Python CSV登录程序异常：错误输入后无法正常登录

核心问题 #

你的代码里的csv.reader是一个一次性迭代器——第一次遍历完所有行之后，这个迭代器就会耗尽，不会再从头开始读取数据。所以当你第一次输入错误凭证、程序进入第二次循环时，for row in reader这部分代码根本不会执行，account_password会一直保持空值，哪怕你输入正确的用户名密码，也会被判定为错误。

修正后的完整代码 #

这里提供两种优化方案，你可以根据场景选择：

方案1：每次验证时重新读取文件（适合小体量账户数据）

import csv

while True:
    username = input("\nEnter username: ")
    password = input("Enter password: ")
    login_success = False
    
    # 每次验证都重新打开文件，保证能从头读取所有账户数据
    with open("accounts.csv") as csvfile:
        reader = csv.reader(csvfile)
        for row in reader:
            # 同时验证用户名和密码，逻辑更紧凑
            if row[0] == username and row[1] == password:
                login_success = True
                break
    
    if login_success:
        break
    else:
        print("Username/password is incorrect. Try again.")

print("\nSuccessfully logged in!")

方案2：提前加载所有账户到内存（效率更高，适合频繁验证场景）

import csv

# 提前把所有账户数据加载到字典中，键为用户名，值为密码
accounts_dict = {}
with open("accounts.csv") as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        accounts_dict[row[0]] = row[1]

while True:
    username = input("\nEnter username: ")
    password = input("Enter password: ")
    
    # 直接从字典查询验证，无需重复读文件
    if accounts_dict.get(username) == password:
        break
    else:
        print("Username/password is incorrect. Try again.")

print("\nSuccessfully logged in!")

修正原理与原代码差异 #

1. 解决迭代器耗尽问题：

   • 原代码：仅在循环外打开一次文件，reader迭代器第一次遍历后就为空，后续循环无法读取数据；
   • 方案1：每次验证都重新打开文件并创建新的reader，确保每次都能从头读取所有账户行；
   • 方案2：提前把所有账户数据加载到内存字典中，后续验证直接从字典查询，既避免了迭代器问题，又大幅提升验证效率。

2. 逻辑与变量优化：

   • 原代码拆分了“查用户名”和“验密码”的逻辑，且account_password变量在循环外初始化，容易残留无效值；
   • 修正后的代码将用户名和密码验证合并，用login_success标志（方案1）或字典查询（方案2）清晰判断登录状态，避免了无效变量残留的问题。

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