Python pickle读取全量数据异常：viewall函数仅返回列表首个条目问题排查

嘿，我一眼就瞅出问题所在啦！你的viewall函数只返回第一条记录，问题其实出在add函数的实现上，咱们一步步捋清楚：

问题根源 #

• 在add函数中，你每次新增账户时，都会创建一个全新的列表Mrecord，把单条账户记录塞进这个列表，然后用pickle.dump(Mrecord, file)把这个仅含一条记录的列表追加到文件里。
• 而viewall函数只调用了一次pickle.load(file)——这个方法只会读取文件中第一个被序列化的对象（也就是你第一次添加的那个单记录列表），后面追加的其他账户对象根本没被读取到，自然就只显示第一条记录了。

修正方案 #

我们需要调整add函数的逻辑，让文件中始终存储一个包含所有账户的完整列表，而不是多个独立的小列表；同时优化viewall函数，让输出更清晰。

修正后的add函数 #

def add(x="BankDatabase.txt"):
    # 先读取已有的账户记录（如果文件存在且非空）
    Mrecord = []
    if os.path.exists(x) and os.path.getsize(x) > 0:
        with open(x, "rb") as file:
            Mrecord = pickle.load(file)
    
    # 收集账户信息
    Account_number = int(input("Enter account number of the account holder - "))
    if countDigit(Account_number) > 20:
        sys.exit("Maximum 20 digits allowed!")
    Name = input("Enter name of the account holder - ")
    Phone_number = int(input("Enter phone number of the account holder - "))
    if countDigit(Phone_number) != 10:
        sys.exit("Invalid phone number!")
    Address = input("Enter address of the account holder - ")
    Date_Of_Birth = input("Enter date of birth of account holder - ")
    if len(Date_Of_Birth) != 10:
        sys.exit("Invalid date of birth!")
    Opening_Balance = int(input("Enter opening balance of the account - "))
    Balance = Opening_Balance
    Trecord = [Account_number, Name, Phone_number, Address, Date_Of_Birth, Opening_Balance, Balance]
    
    # 将新记录添加到总列表中
    Mrecord.append(Trecord)
    print(Trecord)
    
    # 将完整的账户列表写入文件（覆盖原文件）
    with open(x, "wb") as file:
        pickle.dump(Mrecord, file)
    print("Account added successfully!")

修正后的viewall函数 #

def viewall(x="BankDatabase.txt"):
    # 先检查文件是否存在或为空
    if not os.path.exists(x) or os.path.getsize(x) == 0:
        print("No accounts found in the database.")
        return
    
    with open(x, "rb") as file:
        all_records = pickle.load(file)
        # 遍历所有记录，友好输出每个账户的详细信息
        for idx, record in enumerate(all_records, 1):
            print(f"--- Account {idx} ---")
            print(f"Account Number: {record[0]}")
            print(f"Name: {record[1]}")
            print(f"Phone Number: {record[2]}")
            print(f"Address: {record[3]}")
            print(f"Date of Birth: {record[4]}")
            print(f"Opening Balance: {record[5]}")
            print(f"Current Balance: {record[6]}")
            print("------------------------")

为什么这样改？ #

• 现在add函数会先读取文件中已有的所有账户记录，把新记录追加到总列表后，再将完整列表写入文件，保证文件里始终是一个包含全部账户的集合。
• viewall函数读取一次就能拿到所有记录，还能遍历输出每个账户的详细信息，比直接打印列表更易读。

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