如何优化Selenium登录验证逻辑：替换固定等待为动态元素等待

原登录验证逻辑依赖固定5秒等待，之后仅检查错误提示元素判断登录结果，效率较低。现在需要修改为同时监听登录错误提示和成功授权窗口元素，哪个元素先出现就返回对应结果：找到错误元素返回False，找到成功元素返回True。

修改后的代码 #

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

# 假设wait是已初始化的WebDriverWait实例，示例：wait = WebDriverWait(driver, 10)
try:
    # 同时等待错误或成功元素出现，超时时间可根据实际场景调整
    element = wait.until(EC.any_of(
        EC.presence_of_element_located(Locators.MSG_ERROR_LOGIN),
        EC.presence_of_element_located(Locators.MSG_SUCCESS_LOGIN)
    ))
    # 判断触发的是哪种结果
    if element == driver.find_element(*Locators.MSG_ERROR_LOGIN):
        print('Error')
        return False
    else:
        print('SUCCESS')
        return True
except TimeoutException:
    # 处理超时场景（两种元素都未出现）
    print('Timeout: Neither error nor success element appeared')
    return False

逻辑说明 #

• 用EC.any_of实现多条件等待：只要传入的任一元素出现，until就会立即返回对应元素，彻底消除固定等待时间
• 通过匹配返回元素的定位器，即时判断登录结果并输出、返回对应布尔值
• 增加超时异常捕获，避免页面异常导致程序无限等待

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