如何使用Selenium自动点击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("目标网页的URL")
driver.switch_to.frame("iframe的ID或名称")

# 定位链接元素
link_elements = driver.find_elements(By.TAG_NAME, "a")

# 循环点击每个链接
for link_element in link_elements:
    link_element.click()
    
    # 等待页面加载完成
    WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.TAG_NAME, "body")))
    
    # 切换回iframe
    driver.switch_to.default_content()
    driver.switch_to.frame("iframe的ID或名称")

driver.quit()