Python Selenium：如何同时检查多元素以判断登录结果

问题描述 #

我用Selenium编写网站登录代码，需要通过页面元素判断是否登录成功。目前的实现是先检查2FA登录场景的元素，超时后再检查常规登录的元素，整体耗时太长。原代码如下：

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

def handle_response(self):
    try:
        WebDriverWait(self.driver, 3).until(EC.presence_of_element_located((By.XPATH, "//button[@class='s-btn s-btn-primary s-btn--block enterEmailComponent-button']"))) # 2FA登录
        return True
    except TimeoutException:
        try:
            WebDriverWait(self.driver, 3).until(EC.presence_of_element_located((By.XPATH, "//input[@id='search-autocomplete-input']"))) # 常规登录
            return True
        except TimeoutException:
                return False

我想改成同时检查这两个元素，只要其中一个出现就返回成功，避免串行等待的耗时问题。

解决方案 #

可以用Selenium提供的EC.any_of()条件，它允许你同时等待多个预期条件中的任意一个满足，整个等待过程最多只会消耗你设置的超时时间，而非两次等待的总和。

基础改进版 #

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

def handle_response(self):
    try:
        # 同时监听两个元素，任意一个出现就结束等待
        WebDriverWait(self.driver, 3).until(
            EC.any_of(
                # 2FA登录场景元素
                EC.presence_of_element_located((By.XPATH, "//button[@class='s-btn s-btn-primary s-btn--block enterEmailComponent-button']")),
                # 常规登录成功后的元素
                EC.presence_of_element_located((By.XPATH, "//input[@id='search-autocomplete-input']"))
            )
        )
        return True
    except TimeoutException:
        # 两个元素都未出现，判定登录失败
        return False

区分场景版（可选） #

如果需要后续针对不同场景做额外处理，可以捕获any_of的返回结果，判断具体是哪个元素被找到：

def handle_response(self):
    try:
        result = WebDriverWait(self.driver, 3).until(
            EC.any_of(
                EC.presence_of_element_located((By.XPATH, "//button[@class='s-btn s-btn-primary s-btn--block enterEmailComponent-button']")),
                EC.presence_of_element_located((By.XPATH, "//input[@id='search-autocomplete-input']"))
            )
        )
        # 根据元素属性区分场景
        if "enterEmailComponent-button" in result.get_attribute("class"):
            # 执行2FA相关逻辑
            print("进入二次验证流程")
        else:
            # 常规登录成功后的操作
            print("登录成功，进入主页面")
        return True
    except TimeoutException:
        return False

关键说明 #

• EC.any_of()会持续监听页面，只要传入的任意一个条件满足（元素出现），就会立即结束等待，无需等到超时。
• 改进后总等待时间最多为设置的3秒，相比原来最多6秒的等待时间，能大幅缩短判断耗时。

内容的提问来源于stack exchange，提问作者Dang Max 2.0