我们通过 WebDriver 查找控件或者操作控件的时候，如果被查找或被操作的控件不存在，就会报异常。其实很多种情况下由于网速问题，我们查找或操作的控件还没来得及出现，这时候 WebDriver
的代码就会误以为控件不存在而报异常。如果我们的代码能等待一段时间就好了，就是说在一定时间内能查找到或操作到这个控件是合理的，超越我们定的时间就认为控件不存在，然后报异常。
WebDriver 提供了两种类型的等待：显式等待和隐式等待。
WebDriverWait 是由 WebDirver 提供的显示等待类。在设置时间内，默认每隔一段时间检测一次当前页面元素是否存在，如果超过设置时间检测不到则抛出异常。
implicitly_wait() 是由 WebDriver 提供的隐式等待方法，默认设置为 0。它的用法相对来说要简单得多。
显式等待
显式等待使 WebdDriver 等待某个条件成立时继续执行，否则在达到最大时长时抛出超时异常(TimeoutException)。
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
driver = webdriver.Chrome()
driver.get(“http://www.baidu.com”)
”’
类：
WebDriverWait(driver, timeout, poll_frequency=0.5, ignored_exceptions=None)
参数：
driver ：浏览器驱动。
timeout ：最长超时时间，默认以秒为单位。
poll_frequency ：检测的间隔(步长)时间，默认为0.5S。
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() 方法判断元素是否存在。
”’
element = WebDriverWait(driver, 5, 0.5).until(EC.presence_of_element_located((By.ID, “kw”)))
element.send_keys(‘selenium’)
driver.quit()
隐式等待
implicitly_wait() 默认参数的单位为秒，我们下面的例子中设置等待时长为 10 秒。首先这 10
秒并非一个固定的等待时间，它并不影响脚本的执行速度。其次，它并不针对页面上的某一元素进行等待。当脚本执行到某个元素定位时，如果元素可以定位，则继续执行；如果元素定位不到，则它将以轮询的方式不断地判断元素是否被定位到。假设在第
6 秒定位到了元素则继续执行，若直到超出设置时长(10秒)还没有定位到元素，则抛出异常。
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()
自定义等待
如果我们不想使用 selenium 自带的显示等待或隐式等待方法，我们完全可以自己写出等待的逻辑代码。
import time
”’
功能：
securityfind 函数在规定时间查找到控件并返回该控件，否则返回 None，默认查找 30 次，每次查找后等待 1 秒。
参数说明：
driver：控制浏览器的 webdriver。
tagname：要查找的控件名称，打印出来，给用户自己自己看的。
strtag：查找语法。
flag：查找方法，比如 by_id，by_class_name 等等。
intertime：查找一次等待多久，默认 1 秒。
count：总共查找多少次，默认 30 次。
”’
def securityfind(driver, tagname, strtag, flag, intertime = 1, count = 30):
nowcount = 0
while nowcount < count:
time.sleep(intertime)
try:
if flag == “by_id”:
result = driver.find_element_by_id(strtag)
elif flag == “by_name”:
result = driver.find_element_by_name(strtag)
elif flag == “by_class_name”:
result = driver.find_element_by_class_name(strtag)
elif flag == “by_tag_name”:
result = driver.find_element_by_tag_name(strtag)
elif flag == “by_link_text”:
result = driver.find_element_by_link_text(strtag)
elif flag == “by_partial_link_text”:
result = driver.find_element_by_partial_link_text(strtag)
elif flag == “by_xpath”:
result = driver.find_element_by_xpath(strtag)
elif flag == “by_css_selector”:
result = driver.find_element_by_css_selector(strtag)
print “规定时间内找到：” + tagname
return result # 找到 tag 就返回
except:
nowcount += 1
print(“没找到” + tagname + str(nowcount) + “次”)
print(“规定时间次数内没找到：” + tagname)
return None
我们现在举例来使用上面我们自己写的查找等待函数。
from selenium import webdriver
import securityfind # 导入我们自己写的函数
driver = webdriver.Chrome()
driver.get(“http://www.baidu.com”)
element = securityfind(driver, “百度搜索框”, “kw”, “by_id”, 1, 10)
if element != None:
element.send_keys(‘selenium’)
else:
print(“控件不存在”)
driver.quit()