登录功能循环重试Bug求助：成功登录后仍重复执行

核心问题分析 #

你的代码出现无限重试的根本原因有3点：

1. 元素存在性判断逻辑错误：使用driver.find_element()判断登录按钮是否存在时，若元素不存在会直接抛出NoSuchElementException，根本走不到else分支。
2. 递归调用无终止条件：每次重试都递归调用login_attempt()，即使登录成功，之前的递归栈仍会继续执行，导致重复操作。
3. 异常捕获过于宽泛：except:捕获所有异常，包括登录成功后找不到登录按钮的情况，进而触发不必要的重试逻辑。

修复后的代码示例 #

from selenium.common.exceptions import NoSuchElementException
import time
from selenium.webdriver.common.by import By

def login_attempt():
    # 假设此处已完成用户名密码输入逻辑
    print("Username and Password were entered")
    
    # 替换为实际验证码获取逻辑
    result = "valid_captcha"  
    print(f"The stripped captcha text is: {result}")
    time.sleep(1)

    if result:
        try:
            # 输入验证码并提交
            captcha_input = driver.find_element(By.XPATH,"//div[@id='app']//div[contains(@class, 'el-input')]//input[@name='captcha']")
            captcha_input.send_keys(result)
            driver.find_element(By.XPATH,"//div[@id='app']//button[contains(@type,'button')]").click()
            print("Captcha has been entered")
            time.sleep(2)

            # 改用find_elements判断登录按钮是否存在（找不到返回空列表，布尔值为False）
            login_buttons = driver.find_elements(By.XPATH,"//div[@id='app']//form[@class= 'el-form login-form el-form--label-left']//button[contains(@type,'button')]")
            if login_buttons:
                print("Captcha was entered but was wrong, trying again...")
                driver.refresh()
                login_attempt()
            else:
                print("The login attempt worked!")
                # 登录成功后终止递归
                return
        except NoSuchElementException:
            # 捕获找不到元素的异常，说明已登录成功
            print("The login attempt worked!")
            return
        except Exception as e:
            # 处理其他意外异常
            print(f"Unexpected error: {str(e)}")
            time.sleep(2)
            driver.refresh()
            login_attempt()
    else:
        print("The captcha result was maybe blank.")
        time.sleep(2)
        login_buttons = driver.find_elements(By.XPATH,"//div[@id='app']//form[@class= 'el-form login-form el-form--label-left']//button[contains(@type,'button')]")
        if login_buttons:
            driver.refresh()
            time.sleep(2)
            login_attempt()
        else:
            print("The login attempt worked!")

关键修复点说明 #

• 用find_elements替代find_element：find_elements在找不到元素时返回空列表，可通过布尔判断直接确定元素是否存在，避免抛出异常中断逻辑。
• 缩小异常捕获范围：只捕获NoSuchElementException这类与元素查找相关的异常，避免误捕获其他导致重试的情况。
• 添加终止条件：登录成功后通过return终止递归，避免后续代码继续执行。
• 优化成功判断逻辑：建议改为判断登录后页面的标志性元素（比如用户头像、欢迎文案）是否存在，比判断登录按钮不存在更可靠，避免因页面加载延迟导致误判。

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