使用Python Selenium遍历多级菜单按索引截图遇循环问题求助

需要用Python的Selenium工具遍历嵌套菜单并按目录索引截图，菜单位于xpath为/html/body/div[3]/div/iframe的iframe中，共3级结构，返回按钮仅在二级菜单生效。菜单结构如下：

1.1 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[1]"
1.2 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[2]"
1.2.1 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[2]/ul/li"
back_button xpath "/html/body/div[3]/ul/li[3]"
1.3 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[3]"
1.4 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[4]"
1.5 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[5]"
1.5.1 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[5]/ul/li[1]"
1.5.2 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[5]/ul/li[2]"
1.5.2.1 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[5]/ul/li[2]/ul/li[1]"
1.5.2.2 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[5]/ul/li[2]/ul/li[2]"
1.5.2.3 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[5]/ul/li[2]/ul/li[3]"
1.5.3 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[5]/ul/li[3]"
...
1.127 li Element xpath "/html/body/div[3]/ul/li[2]/ul/li[127]"

编写的代码如下：

iframe = driver.find_element(By.XPATH, "/html/body/div[3]/div/iframe")
driver.switch_to.frame(iframe)

xpath_prefix = "/html/body/div[3]/ul/li[2]/ul/li"
back_button_xpath = "/html/body/div[3]/ul/li[3]"

def take_screenshot(driver, index):
    driver.save_screenshot(f"{index}.png")

def click_next_li(driver, index):
    try:
        next_li = driver.find_element(By.XPATH, f"{xpath_prefix}[{index}]")
        next_li.click()
        time.sleep(5)
        take_screenshot(driver, index)
        return True
    except (NoSuchElementException, ElementNotInteractableException):
        return False

def go_back(driver):
    back_button = driver.find_element(By.XPATH, back_button_xpath)
    back_button.click()
    time.sleep(5)

index = 1
while click_next_li(driver, index):
    index += 1

go_back(driver)

index = 1
while click_next_li(driver, index):
    index += 1

# Switch back to the main frame
driver.switch_to.default_content()

遇到的问题：程序进入1.2对应的子菜单后返回，会重新回到1.1，无法继续遍历1.3及后续项。询问问题原因及是否需要追踪已访问元素。

问题根源 #

1. 遍历进度丢失：第一次while循环遍历到1.2进入子菜单，返回后第二次循环直接从index=1重启，导致重复处理1.1，而非从1.3开始继续遍历一级菜单。
2. 未处理嵌套子菜单：当前代码仅处理一级菜单进入二级的逻辑，完全忽略了二级菜单下的三级子项（比如1.2.1、1.5.2.*等）。
3. 硬编码sleep不稳定：固定等待5秒可能导致页面未加载完成就执行后续操作，或浪费不必要的等待时间。

解决方案 #

1. 追踪一级菜单遍历进度 #

返回二级菜单后，从上次中断的索引+1位置继续遍历，而非从头开始。

2. 递归遍历嵌套菜单 #

采用递归方式处理所有层级的子菜单，自动识别并遍历二级、三级子项，同时按目录索引命名截图（比如1.2.png、1.2.1.png）。

3. 替换为显式等待 #

使用Selenium的WebDriverWait等待元素可点击，提升代码稳定性。

修正后的代码 #

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 NoSuchElementException, ElementNotInteractableException

def take_screenshot(driver, index_str):
    """按目录索引保存截图"""
    driver.save_screenshot(f"{index_str}.png")

def get_child_menu_items(parent_element):
    """获取当前元素下的所有子菜单li元素"""
    try:
        return parent_element.find_elements(By.XPATH, "./ul/li")
    except NoSuchElementException:
        return []

def traverse_menu(driver, parent_xpath, current_index_str):
    """递归遍历菜单，处理所有层级"""
    # 等待父菜单元素加载完成
    parent_element = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.XPATH, parent_xpath))
    )
    # 获取当前层级的所有菜单项
    menu_items = parent_element.find_elements(By.XPATH, "./li")
    
    for idx, item in enumerate(menu_items, start=1):
        # 跳过返回按钮（一级菜单中的返回按钮无需处理）
        if parent_xpath == "/html/body/div[3]/ul/li[2]/ul" and idx == 3:
            continue
        
        # 构建当前项的索引字符串（比如1.2、1.2.1）
        current_item_index = f"{current_index_str}.{idx}" if current_index_str else str(idx)
        
        try:
            # 等待元素可点击后点击
            WebDriverWait(driver, 10).until(
                EC.element_to_be_clickable(item)
            ).click()
            # 等待页面切换完成
            WebDriverWait(driver, 10).until(
                EC.staleness_of(item)
            )
            take_screenshot(driver, current_item_index)
            
            # 检查当前项是否有子菜单
            child_items = get_child_menu_items(item)
            if child_items:
                # 递归遍历子菜单
                traverse_menu(driver, f"{parent_xpath}/li[{idx}]/ul", current_item_index)
                # 返回上一级菜单（仅二级菜单生效）
                if len(current_item_index.split(".")) == 2:
                    back_button = WebDriverWait(driver, 10).until(
                        EC.element_to_be_clickable((By.XPATH, "/html/body/div[3]/ul/li[3]"))
                    )
                    back_button.click()
                    WebDriverWait(driver, 10).until(
                        EC.presence_of_element_located((By.XPATH, parent_xpath))
                    )
        except (ElementNotInteractableException, NoSuchElementException):
            continue

# 主逻辑
driver = webdriver.Chrome()  # 根据实际情况选择浏览器驱动
driver.get("你的目标页面URL")

# 切换到iframe
iframe = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.XPATH, "/html/body/div[3]/div/iframe"))
)
driver.switch_to.frame(iframe)

# 开始遍历一级菜单
traverse_menu(driver, "/html/body/div[3]/ul/li[2]/ul", "")

# 切回主frame
driver.switch_to.default_content()
driver.quit()

代码说明 #

• 递归遍历：自动处理所有层级的菜单，无需手动编写多层循环。
• 显式等待：所有元素操作前都等待可点击/存在，避免页面未加载完成导致的错误。
• 索引命名：按菜单层级生成索引字符串（如1.2.1），截图文件名与目录索引对应。
• 返回逻辑：仅在二级菜单返回上一级，符合题目中返回按钮仅在二级生效的要求。

内容的提问来源于stack exchange，提问作者user21037596