WebDriverWait封装

WebDriverWait是一个Selenium库中的类，用于等待Web页面上的某个元素出现或消失。为了方便使用，我们可以对其进行封装。以下是一个简单的WebDriverWait封装的示例代码：

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

class WebDriverHelper:
    def __init__(self, driver):
        self.driver = driver

    def wait_for_element(self, locator, timeout=10):
        element = WebDriverWait(self.driver, timeout).until(
            EC.presence_of_element_located((By.XPATH, locator))
        )
        return element

在上面的代码中，我们创建了一个名为WebDriverHelper的类，它包含一个名为wait_for_element的方法。该方法接受两个参数：locator和timeout。locator是一个XPath表达式，用于定位要等待的元素。timeout是等待元素出现的最长时间，单位为秒。

在wait_for_element方法中，我们使用WebDriverWait类来等待元素的出现。如果元素在指定的时间内出现，那么该方法将返回该元素。否则，它将抛出一个TimeoutException异常。

在上面的示例代码中，我们使用了By.XPATH来定位元素。你也可以使用其他的定位方式，比如By.ID、By.NAME等。