126邮箱新建一个联系人自动化脚本

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
    driver = webdriver.Chrome(executable_path="d:\\chromedriver")
    url = "http://www.126.com"
    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("xxx")
    
        # 直接获取密码输入框
        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 add_contact():
    global driver
    global wait
    try:
        #显式等待“通讯录”按钮可见,且获取元素
        address_book = wait.until(EC.element_to_be_clickable(("xpath",'//div[.="通讯录"]')))
    
        address_book.click()
   
        #显式等待新建联系人按钮可见，且获取按钮元素
        new_contact = wait.until(lambda x : x.find_element_by_xpath('//span[.="新建联系人"]'))
        #wait.until(EC.element_to_clickable('xpath','//span[.="新建联系人"]'))

        new_contact.click()
    
        #显式等待新建联系人页面的“姓名”输入框出现,并获取元素对象
        nameBox = wait.until(EC.visibility_of_element_located(('id','input_N')))
        nameBox.send_keys("hhhhhh")
    
        #分别获取邮箱、手机号等其他元素对象
        emailBox = driver.find_element_by_xpath('//div[@id="iaddress_MAIL_wrap"]//input')
        emailBox.send_keys("[e-mail]")

        
        #获取设置星标联系人复选框
        starBox = driver.find_element_by_xpath(\
             '//span[.="设为星标联系人" and @class="nui-chk-text"]//preceding-sibling::*/b')
        starBox.click()
       
        phoneBox = driver.find_element_by_xpath(\
              '//div[@id="iaddress_TEL_wrap"]//input')

        phoneBox.send_keys("13122222222")

        
        commentBox = driver.find_element_by_id("input_DETAIL")
        commentBox.send_keys("hhhh")
        
        #点击确定按钮
        driver.find_element_by_xpath('//span[.="确 定"]').click()

        #利用新增联系人后生成的页面元素定位
        #emailStr = "[e-mail]"
        if wait.until(EC.visibility_of_element_located(('xpath','//nobr[.="[e-mail]"]'))):
            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 main():
    global driver
    global wait
    
    
    login_126()  
    add_contact()
    close_driver()
    


if __name__ == "__main__":
    main()