Python Selenium循环处理互斥菜单元素异常问题求助

原始代码 #

while True:
    try:
        btnOptions = WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, "/html/...."))).click()
        firstElement = driver.find_elements(By.XPATH, "//*[text()='Click']")
        if not firstElement:
            tnTrash = WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.XPATH, "//*[text()='Move']"))).click()
            WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, "/html/body/..."))).click()
        else:
            firstElement.click()
    except NoSuchElementException:
        break

预期逻辑 #

• 步骤1：找到BtnOptions按钮并点击打开菜单，菜单只会出现element1或element2两者之一，不会同时存在：
  • element1定位：WebDriverWait(driver, 10).until(driver.find_elements(By.XPATH, "//*[text()='element1']"))
  • element2定位：WebDriverWait(driver, 10).until(driver.find_elements(By.XPATH, "//*[text()='element2']"))
• 步骤2：找到element1则点击并重启循环；找到element2也点击并重启循环（删除帖子时菜单显示element1，处理头像时显示element2）

当前问题 #

程序仅在找到element1时正常运行，遇到element2会因找不到element1报错。

尝试过的无效代码 #

while True:
    try:
        btnOptions = WebDriverWait(driver, 20).until(EC.element_to_be_clickable((By.XPATH, "/html/body/div..."))).click()
        WebDriverWait(driver, 20).until(driver.find_element(By.XPATH("//*[text()='move to trash']"))).click() or WebDriverWait(driver, 20).until(driver.find_element(By.XPATH("//*[text()='Do not show in profile']"))).click()
    except NoSuchElementException:
        break

解决方案 #

写法1：使用EC.any_of等待任意目标元素出现 #

利用Selenium的any_of预期条件，等待两个元素中的任意一个可点击，避免单个元素查找失败直接抛异常：

from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import TimeoutException

while True:
    try:
        # 点击选项按钮打开菜单
        WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, "/html/...."))).click()
        
        # 等待element1或element2中的一个可点击，超时10秒
        target_element = WebDriverWait(driver, 10).until(
            EC.any_of(
                EC.element_to_be_clickable((By.XPATH, "//*[text()='element1']")),
                EC.element_to_be_clickable((By.XPATH, "//*[text()='element2']"))
            )
        )
        target_element.click()
        
        # 若element1点击后需要额外确认操作，可添加判断
        if "element1" in target_element.text:
            WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, "/html/body/..."))).click()
            
    except (NoSuchElementException, TimeoutException):
        # 找不到按钮或菜单元素时退出循环
        break

写法2：通过元素列表判断存在性 #

用find_elements获取元素列表（找不到时返回空列表，不抛异常），再判断哪个元素存在：

while True:
    try:
        WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, "/html/...."))).click()
        
        # 获取两个元素的列表，空列表表示元素不存在
        element1 = driver.find_elements(By.XPATH, "//*[text()='element1']")
        element2 = driver.find_elements(By.XPATH, "//*[text()='element2']")
        
        if element1:
            element1[0].click()
            # 处理element1的后续确认操作
            WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, "/html/body/..."))).click()
        elif element2:
            element2[0].click()
        else:
            # 菜单无预期元素，退出循环
            break
            
    except NoSuchElementException:
        break

原代码报错原因 #

1. 原始写法中，except仅捕获NoSuchElementException，但WebDriverWait超时会抛出TimeoutException，未被捕获导致崩溃；
2. 尝试的代码中，By.XPATH("//*[text()='xxx']")写法错误，正确格式应为(By.XPATH, "xxx")，且or逻辑无效——第一个WebDriverWait失败会直接抛异常，无法执行后续代码。

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