Selenium Python中处理点击后新窗口加载冻结问题的策略与实践

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
# driver = webdriver.Chrome() # 或 Firefox, Edge等
# driver.get("your_initial_url_here") # 替换为你的初始URL

# 为了演示，我们创建一个虚拟的WebDriver实例和方法
class MockWebDriver:
    def __init__(self):
        self._window_handles = {"main_window_handle"}
        self._current_window_handle = "main_window_handle"
        print("Mock WebDriver initialized.")

    @property
    def window_handles(self):
        return self._window_handles

    @property
    def current_window_handle(self):
        return self._current_window_handle

    def find_element(self, by, value):
        print(f"Finding element by {by}: {value}")
        # 模拟找到元素并返回一个可点击的对象
        class MockElement:
            def click(self_element):
                print(f"I gonna click in the toaster (mock click on {value})")
                # 模拟点击后打开新窗口
                time.sleep(1) # 模拟网络延迟
                self._window_handles.add("new_window_handle")
                print("Mock: New window handle 'new_window_handle' added.")
                # 注意：实际Selenium不会自动切换，这里只是模拟句柄出现
                # 真实场景下，driver.click()后，需要我们手动切换
                print("Mock: Button clicked, new window initiated.")
        return MockElement()

    def switch_to(self):
        class SwitchTo:
            def window(self_switch, handle):
                self._current_window_handle = handle
                print(f"Switched to window: {handle}")
        return SwitchTo()

    def get(self, url):
        print(f"Navigating to {url}")
        # 模拟页面加载
        time.sleep(2)

# 替换为你的真实WebDriver实例
driver = MockWebDriver() # 真实场景请使用 driver = webdriver.Chrome() 等

# 示例开始
try:
    # 模拟导航到初始页面
    driver.get("http://example.com/initial_page")

    # 1. 获取点击前所有窗口句柄
    original_window_handles = set(driver.window_handles)
    print(f"Original window handles: {original_window_handles}")

    # 2. 执行点击操作，该操作会打开一个新窗口
    # 假设你的按钮 XPath 是 "//div[text()='Clique para visualizar']"
    print("准备点击按钮...")
    driver.find_element(By.XPATH, "//div[text()='Clique para visualizar']").click()
    print("点击操作已触发。")

    # 3. 等待新窗口句柄出现
    # 设置一个WebDriverWait实例，最长等待10秒
    wait = WebDriverWait(driver, 10)

    # 自定义等待条件：等待窗口句柄数量增加
    # 注意：在真实Selenium中，你不需要传入 driver.window_handles
    # EC.number_of_windows_to_be(len(original_window_handles) + 1) 是一个更简洁的条件
    wait.until(EC.number_of_windows_to_be(len(original_window_handles) + 1))
    print("新窗口句柄已出现。")

    # 4. 识别并切换到新窗口
    new_window_handle = None
    for handle in driver.window_handles:
        if handle not in original_window_handles:
            new_window_handle = handle
            break

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

        # 5. 在新窗口中等待特定元素加载（例如，新页面的某个标题或内容）
        # 假设新窗口中有一个ID为'new_page_title'的元素
        # 注意：在MockWebDriver中，我们没有实际的DOM，这里只是模拟等待
        try:
            # wait.until(EC.presence_of_element_located((By.ID, "new_page_title")))
            # 模拟等待新页面元素加载
            print("在新窗口中等待关键元素加载...")
            time.sleep(2) # 模拟新页面加载时间
            print("新窗口关键元素已加载。")
            print(f"当前驱动器位于新窗口: {driver.current_window_handle}")
            # 在新窗口中执行其他操作...
        except Exception as e:
            print(f"在新窗口中等待元素失败: {e}")
            raise
    else:
        print("未找到新的窗口句柄。")

except Exception as e:
    print(f"An error occurred: {e}")
finally:
    # 真实WebDriver需要关闭
    # driver.quit()
    print("示例结束。")