python - 如何编写一个函数来使用pickle从二进制文件中删除记录？

Question

我以列表的形式存储记录。

编写一个菜单驱动程序，在二进制文件中追加、显示和删除学生详细信息 [id, name, percent]。删除应根据学生证进行。

所以我写了这个：

#Code to append display and delete student details
import pickle
def append():
    f = open("student.bin","ab")
    x = int(input("Enter number of students to be appended: "))
    for i in range(x):
        y = eval(input("Enter details in the form [id, name, percentage]: "))
        pickle.dump(y,f)
    f.close()

def display():
    f = open("student.bin","rb")
    while True:
        try:
            x = pickle.load(f)
            print(x)
        except EOFError:
            break
    f.close()

def delete():
    f = open("student.bin","ab+")
    x = input("Enter id to be deleted: ")
    y = pickle.load(f)
    for i in y:
        if str(i[0]) == str(x):

最后，我将用户输入与匹配的 ID 匹配，但我不知道应该如何从二进制文件中完全删除列表。

Answer 1

这是一种方法：从文件中读取所有学生记录，删除我们要删除的记录，然后将整个列表写回文件中。

def delete():
    student_id = int(input("Enter id to be deleted: "))
    records = []
    with open("student.bin", "rb") as stream:
        while True:
            try:
                record = pickle.load(stream)
                if record[0] != student_id:
                    records.append(record)
            except EOFError:
                break
    
    # records now contains all students, except for the one we want to delete
    with open("student.bin", "wb") as stream:
        for record in records:
            pickle.dump(record, stream)

另一种方法是打开文件进行读/写并加载每条记录并将其写回。更难理解。