如何使用Selenium点击网页动态按钮？Python Jupyter Notebook实现方法

问题原因 #

目标按钮位于页面内嵌的iframe框架中，Selenium默认只会读取顶层页面的DOM结构，所以直接查找元素会返回不存在。同时动态加载的元素需要搭配显式等待逻辑，避免在元素未渲染完成时执行操作。

前置依赖安装 #

执行以下命令安装需要的第三方库：

pip install selenium webdriver-manager

完整实现代码（Jupyter Notebook可直接运行） #

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

# 初始化Chrome驱动
driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()))
driver.maximize_window()
# 打开目标网页
driver.get("http://data.cnmc.es/datagraph/jsp/inf_trim.jsp")
# 初始化显式等待，最长等待时间20秒，可根据网络情况调整
wait = WebDriverWait(driver, 20)

# 等待iframe加载完成，切换到内容所在的iframe
target_iframe = wait.until(EC.presence_of_element_located((By.NAME, "Cuerpo")))
driver.switch_to.frame(target_iframe)

# 等待目标按钮可点击后执行点击，此处以页面的Consulta查询按钮为例
# 若要点击其他按钮，可在Chrome F12开发者工具中右键目标元素，复制XPATH/CSS选择器替换下方定位规则即可
target_btn = wait.until(EC.element_to_be_clickable((By.XPATH, "//input[@value='Consulta']")))
# 常规点击
target_btn.click()
# 如果遇到元素被遮挡无法点击的情况，替换为JS点击即可
# driver.execute_script("arguments[0].click();", target_btn)

# 操作完iframe内元素后，如果需要操作顶层页面的元素，执行以下代码切回顶层DOM
# driver.switch_to.default_content()

注意事项 #

• 显式等待会在元素加载完成后立刻执行后续操作，比固定time.sleep的兼容性和执行效率更高
• 所有iframe内部的元素操作都需要先完成iframe切换，否则都会返回元素不存在的报错
• 若需要调整等待时长，修改WebDriverWait初始化的第二个参数即可，单位为秒

内容的提问来源于stack exchange，提问作者Jaime