Python+selenium(11)脚本模块化&参数化

from selenium import webdriver
from time import sleep
from model1 import Login

driver = webdriver.Chrome()
driver.get("http://www.126.com")

print("登录前")
title = driver.title
print("title:" + title)

#调用登录
l = Login(driver)
#邮箱用户名需输入你自己的真实的126邮箱用户名，密码也需要是真实的
l.login("邮箱用户名","邮箱密码")


url = driver.current_url
print("url:"+url)


sleep(5)

print("登陆后：")
title = driver.title
print("url:"+url)

text = driver.find_element_by_id("spnUid").text
print("user name:"+text)

#断言
assert text == "邮箱用户名"

class Login():
    def __init__(self,driver):
        self.driver = driver

    def login(self,username,password):
        driver = self.driver
        driver.switch_to.frame("x-URS-iframe")
        
        driver.find_element_by_name("email").clear()
        driver.find_element_by_name("email").send_keys(username)
        driver.find_element_by_name("password").clear()
        driver.find_element_by_name("password").send_keys(password)
        driver.find_element_by_id("dologin").click()