一个通讯录

QQ小鱼

这是我在别人的启发下写的，他的比较简单，我帮忙写的更加深入

1. import os
   
2. import pickle
   
3. 
   
4. if not os.path.exists('E:/通讯.pkl'):
   
5.     with open('E:/通讯.pkl', 'wb') as f:
   
6.         pickle.dump({},f)
   
7. else:
   
8.     with open('E:/通讯.pkl', 'rb') as f:
   
9.         book = pickle.load(f)
   
10. 
    
11. def phone(a):
    
12.     length = len(a)
    
13.     while (a.isspace() or length == 0):#排除空格和没输入密码
    
14.         a = input('电话不能空，请重新输入电话：')
    
15.         length = len(a)
    
16.     if length == 11:
    
17.         flag = 0
    
18.     else:
    
19.         flag = 1
    
20.     for i in a:
    
21.         if i.isdigit():
    
22.             flag1 = 0
    
23.         else:
    
24.             flag1 = 1
    
25.             break
    
26.     while 1:
    
27.         if flag == 0 and flag1 == 0:
    
28.             break
    
29.         else:
    
30.             a = input('输入不合法，请输入正确的电话:')
    
31.             phone(a)
    
32.             break
    
33.     return a
    
34. 
    
35. def amend(book):
    
36.     s = 1
    
37.     while s:
    
38.         nam = input("请输入要修改的用户名：")
    
39.         if nam not in book:
    
40.             nam = input('用户不存在，请重新输入要修改的用户名：')
    
41.         try:
    
42.             choice = int(input("请问修改哪一项(按1性别，按2电话号码，按3家庭住址,按其他数字退出)："))
    
43.         except ValueError as reason:
    
44.             print('输入出错啦T_T\n错误原因是：' + str(reason))
    
45.         if choice in [1, 2, 3]:
    
46.             book[nam][choice_list[choice]] = input("请输入{}的修改值：".format(choice_list[choice]))
    
47.             s = input("是否继续修改，需要按0，不需要随意：")
    
48.             if s == '0':
    
49.                 s = 1
    
50.                 continue
    
51.             else:
    
52.                 s = 0
    
53.         else:
    
54.             print('本轮结束')
    
55.             s = 0
    
56.     print("-----修改完毕-----")
    
57.     with open('E:/通讯.pkl', 'wb') as f:
    
58.         pickle.dump(book, f)
    
59. 
    
60. def entering(book):
    
61.     print("-----录入模式-----")
    
62.     name = input("请输入用户名：")
    
63.     if name in book:  # 穿插修改部分
    
64.         try:
    
65.             name = int(input("电话簿中已有{}用户，是否修改？(是1/否2)".format(name)))
    
66.         except ValueError as reason:
    
67.             print('输入出错啦T_T\n错误原因是：' + str(reason))
    
68.         if name != 1:
    
69.             print("已结束本轮")
    
70.         elif name == 1:
    
71.             amend(book)
    
72.     else:
    
73.         try:
    
74.             gender = int(input("{}性别(男1/女2/未知3)：".format(name)))
    
75.         except ValueError as reason:
    
76.             print('输入出错啦T_T\n错误原因是：' + str(reason))
    
77.         if gender == 1:
    
78.             gender = "男"
    
79.         elif gender == 2:
    
80.             gender = "女"
    
81.         else:
    
82.             gender = "未知"
    
83.         phonenum = input("{}的电话号码：".format(name))
    
84.         phone(phonenum)
    
85.         home = input("{}的家庭住址(若未知按1)：".format(name))
    
86.         if home == '1':
    
87.             home = '未知'
    
88.         book[name] = {'用户名': name, '性别': gender, '电话号码': phonenum, '家庭住址': home}
    
89. 
    
90.         print("-----录入完毕-----")
    
91.         with open('E:/通讯.pkl', 'wb') as f:
    
92.             pickle.dump(book, f)
    
93.     return book
    
94. 
    
95. '''豆嘉木电话簿'''
    
96. 
    
97. print("-------欢迎使用豆嘉木电话簿-------")
    
98. 
    
99. choice_list = [0, '性别', '电话号码', '家庭住址']
    
100. while True:
     
101.     p = int(input("按1是录入，按2是查询按，按3是删除，按其他数字退出:"))  # 录入部分
     
102.     if p == 1:
     
103.         entering(book)
     
104.     elif p == 2:
     
105.         print("-----查询模式-----")
     
106.         try:
     
107.             check = input("请输入查询对象的名称：")
     
108.         except TypeError:
     
109.             print('输入错误')
     
110.         if check not in book:
     
111.             while 1:
     
112.                 check = int(input("%s不在电话簿中，请问是否增加联系人(是1/否2)："%check))
     
113.                 if check == 1:
     
114.                     entering(book)
     
115.                     break
     
116.                 else:
     
117.                     break
     
118.         else:
     
119.             print('用户名:%s \n性别:%s \n电话号码:%s \n家庭住址:%s'%(book[check]['用户名'], book[check]['性别'],
     
120.                                                               book[check]['电话号码'], book[check]['家庭住址']))
     
121.     elif p == 3:
     
122.         print("-----删除模式-----")
     
123.         delmem = input("需要删除哪一个用户(按数字退出):")
     
124.         if delmem.isdigit():
     
125.             print("-----退出删除模式-----")
     
126.         elif delmem not in book:
     
127.             delmem = int(input("%s不在电话簿中，请问是否增加联系人(是1/否2)：" %delmem))
     
128.             if delmem == 1:
     
129.                 entering(book)
     
130.         else:
     
131.             del book[delmem]
     
132.         with open('E:/通讯.pkl', 'wb') as f:
     
133.             pickle.dump(book, f)
     
134.         print("-----删除完毕-----")
     
135. 
     
136.     else:
     
137.         print("-----程序终止-----")
     
138.         break

复制代码

shiyouroc

给张图