[已解决]求助通讯录小程序数据保存的问题

lxx213

import pickle

print('|--- 欢迎进入通讯录程序 ---|')
print('|--- 1：查询联系人资料  ---|')
print('|--- 2：插入新的联系人  ---|')
print('|--- 3：删除已有联系人  ---|')
print('|--- 4：退出通讯录程序  ---|')

try:
     
    f = open('通讯录数据库.pki','rb')
        
except OSError:
   
    contacts = dict()

else:
    contacts = pickle.load(f)
   
    print(contacts)
    f.close()



print(contacts)   

while 1:

  
    instr = int(input('\n请输入相关的指令代码：'))


    if instr == 1:
        name = input('请输入联系人姓名：')
        if name in contacts:
            print(name + ' : ' + contacts[name])
        else:
            print('您输入的姓名不再通讯录中！')

    if instr == 2:
        name = input('请输入联系人姓名：')
        if name in contacts:
            print('您输入的姓名在通讯录中已存在 -->> ', end='')
            print(name + ' : ' + contacts[name])
            if input('是否修改用户资料（YES/NO）：') == 'YES':
                contacts[name] = input('请输入用户联系电话：')
        else:
            contacts[name] = input('请输入用户联系电话：')

    if instr == 3:
        name = input('请输入联系人姓名：')
        if name in contacts:
            del(contacts[name])         # 也可以使用dict.pop()
        else:
            print('您输入的联系人不存在。')
            
    if instr == 4:
        break
   

print('|--- 感谢使用通讯录程序 ---|')

f =open('通讯录数据库.pki','ab')

pickle.dump(contacts,f)
f.close()
print(contacts)   
这个通讯录程序 我想实现每次新添加的用户信息在退出后 还可以保存 下次进入程序后还有原有信息 可以从第一次运行程序 添加的用户信息保留了，但是第二次运行再添加的信息，第三次运行时就没有了，还是只有第一次运行添加进去的
三个print(contacts)   是自己测试数据添加是否成功的

最佳答案

月排行榜 / 总排行榜

suchocolate

2021-1-13 11:22:20

pickle写入的是对象，执行一次load，只是加载1个对象，想加载所有需要再次使用load：

1. import pickle
   
2. 
   
3. 
   
4. with open('test2.pkl', 'wb') as f:
   
5.     pickle.dump({1: 'a'}, f)
   
6. 
   
7. with open('test2.pkl', 'ab') as f:
   
8.     pickle.dump({2: 'b'}, f)
   
9.     pickle.dump({3: 'c'}, f)
   
10. 
    
11. 
    
12. with open('test2.pkl', 'rb') as f:
    
13.     print('只加载一个对象：', pickle.load(f))
    
14. 
    
15. 
    
16. with open('test2.pkl', 'rb') as f:
    
17.     print('加载所有对象：')
    
18.     while True:
    
19.         try:
    
20.             print(pickle.load(f))
    
21.         except EOFError:
    
22.             exit(0)

复制代码

如果不想考虑追加问题，那么就在保存时wb重写，这样读取一次就够了。

1. import pickle
   
2. import os
   
3. 
   
4. 
   
5. def show_menu():
   
6.     print('|--- 欢迎进入通讯录程序 ---|')
   
7.     print('|--- 1：查询联系人资料  ---|')
   
8.     print('|--- 2：插入新的联系人  ---|')
   
9.     print('|--- 3：删除已有联系人  ---|')
   
10.     print('|--- 4：退出通讯录程序  ---|')
    
11. 
    
12. 
    
13. def load_file():
    
14.     """
    
15.     定义加载文件函数
    
16.     :return: 联系人字典
    
17.     """
    
18.     if not os.path.exists('contact_db.pkl'):
    
19.         save_file(dict())
    
20.     with open('contact_db.pkl', 'rb') as f:
    
21.         result = pickle.load(f)
    
22.         return result
    
23. 
    
24. 
    
25. def save_file(f_c):
    
26.     """
    
27.     定义保存文件函数
    
28.     :param f_c:
    
29.     :return: none
    
30.     """
    
31.     with open('contact_db.pkl', 'wb') as f:
    
32.         pickle.dump(f_c, f)
    
33. 
    
34. 
    
35. def main():
    
36.     """
    
37.     定义主函数
    
38.     :return: none
    
39.     """
    
40.     show_menu()
    
41.     c = load_file()
    
42.     while True:
    
43.         cid = int(input('请输入相关的指令代码: '))
    
44.         if cid == 1:
    
45.             name = input('请输入联系人姓名: ')
    
46.             if name in c:
    
47.                 print(f'{name}: {c[name]}')
    
48.             else:
    
49.                 print('您输入的姓名不再通讯录中!')
    
50.         elif cid == 2:
    
51.             name = input('请输入联系人姓名: ')
    
52.             if name in c:
    
53.                 print('您输入的姓名在通讯录中已存在 -->> ', end='')
    
54.                 print(f'{name}: {c[name]}')
    
55.                 if input(r'是否修改用户资料(Y/n):') in ['Y', 'y']:
    
56.                     c[name] = input('请输入用户联系电话：')
    
57.             else:
    
58.                 c[name] = input('请输入用户联系电话：')
    
59.         elif cid == 3:
    
60.             name = input('请输入联系人姓名：')
    
61.             if name in c:
    
62.                 del (c[name])
    
63.             else:
    
64.                 print('您输入的联系人不存在。')
    
65.         elif cid == 4:
    
66.             break
    
67.         else:
    
68.             print('请输入正确的id。')
    
69.     print('|--- 感谢使用通讯录程序 ---|')
    
70.     save_file(c)
    
71. 
    
72. 
    
73. if __name__ == '__main__':
    
74.     main()

复制代码

跳转到最佳答案楼层

suchocolate

pickle写入的是对象，执行一次load，只是加载1个对象，想加载所有需要再次使用load：

1. import pickle
   
2. 
   
3. 
   
4. with open('test2.pkl', 'wb') as f:
   
5.     pickle.dump({1: 'a'}, f)
   
6. 
   
7. with open('test2.pkl', 'ab') as f:
   
8.     pickle.dump({2: 'b'}, f)
   
9.     pickle.dump({3: 'c'}, f)
   
10. 
    
11. 
    
12. with open('test2.pkl', 'rb') as f:
    
13.     print('只加载一个对象：', pickle.load(f))
    
14. 
    
15. 
    
16. with open('test2.pkl', 'rb') as f:
    
17.     print('加载所有对象：')
    
18.     while True:
    
19.         try:
    
20.             print(pickle.load(f))
    
21.         except EOFError:
    
22.             exit(0)

复制代码

如果不想考虑追加问题，那么就在保存时wb重写，这样读取一次就够了。

1. import pickle
   
2. import os
   
3. 
   
4. 
   
5. def show_menu():
   
6.     print('|--- 欢迎进入通讯录程序 ---|')
   
7.     print('|--- 1：查询联系人资料  ---|')
   
8.     print('|--- 2：插入新的联系人  ---|')
   
9.     print('|--- 3：删除已有联系人  ---|')
   
10.     print('|--- 4：退出通讯录程序  ---|')
    
11. 
    
12. 
    
13. def load_file():
    
14.     """
    
15.     定义加载文件函数
    
16.     :return: 联系人字典
    
17.     """
    
18.     if not os.path.exists('contact_db.pkl'):
    
19.         save_file(dict())
    
20.     with open('contact_db.pkl', 'rb') as f:
    
21.         result = pickle.load(f)
    
22.         return result
    
23. 
    
24. 
    
25. def save_file(f_c):
    
26.     """
    
27.     定义保存文件函数
    
28.     :param f_c:
    
29.     :return: none
    
30.     """
    
31.     with open('contact_db.pkl', 'wb') as f:
    
32.         pickle.dump(f_c, f)
    
33. 
    
34. 
    
35. def main():
    
36.     """
    
37.     定义主函数
    
38.     :return: none
    
39.     """
    
40.     show_menu()
    
41.     c = load_file()
    
42.     while True:
    
43.         cid = int(input('请输入相关的指令代码: '))
    
44.         if cid == 1:
    
45.             name = input('请输入联系人姓名: ')
    
46.             if name in c:
    
47.                 print(f'{name}: {c[name]}')
    
48.             else:
    
49.                 print('您输入的姓名不再通讯录中!')
    
50.         elif cid == 2:
    
51.             name = input('请输入联系人姓名: ')
    
52.             if name in c:
    
53.                 print('您输入的姓名在通讯录中已存在 -->> ', end='')
    
54.                 print(f'{name}: {c[name]}')
    
55.                 if input(r'是否修改用户资料(Y/n):') in ['Y', 'y']:
    
56.                     c[name] = input('请输入用户联系电话：')
    
57.             else:
    
58.                 c[name] = input('请输入用户联系电话：')
    
59.         elif cid == 3:
    
60.             name = input('请输入联系人姓名：')
    
61.             if name in c:
    
62.                 del (c[name])
    
63.             else:
    
64.                 print('您输入的联系人不存在。')
    
65.         elif cid == 4:
    
66.             break
    
67.         else:
    
68.             print('请输入正确的id。')
    
69.     print('|--- 感谢使用通讯录程序 ---|')
    
70.     save_file(c)
    
71. 
    
72. 
    
73. if __name__ == '__main__':
    
74.     main()

复制代码