您的位置:

implicitly_wait详解

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 更为好用,因为它可以更为精细地对待可见性的问题,并且不需要猜测一个合适的等待时间。

参考资料