from base.base_page import BasePage
from elements.login_elements import LoginElements as Login


class LoginPage(BasePage):
    """
    登录页面
    """

    def username_input(self, username):
        """
        填写用户名
        :param username: 用户名
        :return:
        """
        self.send_text(username, Login.username_input)

    def password_input(self, password):
        """
        填写密码
        :param password: 密码
        :return:
        """
        self.send_text(password, Login.password_input)

    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):
        """
        判断是否登录成功
        :return: bool
        """
        current_url = self.driver.current_url
        if "index" in self.driver.current_url:
            return True
        else:
            return False