Python Selenium弹窗处理：自动化下载时切换弹窗的方法求助

问题描述 #

我正在用Selenium实现网页点击及下载自动化操作，目前遇到难题：点击下载按钮后会弹出弹窗，我不知道如何通过Selenium切换到该弹窗进行交互。简言之，我需要与弹窗交互，但不熟悉相关的函数、语法及操作流程，恳请帮助！若找到解决方案会及时更新。

原代码（修正后） #

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import TimeoutException
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service as ChromeService
import time

# Chrome可执行文件路径（按需替换）
chrome_path = "你的Chrome路径"
url = "目标网址"

# 使用ChromeDriverManager初始化WebDriver（推荐方式）
driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()))

# 打开目标URL
driver.get(url)

# 最大化浏览器窗口
driver.maximize_window()

# 提示用户终止程序的方式
print("按CTRL+C停止程序，可关闭Chrome浏览器并在任务管理器中结束Chrome进程")

# 定义主窗口中的元素定位器
iframe_xpath = '//*[@id="viz"]/iframe'
xpath1 = '/html/body/div[2]/div[2]/div[1]/div[2]/div[2]/div[5]/button'
xpath2 = '/html/body/div[6]/div/div/div/div[2]/div'
xpath3 = ''

# 定义弹窗中的元素定位器（需补充）

# 循环检查元素（效率较低，可优化）
try:
    # 等待iframe加载完成
    iframe = WebDriverWait(driver, 15).until(
        EC.presence_of_element_located((By.XPATH, iframe_xpath))
    )

    # 切换到iframe
    driver.switch_to.frame(iframe)

    # 等待目标元素出现并点击
    element = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.XPATH, xpath1))
    )
    element.click()
    
    # 等待第二个元素出现并点击
    element2 = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.XPATH, xpath2))
    )
    element2.click()
    
    # 切回主文档
    driver.switch_to.default_content()

    
except TimeoutException:
    # 超时处理：提示并重试
    print("未找到元素，正在重试...")
    time.sleep(10)  # 按需调整重试间隔

# 切回主文档
driver.switch_to.default_content()

分场景解决方案 #

1. 浏览器原生弹窗（Alert/Confirm/Prompt） #

这类弹窗是浏览器自带的系统级弹窗，Selenium提供直接操作API：

• 切换到弹窗：driver.switch_to.alert
• 核心操作：
  • 获取弹窗文本：alert.text
  • 确认弹窗：alert.accept()
  • 取消弹窗：alert.dismiss()
  • 输入内容（仅Prompt类型弹窗）：alert.send_keys("输入内容")

示例代码：

# 点击按钮触发弹窗后，等待弹窗出现
alert = WebDriverWait(driver, 10).until(EC.alert_is_present())
# 获取弹窗文本
print(alert.text)
# 确认弹窗
alert.accept()
# 如需取消弹窗，替换为：
# alert.dismiss()

2. 网页自定义模态弹窗（Div/Iframe形式） #

这类弹窗是网页通过HTML/CSS/JS实现的，本质是普通DOM元素，无需切换窗口，直接定位操作即可：

1. 确保已切回主文档（如果之前在iframe中）：driver.switch_to.default_content()
2. 用显式等待定位弹窗及内部元素：

示例代码：

# 等待自定义弹窗可见
popup = WebDriverWait(driver, 10).until(
    EC.visibility_of_element_located((By.XPATH, "你的弹窗元素XPATH"))
)
# 定位弹窗内的确认按钮并点击
confirm_btn = popup.find_element(By.XPATH, "确认按钮XPATH")
confirm_btn.click()

3. 新窗口/新标签页弹窗 #

点击按钮后打开新窗口的场景，需通过窗口句柄切换：

1. 记录主窗口句柄
2. 等待新窗口出现后切换

示例代码：

# 点击前记录主窗口句柄
main_window = driver.current_window_handle
# 点击触发新窗口的按钮
element.click()
# 等待新窗口加载完成
WebDriverWait(driver, 10).until(lambda d: len(d.window_handles) > 1)
# 切换到新窗口
for handle in driver.window_handles:
    if handle != main_window:
        driver.switch_to.window(handle)
        break
# 新窗口操作完成后，如需切回主窗口：
# driver.switch_to.window(main_window)

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