快速掌握selenium.webdriver.support.wait在python中的使用

selenium.webdriver.support.wait是Selenium WebDriver中的一个类，用于实现等待某个条件发生后再继续执行下一步操作。在Python中使用selenium.webdriver.support.wait需要导入相关的库和模块，然后创建一个WebDriverWait对象，设置等待条件并设置等待时间，最后调用until或until_not方法等待条件满足或超时。下面是一个使用selenium.webdriver.support.wait的使用例子：

首先，我们需要导入相关的库和模块：

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

然后，创建一个WebDriver对象：

driver = webdriver.Chrome()
driver.get('https://www.example.com')

接下来，我们创建一个WebDriverWait对象，设置等待条件和等待时间：

wait = WebDriverWait(driver, 10)
element = wait.until(EC.presence_of_element_located((By.ID, 'element_id')))

上面的代码中，我们设置了等待10秒钟，直到页面上的元素id为'element_id'出现。如果元素在10秒内出现，则会返回该元素；如果元素在10秒内没有出现，则会抛出一个TimeoutException异常。

我们也可以使用其他的等待条件，比如等待元素可点击：

element = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, 'button')))

上面的代码中，我们设置了等待页面上的一个按钮可点击，然后进行下一步操作。

除了等待元素，还可以等待页面标题或URL发生变化：

wait.until(EC.title_contains('example'))
wait.until(EC.url_contains('example'))

上面的代码中，我们设置了等待页面标题或URL中包含'example'，然后进行下一步操作。

在实际使用中，我们可能需要多次等待来保证页面的稳定性，我们可以使用多个等待条件，例如等待元素的文本内容发生变化：

wait.until(EC.text_to_be_present_in_element((By.CSS_SELECTOR, 'span'), 'new text'))

上面的代码中，我们设置了等待一个span元素的文本内容为'new text'，然后进行下一步操作。

当然，我们还可以自定义一个等待条件，只需要继承selenium.webdriver.support.expected_conditions类并实现__call__方法即可：

class MyCondition(object):
    def __init__(self, locator):
        self.locator = locator
    
    def __call__(self, driver):
        element = driver.find_element(*self.locator)
        if element.get_attribute('class') == 'condition-met':
            return element
        else:
            return False

然后，我们可以使用自定义的等待条件：

element = wait.until(MyCondition((By.ID, 'element_id')))

上面的代码中，我们设置了等待一个元素的class属性为'condition-met'，然后进行下一步操作。

总结来说，selenium.webdriver.support.wait是Selenium WebDriver中用于实现等待机制的类，它提供了多种等待条件，可以灵活地等待页面的变化，并且还可以自定义等待条件。通过合理使用selenium.webdriver.support.wait，我们可以更好地控制自动化测试的执行流程，提高测试的稳定性。