如何在使用Selenium的Python上点击iframe按钮？

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")

# 等待iframe加载完成
wait = WebDriverWait(driver, 10)
iframe = wait.until(EC.presence_of_element_located((By.TAG_NAME, "iframe")))

# 切换到iframe上下文
driver.switch_to.frame(iframe)

# 定位并点击按钮
button = driver.find_element(By.XPATH, "//button[@id='button-id']")
button.click()

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

# 关闭浏览器
driver.quit()