Driver 驱动和 login 封装

from selenium import webdriver
from utils.mysettings import url,username,password

class Driver:
    # 初始化为空
    _driver = None
    @classmethod
    def getDriver(cls,browername='chrome'):
        if cls._driver is None:
            if browername == 'chrome':
                cls._driver = webdriver.Chrome(r'D:\tools\chromedriver_89\chromedriver.exe')
            elif browername == 'firefox':
                cls._driver = webdriver.Chrome(r'D:\tools\chromedriver_89\chromedriver.exe')
            else:
                raise ('brower 配置路径问题')


            cls._driver.maximize_window()
            cls._driver.implicitly_wait(5)
            cls._driver.get(url)
            cls.login_wangxiao()

        return cls._driver



    @classmethod
    def login_wangxiao(cls):

        # 首页登录按钮
        cls._driver.find_element_by_css_selector('div[class="section-status"] div:nth-child(1)>a:nth-child(1)').click()
        # 输入账号
        cls._driver.find_element_by_css_selector('[type="text"]').send_keys(username)
        # 输入密码
        cls._driver.find_element_by_css_selector('[type="password"]').send_keys(password)
        # 登录按钮
        cls._driver.find_element_by_link_text('登录').click()


if __name__ == '__main__':
    Driver.getDriver('chrome')