WebDriver 提供了两种类型的等待:显式等待和隐式等待。
显式等待
显式等待使 WebdDriver 等待某个条件成立时继续执行,否则在达到最大时长时抛出超时异常
(TimeoutException)。
from selenium import webdriver
from selenium.webdriver.common.by import By #里面包含xpath、id、name等
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
driver = webdriver.Firefox()
driver.get("http://www.baidu.com")
element = WebDriverWait(driver, 5, 0.5).until(
EC.presence_of_element_located((By.ID, "kw")) )#by里面也包含xpath、name等
element.send_keys('selenium')
driver.quit()
WebDriverWait 类是由 WebDirver 提供的等待方法。在设置时间内,默认每隔一段时间检测一次当前页面元素是否存在,如果超过设置时间检测不到则抛出异常。具体格式如下:
WebDriverWait(driver, timeout, poll_frequency=0.5, ignored_exceptions=None)
driver :浏览器驱动。
timeout :最长超时时间,默认以秒为单位。
ignored_exceptions :超时后的异常信息,默认情况下抛 NoSuchElementException 异常。
WebDriverWait()一般由 until()或 until_not()方法配合使用,下面是 until()和 until_not()方法的说明。
until(method, message=‘ ’)
调用该方法提供的驱动程序作为一个参数,直到返回值为 True。
until_not(method, message=’ ’)
调用该方法提供的驱动程序作为一个参数,直到返回值为 False。
在本例中,通过 as 关键字将 expected_conditions 重命名为 EC,并调用 presence_of_element_located()方法判断元素是否存在。 expected_conditions 类所提供的预期条件判断的方法如表 4.3 所示。
除 expected_conditions 所提供的丰富的预期条件判断方法外,还可以使用前面学过的 is_displayed()方法来判断元素是否可见。
from selenium import webdriver from time import sleep, ctime
driver = webdriver.Firefox()
driver.get("http://www.baidu.com")
print(ctime())
for i in range(10):
try:
el = driver.find_element_by_id("kw22")
if el.is_displayed():
break
except:
pass
sleep(1)
else: print("time out")
driver.close()
print(ctime())
相对来说,这种方式更容易理解,通过 for 循环 10 次,每次循环判断元素的 is_displayed()状态是否为 True:
如果为 True,则 break 跳出循环;否则 sleep(1)后继续循环判断,直到 10 次循环结束后,打印“time out”信息。执行结果如下:
========================= RESTART: D:/pyse/baidu.py =========================
Fri Oct 23 22:51:25 2015 time out
Fri Oct 23 22:51:35 2015
隐式等待
WebDriver 提供了 implicitly_wait()方法来实现隐式等待,默认设置为 0。它的用法相对来说要简单得多。
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from time import ctime
driver = webdriver.Firefox()
# 设置隐式等待为10秒
driver.implicitly_wait(10)
driver.get("http://www.baidu.com")
try:
print(ctime())
driver.find_element_by_id("kw22").send_keys('selenium')
except NoSuchElementException as e:
print(e)
finally: print(ctime())
driver.quit()
implicitly_wait()默认参数的单位为秒,本例中设置等待时长为 10 秒。首先这 10 秒并非一个固定的等待时间,它并不影响脚本的执行速度。其次,它并不针对页面上的某一元素进行等待。当脚本执行到某个元素定位时,如果元素可以定位,则继续执行;如果元素定位不到,则它将以轮询的方式不断地判断元素是否被定位到。假设在第 6 秒定位到了元素则继续执行,若直到超出设置时长(10 秒)还没有定位到元素,则抛出异常。