python3+selenium3登录封装与调用(把登陆做成公共模块)

 

登录封装与调用

 

一、登录封装

把登录写成一个类，里面写登录的方法，保存文件为login.py，下面作为我写的一个例子，方便参考

class Loginfor():

    # 封装
    def __init__(self, driver):
        self.driver = driver

    # 定义登陆函数，将登陆作为公共调用的模块，进行数据传递,因此不需要导入webdriver这个模块
    def login(self, username, password, code):
        # 输入用户名和密码,点击登录
        self.driver.find_element_by_id("userName").clear()
        self.driver.find_element_by_id("userName").send_keys(username)
        self.driver.find_element_by_id("password").clear()
        self.driver.find_element_by_id("password").send_keys(password)
        self.driver.find_element_by_class_name("code").clear()
        self.driver.find_element_by_class_name("code").send_keys(code)
        self.driver.find_element_by_xpath(
            "//button[@class='ant-btn__block login-button ant-btn ant-btn-primary ant-btn-lg']").submit()

 

二、调用

import unittest
from selenium import webdriver
from itracker login import Loginfor


class PatientTrouble(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.maximize_window()
        self.driver.get("XXXXXXXXXXXXXX")
        self.driver.implicitly_wait(15)

    def test_trouble(self):
        Loginfor(self.driver).login("XXXXXX", "XXX", "XXXX")
       其他的你想要继续执行的步骤
       
        self.driver.close()

    def tearDown(self):
        pass

    if __name__ == "__main__":
        unittest.main(verbosity=2)