selenium中的三种等待方法

在selenium执行的过程中，我们经常会遇到这样一个问题：当我们使用find_element方法去定位元素时，发现实际页面中存在这个元素，但是我们的代码却抛出NoSuchElementException的异常。

这是因为代码的执行速度比浏览器服务器的相应速度快得多，浏览器还没来得及将结果返回给我们，代码就已经跑到下面去了。

为此，我们需要一个等待时间来等待浏览器的响应结果。selenium有三种常用方法设置等待时间。

1.强制等待

通过time模块中的sleep()方法，让我们的代码在执行到某个位置时强制等待一段时间，如下例：

# 导入time模块
import time
from selenium import webdriver

# 创建webdrive的实例对象
driver = webdriver.Chrome()

# 通过get()方法打开百度
driver.get("https://www.baidu.com/")

# 设置等待时间，等待页面元素加载，10s后代码继续往下执行
time.sleep(10) 

使用sleep()方法的缺点是不够灵活，不能准确把握等待时间。

2.隐式等待

通过webdriver对象的implicitly_wait()方法，该方法一次设置，全局有效。当我们去定位页面元素时，如果元素没有找到，不会立即抛出异常，而是周期性地（通常为 0.5s）去重新寻找，直到该元素找到或者超过最大等待时间才结束（抛出异常或者返回 None），如下例：

# 设置隐式等待,最多等待10s
driver.implicitly_wait(10)

# 定位属性id值为kw的元素时，自动执行隐式等待
driver.find_element("id", "kw")

注意：当我们使用implicitly_wait()时，如果想要定位的元素已经找到，但是它的内容（如文本内容，属性等）没有加载出来，此时隐式等待无效，仍会直接抛出NoSuchElementException异常，这也是为什么我们很多时候仍需要使用time.sleep()的原因。

3.显示等待

不同于隐式等待全局等待的方式，显示等待更加灵活，可以让我们针对某一个元素进行等待设置。

通过WebDriverWait类的until()和until_not()方法来实现。

 until()：条件为成立返回True，等待结束，如果超时，抛出TimeoutException异常。

 until_not()：条件不成立返回True，等待结束，如果超时，抛出TimeoutException。

实现如下：

导入模块

from selenium import webdriver
# 导入 WebDriverWait类
from selenium.webdriver.support.wait import WebDriverWait

调用格式

# 调用格式
WebDriverWait(driver,timeout,poll_frequency,ignored_exception).until(func,error_message)

WebDriverWait(driver,timeout,poll_frequency,ignored_exception).until_not(func,error_message)

"""
driver：webdriver实例对象
timeout：最大等待时间
poll_freqency：调用频率，可选的，默认0.5s一周期
ignored_exception:可选的，要忽略的异常，默认为None
func：要执行的方法
error_message：自定义异常信息，可选的
"""

func参数可以是自定义方法，也可以是webelement对象的方法，也可以是expected_conditions模块中方法，且一定要是可调用的（即该对象拥有__call__()方法），否则就会抛出如下异常：

（1）自定义方法，通常借助匿名函数（lambda）来实现

driver.get("https://www.baidu.com/")

# 自定义方法查找name属性为wd的输入框，查找成功打印消息

flag =  WebDriverWait(driver,10).until(lambda driver:driver.find_element('xpath',"//input[@name='wd']"),"查找不成功")
if flag:
    print("查找成功")

 （2）调用webelement对象的方法，常用的有：

        is_displayed():判断元素是否被展示出来

        is_enable():判断元素是否可操作

        is_select():判断元素是否被选择

tag = WebDriverWait(driver, 10).until(lambda driver: driver.find_element('xpath', "//input[@name='wd']").is_displayed(), "查找不成功")
if tag:
    print("查找成功")

（3）借助webdriver的expected_conditions模块，该模块中包含了一系列用于判断元素特征的方法，常见的有：

        title_is(str):判断当前页面的title是否精确等于预期，返回布尔值

        title_contains(str):判断当前页面的title是否包含预期字符串，返回布尔值

        presence_of_element_located(locator):判断某个元素是否被加到了DOM树中，返回元素对象

        visibility_of_element_located(locator):判断某个元素是否可见，返回元素对象

        invisibility_of_element_located(locator):判断某个元素是否不可见，返回元素对象

        element_to_be_clickable(locator):判断某个元素是否是enable的，返回元素对象

        text_to_be_present_in_element(locator,str):判断某个元素是否包含了预期文本，返回布尔值

# 借助webdriver的expected_conditions模块中的元素特征判断方法
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()
driver.get("https://www.baidu.com/")
# 判断定位的元素是否可以被点击
flag =  WebDriverWait(driver,5).until(EC.element_to_be_clickable(('xpath', "//input[@name='wd']")),"查找不成功")
if flag:
    print("查找成功\n")
# 判断定位的元素是否可见
tag =  WebDriverWait(driver,5).until(EC.visibility_of_element_located(('xpath', "//input[@name='wd']")),"查找不成功")
if tag:
    print("查找成功")