Selenium Python：解决点击事件后代码阻塞与新窗口加载问题

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

# 假设 driver 已经被初始化
# driver = webdriver.Chrome()
# driver.get("your_initial_url")

print("准备点击按钮...")
button_xpath = "//div[text()='Clique para visualizar']"

try:
    # 等待按钮可点击
    WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, button_xpath))).click()
    print("按钮已成功点击。")
except Exception as e:
    print(f"点击按钮失败: {e}")
    # 根据需要进行错误处理，例如截图、退出等

import time # 仅用于示例中的短暂暂停，实际应优先使用 WebDriverWait

# ... (接上文代码)

original_window_handle = driver.current_window_handle
initial_window_handles = driver.window_handles # 在点击前获取所有窗口句柄

# ... (点击按钮的代码) ...

# 等待新窗口出现
try:
    # 等待直到窗口数量增加
    WebDriverWait(driver, 15).until(EC.new_window_is_opened(initial_window_handles))
    # 或者，如果确定只会打开一个新窗口，可以等待窗口数量变为初始数量+1
    # WebDriverWait(driver, 15).until(EC.number_of_windows_to_be(len(initial_window_handles) + 1))

    all_window_handles = driver.window_handles
    new_window_handle = None

    # 遍历所有句柄，找到与原始句柄不同的那个
    for handle in all_window_handles:
        if handle != original_window_handle:
            new_window_handle = handle
            break

    if new_window_handle:
        driver.switch_to.window(new_window_handle)
        print(f"已成功切换到新窗口，标题为: {driver.title}")

        # 在新窗口中等待特定元素加载，确保页面稳定
        # 替换为新页面上实际存在的元素
        new_page_element_xpath = "//h1[contains(text(), '欢迎来到新页面')]"
        WebDriverWait(driver, 20).until(EC.presence_of_element_located((By.XPATH, new_page_element_xpath)))
        print("新窗口内容已加载完毕，可以继续在新窗口中操作。")

        # 在新窗口中执行其他操作，例如输入文本、点击元素等
        # driver.find_element(By.ID, "some_input_field").send_keys("Hello New Window!")

        # 完成新窗口操作后，可以选择关闭新窗口并切换回原始窗口
        # driver.close() # 关闭当前焦点所在的窗口（即新窗口）
        # driver.switch_to.window(original_window_handle)
        # print(f"已关闭新窗口并切换回原始窗口，当前标题: {driver.title}")

    else:
        print("错误：未能找到新打开的窗口。")

except Exception as e:
    print(f"等待或切换新窗口时发生错误: {e}")
    # 错误处理逻辑

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
import time

# 初始化 WebDriver (这里以 Chrome 为例)
driver = webdriver.Chrome()
driver.maximize_window() # 最大化窗口以确保元素可见
driver.get("http://example.com/page_with_button_opening_new_window") # 替换为你的初始URL

print("访问初始页面...")

try:
    # 1. 获取原始窗口句柄和当前所有窗口句柄列表
    original_window_handle = driver.current_window_handle
    initial_window_handles = driver.window_handles
    print(f"原始窗口句柄: {original_window_handle}")
    print(f"初始窗口数量: {len(initial_window_handles)}")

    # 2. 等待按钮可点击并执行点击操作
    button_xpath = "//div[text()='Clique para visualizar']" # 替换为你的按钮XPath
    print("准备点击按钮...")
    WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, button_xpath))).click()
    print("按钮已成功点击。")

    # 3. 等待新窗口出现
    print("等待新窗口打开...")
    WebDriverWait(driver, 15).until(EC.new_window_is_opened(initial_window_handles))

    # 4. 获取所有窗口句柄，并识别新窗口
    all_window_handles = driver.window_handles
    new_window_handle = None
    for handle in all_window_handles:
        if handle != original_window_handle:
            new_window_handle = handle
            break

    if new_window_handle:
        # 5. 切换到新窗口
        driver.switch_to.window(new_window_handle)
        print(f"已切换到新窗口，标题为: {driver.title}")

        # 6. 在新窗口中等待元素加载，确保页面稳定后进行操作
        # 替换为新页面上实际存在的元素，例如一个标题或某个关键组件
        new_page_element_xpath = "//h1[contains(text(), '新窗口内容')]" # 假设新窗口有一个H1标题
        print(f"在新窗口中等待元素: {new_page_element_xpath}")
        WebDriverWait(driver, 20).until(EC.presence_of_element_located((By.XPATH, new_page_element_xpath)))
        print("新窗口内容已加载完毕，可以继续操作。")

        # 在这里执行新窗口中的自动化操作
        # driver.find_element(By.ID, "some_input_on_new_page").send_keys("Test Data")
        # driver.find_element(By.XPATH, "//button[text()='Submit']").click()

        # 7. (可选) 完成操作后，关闭新窗口并切换回原始窗口
        # driver.close()
        # driver.switch_to.window(original_window_handle)
        # print(f"已关闭新窗口并切换回原始窗口，当前标题: {driver.title}")

    else:
        print("错误：未能识别新打开的窗口。")

except Exception as e:
    print(f"自动化过程中发生错误: {e}")

finally:
    # 确保在任何情况下都关闭浏览器
    print("自动化测试完成，即将关闭浏览器。")
    driver.quit()