重写查找元素的方法加入显式等待

Wink1993

于 2020-06-04 09:30:47 发布

阅读量277

点赞数

分类专栏：自动化测试学习

本文链接：https://blog.youkuaiyun.com/weixin_42578892/article/details/106538613

版权

通过在find_elements和find_element中应用显式等待，可以替代传统的sleep强制等待，提高自动化脚本执行效率，避免不必要的等待时间。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from selenium.webdriver.support.ui import WebDriverWait
from selenium.common.exceptions import NoSuchElementException, TimeoutException
from selenium import webdriver


class Base(object):
    def __init__(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()

    def find_element(self, by, locator, timeout=30):
        """

        :param by: 定位方式 eg:By.ID
        :param locator: 定位表达式
        :param timeout: 显式等待的超时时间
        :return:
        """
        try:
            element = WebDriverWait(self.driver, timeout).until(lambda driver: driver.find_element(by, locator))
            # WebDriverWait(self.driver, timeout, 0.5).until(EC.visibility_of_element_located(by, locator))
        except(NoSuchElementException, TimeoutException) as e:
            raise e
        else:
            return element

    def find_elements(self, by, locator, timeout=30):
        """
        :param by: 定位方法eg:By.ID

最低0.47元/天解锁文章