import time

from base.base_page import BasePage
from elements.product_accredit__elements import ProductAccreditElements
from selenium.common.exceptions import TimeoutException, NoSuchElementException


class ProductAccreditPage(BasePage):
    """
    登录页面
    """
    # 页面元素
    es = ProductAccreditElements

    def __init__(self, driver):
        super().__init__(driver)
        try:
            self.driver.switch_to.default_content()
            self.switch_iframe(self.es.this_iframe)
        except:
            pass

    def wait_layer_reload_hide(self, timeout=600):
        """
        等待加载框消失
        :param timeout: 超时时间
        :return:
        """
        for i in range(1, 600):
            try:
                # 加载框存在睡眠一秒
                self.get_ele(self.es.layer_reload)
                time.sleep(1)
            except NoSuchElementException:
                # 不存在跳出循环
                break