如何使用python-selenium在弹出窗口中点击按钮

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

driver = webdriver.Chrome()

driver.get("https://example.com")
button = driver.find_element(By.ID, "button-id")

button.click()

# 获取当前窗口的句柄
current_window = driver.current_window_handle

# 获取所有窗口的句柄
all_windows = driver.window_handles

# 切换到新窗口
for window in all_windows:
    if window != current_window:
        driver.switch_to.window(window)
        break

# 切换到iframe
iframe = driver.find_element(By.ID, "iframe-id")
driver.switch_to.frame(iframe)

# 切换回主窗口
driver.switch_to.default_content()

# 切换回默认上下文
driver.switch_to.default_content()

driver.quit()