from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.common.exceptions import TimeoutException, NoSuchElementException
import time, traceback

driver = None
wait = None

#登录函数
def login_126():

    global driver
    global wait
    url = "http://www.126.com"
    driver = webdriver.Chrome(executable_path="d:\\chromedriver")
    driver.get(url)
    # 定义显式等待对象
    wait = WebDriverWait(driver, 10, 0.2)
    try:
        # 切入登录框所在的iframe
        wait.until(EC.frame_to_be_available_and_switch_to_it(( \
            By.XPATH, '//iframe[contains(@id,"x-URS-iframe")]')))
        time.sleep(3)
        # driver.switch_to.frame(driver.find_element_by_xpath(\
        # '//iframe[contains(@id,"x-URS-iframe")]'))

        # 显式等待用户名输入框可见且获取输入框元素对象
        username = wait.until(lambda x: x.find_element_by_xpath( \
            '//input[@placeholder="邮箱帐号或手机号"]'))

        username.send_keys("xxxx")

        # 直接获取密码输入框
        passwd = driver.find_element_by_xpath('//input[@name="password"]')
        passwd.send_keys("xxxxx")

        # 获取登录按钮并点击
        driver.find_element(By.ID, "dologin").click()

        #且出到正文
        driver.switch_to.default_content()

        # 显式等待登录成功后“退出”按钮出现在页面上
        if wait.until(EC.visibility_of_element_located(('xpath', '//a[.="退出"]'))):
            print("登录成功！")
        else:
            print("登录失败！")

    except NoSuchElementException as e:
        print(traceback.print_exc())
    except TimeoutException as e:
        print(traceback.print_exc())
    except Exception as e:
        print(traceback.print_exc())

#退出驱动并关闭所有浏览器窗口
def close_driver():
    global driver
    driver.quit()

def send_mail():
    global driver
    global wait
    try:
        #显式等待写信按钮可点击并获取写信按钮元素对象
        write_button = wait.until(EC.element_to_be_clickable(('xpath','//span[.="写 信"]')))
        write_button.click()

        #显式等待收件人输入框可见
        receiver = wait.until(lambda x : x.find_element_by_xpath(\
                 '//input[@class="nui-editableAddr-ipt"]'))
        #receiver = wait.until(EC.visibility_of_element_located(('xpath','//span[.="写 信"]')))
        receiver.send_keys("hhq8286@126.com")

        #获取主题输入框页面元素对象
        subject = driver.find_element_by_xpath('//input[contains(@id,"subjectInput")]')
        subject.send_keys("2019你要怎么做？")

        #获取添加附件元素对象（用send_keys()需要定位到input元素）
        attachment = driver.find_element_by_xpath('//input[@class="O0"]')

        attachment.send_keys("e:\\python\\111.txt")

        #等待附件上传完成
        wait.until(EC.visibility_of_element_located(('xpath','//span[.="上传完成"]')))
        print("附件上传完成！")

        time.sleep(3)

        #切入正文frame 
        wait.until(EC.frame_to_be_available_and_switch_to_it(\
             ('xpath','//iframe[@class="APP-editor-iframe"]')))

        contentBox = driver.find_element_by_xpath('/html/body')
        contentBox.send_keys("学习 理财 看书 如何转更多钱钱，不为钱而工作")

        #切出frame，切到正文
        driver.switch_to.default_content()

        #发送
        driver.find_element_by_xpath(\
            '//footer[@class="jp0"]//span[text()="发送"]//preceding-sibling::span//b').click()

        #显式等待发送成功的//h2[contains(@id,"succInfo")]    //h2[text()="发送成功"]
        if wait.until(EC.visibility_of_element_located(('xpath','//h2[text()="发送成功"]'))):
            print("邮件发送成功！")
        else:
            print("邮件发送失败！")

        time.sleep(5)

    except NoSuchElementException as e:
        print(traceback.print_exc())
    except TimeoutException as e:
        print(traceback.print_exc())
    except Exception as e:
        print(traceback.print_exc())

def main():
    login_126()  
    send_mail()
    close_driver()

if __name__ == "__main__":
    main()