python3 selenium自动化测试 登陆新浪邮箱之函数的封装和调用

python3 selenium自动化测试 登陆新浪邮箱之函数的封装和调用
以下是具体的脚本。

from selenium import webdriver
from time import sleep
dr = webdriver.Chrome()
'''操作步骤如下：
   1.打开邮箱网页
   2.输入账户
   3.输入密码
   4.点击登陆
   5.退出邮箱账户
   6.关闭浏览器'''

'''下面函数的封装'''
#1.打开邮箱网页
def openUrl(url):
    return dr.get(url)

#2.输入账户
def input_account(locator,account):
    return dr.find_element_by_xpath(locator).send_keys(account)

#3.输入密码
def input_password(locator,password):
    return dr.find_element_by_xpath(locator).send_keys(password)

#4.点击登陆
def login(locator):
    return dr.find_element_by_xpath(locator).click()

#5.退出邮箱账户
def logOut(locator):
    return dr.find_element_by_xpath(locator).click()

#7.关闭浏览器
def close_brower():
    return dr.quit()


'''下面是函数的调用'''
URL = 'https://mail.sina.com.cn/'
openUrl(URL)
sleep(3)
dr.maximize_window()
sleep(3)
input_account('//*[@id="freename"]','xxxxx@sina.com')
input_password('//*[@id="freepassword"]','xxxxxxxxxxx')
login('/html/body/div[1]/div/div[2]/div/div/div[4]/div[1]/div[1]/div[7]/div[1]/a[1]')
sleep(2)
logOut('//*[@id="warpMain"]/div[4]/div[1]/div[3]/div[3]/span/a')
sleep(2)
close_brower()