implicitly_wait详解

小蓝 • 2024-12-22 15:42 • 编程

webdriver 提供了两种等待方式，一种是显式等待，另一种是隐式等待。implicitly_wait 函数就是隐式等待代表，主要用于处理元素不存在的情况以及元素加载时间不够的情况。本文将从多个方面对implicitly_wait做详细的阐述。

一、等待时间设置

implicitly_wait函数默认等待时间是0秒，也就是说webdriver在启动的时候默认是不会等待的。我们可以手动设置一个等待时间。下面是一个等待10秒的例子：

from selenium import webdriver

driver = webdriver.Chrome()
driver.implicitly_wait(10)

可以看到，我们需要在创建driver后，再通过implicitly_wait()方法设置等待时间。同时，需要注意传入的参数是秒。在等待的过程中，webdriver会轮询去查看元素是否存在，如果元素在等待时间内存在，webdriver会马上执行下一步操作；如果元素不存在，webdriver会将等待时间完全耗尽，然后再往下执行下一步操作。

二、等待条件设置

除了等待时间，我们也可以通过设置等待条件来实现更为精细化的特定等待效果。implicitly_wait()方法中支持的条件类型可以参考官方文档。以visibility_of_element_located为例：

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

driver = webdriver.Chrome()
wait = WebDriverWait(driver, 10)
wait.until(EC.visibility_of_element_located((By.XPATH, '//input[@name="username"]')))

上述代码等待10秒，直到元素“username”可见，才继续往下执行操作。如果等待时间过期，webdriver 会抛出 timeout error。

三、等待前后顺序

implicitly_wait() 属于隐式等待，它会在每一个操作执行前，先进行等待检测。这意味着如果我们放在代码的最前面，那么等待时间会被应用到全部的find element 操作中。当然我们也可以在代码的后面逐个添加 wait 检测，除了重复且不能仅检测想要等待的元素外，这种方式会使我们的代码变得冗长。

一种常见的方法是将element 操作的代码放在 helper function 中，并且将其和等待时间包裹在同一个 block 中。下面以login 动作为例：

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

wait_time = 10

def login(driver):
    wait.until(EC.visibility_of_element_located((By.XPATH, '//input[@name="username"]')))
    username = driver.find_element(By.XPATH, '//input[@name="username"]')
    password = driver.find_element(By.XPATH, '//input[@name="password"]')
    submit = driver.find_element(By.XPATH, '//input[@name="submit"]')
    username.send_keys("user")
    password.send_keys("password")
    submit.click()

driver = webdriver.Chrome()
wait = WebDriverWait(driver, wait_time)
driver.get('https://www.baidu.com')
login(driver)

通过这种方式，我们可以将等待包裹在driver 实例中，因为每次在调用login 动作时，webdriver 都会去等待所有的element。

四、等待的陷阱

最后需要提到的是，implicitly_wait存在一些陷阱。例如：如果等待时间过短，webdriver可能还没有来得及检测元素，就已经执行下一个操作，这会导致找不到元素的错误。反之则会导致测试用例执行很慢。

遇到这种情况，我们可以使用更为精细化的等待条件来规避问题。例如，对于一个使用 Ajax 加载信息的 page，使用 visibility_of_element_located 就比 implicitly_wait 更为好用，因为它可以更为精细地对待可见性的问题，并且不需要猜测一个合适的等待时间。

参考资料

Selenium官方文档

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/284983.html