basepages 封装显示等待

from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from utils.mysettings import timeout,poll_time
from pages.myDriver import Driver

class Basepage:
    # 初始化驱动
    def __init__(self):
        self.driver = Driver.getDriver()

    def get_element(self,locator):
        WebDriverWait(driver=self.driver,
                      timeout=timeout,
                      poll_frequency=poll_time).until(
            EC.visibility_of_element_located(locator)
        )

        return self.driver.find_element(*locator)


    def get_elements(self, locator):
        WebDriverWait(driver=self.driver,
                      timeout=timeout,
                      poll_frequency=poll_time).until(
            EC.invisibility_of_element_located(locator)
        )
        return self.driver.find_elements(*locator)