Python CSV模块注册函数问题排查及正确实现需求

原代码的问题出在哪？ #

• 用户名检查逻辑完全错误：原代码遍历CSV每一行时，只要当前行用户名和输入不一样，就直接执行写入操作。比如输入已存在的user，轮到user2那行时，会误判为“用户名未被占用”并写入数据，导致同时弹出“用户名已占用”和“注册成功”的矛盾提示。
• 无限循环无法终止：外层while True没有设置退出条件，就算注册成功，程序还会一直要求用户输入。
• 文件指针位置异常：用csv.reader读完文件后，文件指针会移动到末尾，后续循环再读取时会获取不到任何内容，等于跳过了用户名检查步骤。
• 异常处理过于敷衍：except: pass会吞掉所有异常（比如文件权限错误），导致无法排查潜在问题。
• 写入方法使用错误：writerows是用于多行写入的方法，这里只需要写入一行数据，用writerow才符合需求。

修复好的注册函数 #

import csv

def register():
    # 确保文件存在，不存在则创建空文件
    try:
        with open('users.csv', 'x', newline='') as f:
            pass
    except FileExistsError:
        # 文件已存在，无需重复创建
        pass

    while True:
        username = input('请输入用户名: ')
        password = input('请输入密码（必须仅包含数字）: ')

        # 先校验密码格式
        if not password.isdigit():
            print('密码必须仅包含数字，请重新输入')
            continue

        # 读取所有已存在的用户名，存入集合便于快速查询
        existing_users = set()
        try:
            with open('users.csv', 'r', newline='') as f:
                reader = csv.reader(f)
                for row in reader:
                    if row:  # 跳过文件中的空行
                        existing_users.add(row[0])
        except Exception as e:
            print(f"读取用户文件失败: {str(e)}")
            continue

        # 检查用户名是否已被占用
        if username in existing_users:
            print('该用户名已被占用，请重新输入')
            continue

        # 注册成功，将用户数据追加写入文件
        try:
            with open('users.csv', 'a', newline='') as f:
                writer = csv.writer(f)
                writer.writerow([username, password])
            print('注册成功！')
            break  # 注册成功后退出循环
        except Exception as e:
            print(f"写入用户文件失败: {str(e)}")
            continue

修复说明 #

1. 优化文件处理逻辑：
   • 仅捕获FileExistsError处理文件已存在的情况，其他异常会给出明确提示，避免隐藏问题。
   • 读取文件用r模式，写入用a（追加）模式，彻底避开文件指针位置混乱的问题。
2. 修正用户名检查逻辑：
   • 先将所有已存在的用户名读取到集合中，通过集合的快速查询特性判断用户名是否被占用，解决了原代码逐行判断的逻辑错误。
3. 调整校验顺序：先校验密码格式，不符合要求直接让用户重输，减少不必要的文件读取操作。
4. 添加循环终止条件：注册成功后通过break跳出while True循环，结束函数执行。
5. 完善异常提示：对文件读写操作的异常进行捕获并给出具体信息，便于排查问题。
6. 修正写入方法：用writerow替代writerows，匹配单行数据写入的需求。

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