阳洁
2023-07-28 ddf66cf155af3805323ec55ae6814f2d23d40c07
po/login_page.py
@@ -1,65 +1,43 @@
from base.base_page import BasePage
from elements.login_elements import LoginElements as Login
from elements.login_elements import LoginElements
from selenium.common.exceptions import TimeoutException
class LoginPage(BasePage):
    """
    登录页面
    """
    # 页面元素
    es = LoginElements
    def username_input(self, username):
        """
        填写用户名
        :param username: 用户名
        :return:
        """
        self.send_text(username, Login.username_input)
    def __init__(self, driver):
        super().__init__(driver)
        self.driver = driver
        # 切换到最顶层文档
        driver.switch_to.default_content()
    def password_input(self, password):
        """
        填写密码
        :param password: 密码
        :return:
        """
        self.send_text(password, Login.password_input)
    def login(self, username, password, code):
        # 输入用户名
        self.fill(self.es.username_input, username)
        # 输入密码
        self.fill(self.es.password_input, password)
        # 输入验证码
        self.fill(self.es.validate_code_input, code)
        # 点击登录按钮
        self.click(self.es.login_btn)
    def validate_code_input(self, validate_code):
        """
        填写验证码
        :param validate_code: 验证码
        :return:
        """
        self.send_text(validate_code, Login.validate_code_input)
    def login_btn(self):
        """
        点击登录按钮
        :return:
        """
        self.click(Login.login_btn)
    def login(self, username, password, validate_code):
        """
        登录
        :param username: 用户名
        :param password: 密码
        :param validate_code: 验证码
        :return:
        """
        self.username_input(username)
        self.password_input(password)
        self.validate_code_input(validate_code)
        self.login_btn()
    def is_login_success(self):
    def login_assert(self):
        """
        判断是否登录成功
        :return: bool
        """
        current_url = self.driver.current_url
        if "index" in self.driver.current_url:
        try:
            ele = self.get_ele(self.es.home_logo, timeout=10)
            if ele.text == "TAI-OTS":
            return True
        else:
        except TimeoutException:
            return False