Python Selenium 等待元素出现|极客教程

Python Selenium 等待元素出现

在使用 Selenium 进行 Web 自动化测试时，经常会遇到需要等待页面中某个元素出现的情况。如果在元素加载完成之前就开始执行脚本，可能会导致脚本失败。因此，需要在代码中加入等待元素出现的机制。

本文将介绍如何使用 Python 编写 Selenium 脚本，并通过不同的等待方式，等待页面中的元素出现。

为什么需要等待元素出现

在进行 Web 自动化测试时，页面中的元素并不是立即加载完成的，而是需要一定的时间进行异步加载。如果在元素加载完成之前就开始执行脚本，可能会导致定位不到元素的情况，从而导致脚本执行失败。

为了解决这个问题，我们需要在脚本中等待页面中的元素出现，确保元素加载完成后再进行操作。

Selenium 等待元素出现的方法

Selenium 提供了多种等待元素出现的方法，常见的方法包括隐式等待、显式等待和智能等待。接下来我们将分别介绍这三种方法的用法。

隐式等待

隐式等待是一种全局性的等待方式，它会在整个 WebDriver 生命周期内起作用。当使用隐式等待时，Selenium 将在查找每个元素之前等待一定的时间，如果超过了设定的时间仍然找不到元素，将抛出 ElementNotVisibleException 异常。

from selenium import webdriver

driver = webdriver.Chrome()
driver.implicitly_wait(10)  # 设置隐式等待时间为10秒
driver.get("https://www.example.com")

element = driver.find_element_by_xpath("//input[@id='username']")
element.send_keys("username")

上面的代码中，我们通过 driver.implicitly_wait(10) 设置了隐式等待时间为10秒，然后使用 find_element_by_xpath 查找页面中的用户名输入框。如果在10秒内未找到该元素，将抛出 ElementNotVisibleException 异常。

显式等待

显式等待是一种针对特定元素的等待方式，它可以指定一个最长等待时间，并且可以定义等待条件。只有当指定的条件满足时才会继续执行脚本。

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("https://www.example.com")

element = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.XPATH, "//input[@id='username']"))
)
element.send_keys("username")

上面的代码中，我们通过 WebDriverWait(driver, 10) 设置了最长等待时间为10秒，然后使用 until 方法指定了等待条件为 presence_of_element_located，并通过 By.XPATH 指定了要查找的元素。只有当用户名输入框出现时，才会继续执行脚本。

智能等待

智能等待是一种结合了隐式等待和显式等待的等待方式，它会根据实际情况自动选择适合的等待方式。Selenium 4 中引入了新的智能等待方法 driver.implicitly_wait_and_sleep()，可以在设置的隐式等待时间内自动采用隐式等待，超出隐式等待时间后自动切换为显式等待。

from selenium import webdriver

driver = webdriver.Chrome()
driver.implicitly_wait_and_sleep(10)  # 设置智能等待时间为10秒
driver.get("https://www.example.com")

element = driver.find_element_by_xpath("//input[@id='username']")
element.send_keys("username")

上面的代码中，我们通过 driver.implicitly_wait_and_sleep(10) 设置了智能等待时间为10秒，然后使用 find_element_by_xpath 查找页面中的用户名输入框。如果在10秒内未找到该元素，将自动切换为显式等待。