from base.base_page import BasePage
from elements.login_elements import LoginElements
from selenium.common.exceptions import TimeoutException


class LoginPage(BasePage):
    """
    登录页面
    """
    # 页面元素
    es = LoginElements

    def __init__(self, driver):
        super().__init__(driver)
        self.driver = driver
        # 切换到最顶层文档
        driver.switch_to.default_content()

    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 login_assert(self):
        """
        判断是否登录成功
        :return: bool
        """
        try:
            ele = self.get_ele(self.es.home_logo, timeout=10)
            if ele.text == "TAI-OTS":
                return True
        except TimeoutException:
            return False