寻找一种在python自动测试中的for循环期间检查意外窗口的方法

from selenium import webdriver

driver = webdriver.Chrome()

for i in range(10):
    # 执行测试步骤
    driver.find_element_by_id("button_id").click()

for i in range(10):
    # 执行测试步骤
    driver.find_element_by_id("button_id").click()

    # 检查是否有意外窗口弹出
    if len(driver.window_handles) > 1:
        # 处理意外窗口
        driver.switch_to.window(driver.window_handles[1])
        # 进行相关操作
        driver.close()
        # 切换回主窗口
        driver.switch_to.window(driver.window_handles[0])