Python CSV处理问题：添加新用户时输出文件未保留已有用户

问题根源 #

原脚本的核心问题在于首次生成输出文件时未写入CSV表头，导致后续用csv.DictReader读取已有用户时无法正确解析列，进而出现数据识别错误；另外，单纯追加模式若未处理表头，会导致文件结构混乱，看起来像是原有用户丢失。若要确保输出文件始终包含完整的用户列表（原有+新增），更稳妥的方式是先读取所有已有用户数据，再重新写入完整内容，而非单纯追加。

推荐解决方案（重新生成完整文件） #

此方案会先读取已有用户数据，再重新生成包含所有用户（原有+新增）的输出文件，彻底避免数据丢失或结构混乱：

import csv
import secrets
import subprocess
from pathlib import Path

# 设置数据目录路径
data_dir = Path("/home/shayan/Desktop/Python Script/Script_1/data")
output_file = data_dir / "users_out.csv"

# 读取已有用户数据和用户名集合
existing_users = []
existing_usernames = set()

try:
    with open(output_file, "r") as f:
        reader = csv.DictReader(f)
        existing_users = list(reader)
        existing_usernames = {user["username"] for user in existing_users}
except FileNotFoundError:
    pass  # 文件不存在时跳过

# 读取输入文件并处理新用户
with open(data_dir / "users_in.csv", "r") as file_input:
    reader = csv.DictReader(file_input)
    fieldnames = ["username", "password", "real_name"]

    # 重新写入完整的输出文件（原有用户+新增用户）
    with open(output_file, "w", newline="") as file_output:
        writer = csv.DictWriter(file_output, fieldnames=fieldnames)
        writer.writeheader()

        # 先写入已有用户
        writer.writerows(existing_users)

        # 处理并写入新用户
        for user in reader:
            if "username" in user and user["username"] not in existing_usernames:
                # 生成16位随机密码
                user["password"] = secrets.token_hex(8)

                # 执行useradd命令创建用户
                useradd_cmd = [
                    "/sbin/useradd",
                    "-c", user["real_name"],
                    "-m",
                    "-G", "users",
                    "-p", user["password"],
                    user["username"]
                ]
                try:
                    subprocess.run(useradd_cmd, check=True)
                except subprocess.CalledProcessError:
                    print(f"用户 '{user['username']}' 已存在，跳过...")
                    continue

                # 写入新用户数据
                writer.writerow(user)
                existing_usernames.add(user["username"])  # 更新集合，避免重复

print("用户处理完成。")

修改说明 #

• 读取已有用户时，同时保存完整用户数据和用户名集合，确保后续能完整写入原有数据。
• 使用"w"模式重新生成输出文件，先写入表头，再依次写入已有用户和新增用户，保证文件结构清晰、数据完整。
• 添加newline=""参数，避免CSV文件出现多余空行。
• 处理新用户时实时更新用户名集合，避免同一输入文件内重复添加同一用户。

备选方案（追加模式） #

若不需要重新生成完整文件，仅需在已有文件后追加新用户，可使用此方案（需处理表头问题）：

import csv
import secrets
import subprocess
from pathlib import Path

# 设置数据目录路径
data_dir = Path("/home/shayan/Desktop/Python Script/Script_1/data")
output_file = data_dir / "users_out.csv"

# 读取已有用户用户名集合
existing_usernames = set()
file_exists = output_file.exists()

if file_exists:
    with open(output_file, "r") as f:
        reader = csv.DictReader(f)
        existing_usernames = {user["username"] for user in reader}

# 读取输入文件并追加新用户
with open(data_dir / "users_in.csv", "r") as file_input, \
     open(output_file, "a", newline="") as file_output:
    fieldnames = ["username", "password", "real_name"]
    writer = csv.DictWriter(file_output, fieldnames=fieldnames)

    # 首次创建文件时写入表头
    if not file_exists:
        writer.writeheader()

    # 处理新用户
    reader = csv.DictReader(file_input)
    for user in reader:
        if "username" in user and user["username"] not in existing_usernames:
            # 生成16位随机密码
            user["password"] = secrets.token_hex(8)

            # 执行useradd命令创建用户
            useradd_cmd = [
                "/sbin/useradd",
                "-c", user["real_name"],
                "-m",
                "-G", "users",
                "-p", user["password"],
                user["username"]
            ]
            try:
                subprocess.run(useradd_cmd, check=True)
            except subprocess.CalledProcessError:
                print(f"用户 '{user['username']}' 已存在，跳过...")
                continue

            # 追加写入新用户
            writer.writerow(user)
            existing_usernames.add(user["username"])

print("用户处理完成。")

修改说明 #

• 先判断输出文件是否存在，不存在则在追加时写入表头，避免表头缺失导致的解析错误。
• 追加新用户时，通过用户名集合过滤重复数据，确保不会重复写入已有用户。

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