python - 如何使用 Python 删除存储在 Pickle 文件中的对象
问题描述
我正在开发一个将对象存储在 pickle 文件中的数据库程序。这些对象具有三个属性:名称、等级和平均。我已经能够将每个学生对象保存到一个泡菜文件中,我可以读回文件并显示它的内容。但是,我无法弄清楚如何使用索引号从文件中删除对象。我希望用户能够输入一个数字,然后删除该编号的对象(因此,如果我输入 2,第二个对象将被删除。)现在我的代码能够从泡菜列表中删除一个项目,但是我无法让它与对象一起工作。当我尝试删除学生时,我收到以下错误:TypeError: 'Student' object does not support item删除。有谁知道我如何从我的泡菜文件中删除一个对象?
我的代码如下:
import pickle
class Student():
def __init__(self,nam,grd,avg):
self.name = nam
self.grade = grd
self.average = avg
def get_details(self):
print(self.name, self.grade, self.average)
def create_item():
new_student = Student(input("Enter name: "),input("Enter grade: "), input("Enter average: "))
save_object(new_student, 'student_data.pkl')
def clear_database():
file = open('student_data.pkl', 'w')
file.close()
def save_object(obj, filename):
with open(filename, 'ab') as output:
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
def unpickle_database(filename):
with open(filename, 'rb') as f:
while True:
try:
yield pickle.load(f)
except EOFError:
break
def display_database():
print("\nName: Grade: Average: ")
for student in unpickle_database('student_data.pkl'):
property_list = [student.name, student.grade, student.average]
print(''.join([v.ljust(20,' ') for v in property_list]))
def delete_student(student_to_delete):
with open('student_data.pkl', 'rb') as f:
db = pickle.load(f)
try:
del db[student_to_delete]
except KeyError:
print("{user} doesn't exist in db".format(user=user_to_delete))
while True:
user_input = input("\nType \"Clear\" to clear the database. Type \"Add\" to add a student. Type \"Display\" to display the database contents. Type \"Quit\" to quit the program. Type \"Remove\" to remove a student.\n")
if user_input == "Quit":
break
if user_input == "Clear":
Student.clear_database()
print("\nThe database has been cleared.")
elif user_input == "Add":
Student.create_item()
display_database()
elif user_input == "Display":
display_database()
elif user_input == "Remove":
student_to_delete = input("Type the student number that you would like to delete: ")
delete_student(student_to_delete)
解决方案
您不能直接从 Pickle 文件中删除对象。我的意思是,你可以这样做,但它需要你解析它的结构来确定单个对象的数据在哪里开始和结束。这很乏味,而且不是很有价值。
但是,您可以将所有腌制学生作为列表读取:
students = list(unpickle_database(filename))
然后,从列表中删除其中一名学生del students[index]
,最后使用所述列表重新腌制所有剩余的学生:
for student in students:
save_object(student, filename)
推荐阅读
- java - 广播意图回调:result=CANCELLED forIntent { act=com.google.android.c2dm.intent.RECEIVE
- python - lgbm模型开发过程中的local_test
- xslt - 如何在xslt中的特定元素之后复制元素
- sql - MariaDB,如何使用插入复制现有记录?
- android - 适用于 Android 的离子电容器插件未构建
- python - conda create from environment file vs. conda clone
- javascript - 解析多部分结果
- eclipse - 如何解决 Eclipse 的“Auth error to push the code --Eclipse to Github”的问题
- mysql - mysql 中导出和导入生成的存储/虚拟列的问题
- twitter-bootstrap - bootstrap 4中类崩溃的含义是什么?