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
from selenium.common.exceptions import TimeoutException # 导入TimeoutException
import time 

# 假设已经初始化了WebDriver，例如Chrome
# driver = webdriver.Chrome() 
# driver.get("your_initial_url") # 替换为你的初始URL

# 示例：初始化一个虚拟的WebDriver以使代码可运行
# 在实际应用中，你需要根据浏览器类型进行初始化
class MockWebDriver:
    def __init__(self):
        self.current_window_handle = "original_handle"
        self.window_handles = ["original_handle"]
        self.current_url = "about:blank"
        self.title = "Mock Page"

    def find_element(self, by, value):
        print(f"Mock: 查找元素 by={by}, value={value}")
        if value == "Clique para visualizar":
            # 模拟找到元素
            class MockElement:
                def click(self):
                    print("Mock: 元素被点击，模拟新窗口打开...")
                    # 模拟新窗口打开
                    self.window_handles.append("new_handle_123")
                    self.current_url = "https://new.example.com"
                    self.title = "New Window Title"
            return MockElement()
        raise Exception("Mock: 元素未找到")

    def switch_to(self):
        class SwitchTo:
            def window(self, handle):
                self.current_window_handle = handle
                print(f"Mock: 切换到窗口 {handle}")
        return SwitchTo()

# driver = MockWebDriver() # 实际使用时请注释此行，并使用真实的WebDriver
# 如果是真实的WebDriver，请取消注释下一行并替换为你的浏览器驱动
driver = webdriver.Chrome() # 例如使用Chrome
driver.get("https://www.example.com") # 替换为你的初始URL


print("--- 准备点击按钮 ---")
# 1. 获取当前窗口句柄
original_window_handle = driver.current_window_handle
print(f"原始窗口句柄: {original_window_handle}")

# 2. 找到并点击触发新窗口的元素
# 假设你的元素是 'Clique para visualizar'
try:
    print("正在点击 'Clique para visualizar' 按钮...")
    # 注意：这里的XPATH可能需要根据实际页面结构调整
    site_element = driver.find_element(By.XPATH, "//div[text()='Clique para visualizar']")
    site_element.click()
    print("按钮已点击。")
except Exception as e:
    print(f"点击按钮时发生错误: {e}")
    # 可以在此处添加错误处理或截图

# 3. 等待新窗口出现
# 使用WebDriverWait等待窗口句柄数量增加到2（原始窗口+新窗口）
try:
    WebDriverWait(driver, 10).until(EC.number_of_windows_to_be(2))
    print("新窗口已检测到。")
except TimeoutException:
    print("等待新窗口超时。")
    # 可以在此处添加错误处理
except Exception as e:
    print(f"等待新窗口时发生未知错误: {e}")


# 4. 获取所有窗口句柄
all_window_handles = driver.window_handles
print(f"所有窗口句柄: {all_window_handles}")

# 5. 遍历并切换到新窗口
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"已切换到新窗口，句柄: {new_window_handle}")
    # 现在，驱动的上下文已在新窗口，可以执行后续操作
    # 例如，打印新窗口的URL或标题
    print(f"新窗口URL: {driver.current_url}")
    print(f"新窗口标题: {driver.title}")
else:
    print("未找到新的窗口句柄。")

# 可以在新窗口中执行操作...
# 例如：driver.find_element(By.ID, "some_element_in_new_window").send_keys("test")

# 完成新窗口操作后，如果需要返回原始窗口，可以再次切换
# driver.switch_to.window(original_window_handle)
# print(f"已切换回原始窗口，句柄: {original_window_handle}")

# driver.quit() # 最后关闭浏览器

# ... (上面切换到新窗口的代码) ...

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

    # 在新窗口中等待特定元素加载
    try:
        print("正在等待新窗口中的特定元素...")
        # 假设新窗口中有一个ID为 'newWindowContent' 的元素
        # 你需要替换 'newWindowContent' 为新窗口中实际存在的元素ID、类名、XPath等
        new_window_element = WebDriverWait(driver, 20).until(
            EC.presence_of_element_located((By.ID, "newWindowContent"))
        )
        print("新窗口中的元素已加载:", new_window_element.text if new_window_element.text else "无文本内容")
        # 现在可以安全地与新窗口中的元素进行交互
        # new_window_element.click()
    except TimeoutException:
        print("在新窗口中等待元素超时。")
    except Exception as e:
        print(f"在新窗口中查找元素时发生错误: {e}")
else:
    print("未找到新的窗口句柄，无法在新窗口中执行操作。")

# 最后，不要忘记关闭浏览器
# driver.quit()