selenium

python爬虫之selenium--等待的三种方式

发表于 2021年12月29日阅读 2925 评论 0

前言

在我们做UI自动化的时候，最不稳定的就是页面了，由于各种原因页面的元素没有加载出来，比如网速慢，服务器响应慢等等，这时候如果我们去操作页面的元素，selenium会给我们抛出一个NoSuchElementException的异常。我们可以等待页面页面加载出来或者一段时间再去操作，这样的话大大降低了这种错误的出现。

我们可以用python内置的sleep来强制等待，或者用selenium提供的显示等待或者隐式等待的方法

相关模块：
WebDriverWait 显示等待针对元素必用
expected_conditions 预期条件类（用于显示等待）
NoSuchElementException 用于隐式等待抛出异常

sleep

sleep强制等待，不管元素有没有加载出来，都必须等到时间才会往下去执行

from selenium import webdriver

from time import sleep

driver = webdriver.Chrome()
driver.get('http://www.baidu.com')
sleep(2)
driver.find_element_by_css_selector('#kw').send_keys('python')
sleep(2)
driver.quit()

隐式等待

隐式等待不针对某一个元素进行等待，全局元素等待，隐式等待会等待整个页面加载完成，也就是说浏览器窗口标签栏中不在出现转动的小圆圈，才会执行下一步

from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from time import sleep, ctime

driver = webdriver.Chrome()
driver.get('http://www.baidu.com')
sleep(2)
# 设置隐式等待，最多等待 5s，如果五秒内什么时候加载完就什么时候进行操作
driver.implicitly_wait(5)
try:
    print(ctime())
    driver.find_element_by_css_selector('#kw').send_keys('python')
    driver.find_element_by_css_selector('#su').click()
except NoSuchElementException  as msg:
    print(msg)
finally:
    print(ctime())
sleep(2)
driver.quit()

显示等待

显示等待是针对某一个元素进行相关等待判定

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

driver = webdriver.Chrome()
driver.get('http://www.baidu.com')
sleep(2)
driver.find_element_by_css_selector('#kw').send_keys('python')

# 一共等待 5 秒钟，每 0.5s 找一次，直到通过 ID 找到
element = WebDriverWait(driver, 5, 0.5).until(EC.presence_of_element_located((By.ID, 'su')))

element.click()
sleep(2)
driver.quit()