[已解决]代码优化和增加功能，求高手指导一下

zua

代码如下：
0：我想实现每次调用代码输入的帐号密码存放在c盘一个txt文档（已经实现）。
1：判断帐号密码是否存在这个过程是从这个txt文档调用数据来判断（不知道怎么实现）。
2：判断密码这块搞不明白了，加入到txt文本中的帐号和密码应该用什么格式去存放？才能实现调用的时候匹配密码呢？
3：代码优化，def函数不怎么会。

1. f_nr = open("C:\\帐号密码1.txt","a")
   
2. f_nr.close()
   
3. count = {}
   
4. flag = 2
   
5. while flag:
   
6.     flag = 2
   
7.     print("\n| --- 新建用户：N/n --- |")
   
8.     print("| --- 登录帐号：E/e --- |")
   
9.     print("| --- 退出程序：Q/q --- |")
   
10.     order = str(input("| ---   请输入指令  --- |\n"))
    
11. 
    
12.     if order == "n" or order == "N":
    
13.         f_nr = open("C:\\帐号密码1.txt","r")
    
14.         f = f_nr.read()
    
15.         f_nr.close()
    
16.         
    
17.         while (flag -1):
    
18.             name = str(input("请输入用户名："))
    
19.             while (flag-1):
    
20.                 if name not in f:
    
21.                     password = str(input("请输入密码："))
    
22.                     count[name] = password
    
23.                     print("注册成功，赶紧试试登录吧 ~")
    
24.                     flag = flag -1
    
25.                     
    
26.                     f_in = open("C:\\帐号密码1.txt","a")
    
27.                     f_in.write(str(count))
    
28.                     f_in.close()
    
29.                     break
    
30.                 else:
    
31.                     name = str(input("该帐号已存在，请重新输入："))
    
32. 
    
33. 
    
34.     elif order == "E" or order == "n":
    
35.         name = str(input("请输入用户名："))
    
36.         while flag:
    
37.             if name in count:               
    
38.                 password = str(input("请输入密码："))
    
39.                 while True:
    
40.                     if password == count[name]:
    
41.                         print("欢迎进入xxoo系统！")
    
42.                         flag = flag - 2
    
43.                         break
    
44. 
    
45.                     else:
    
46.                         password = str(input("输入错误，请重新输入密码："))
    
47. 
    
48.             else:
    
49.                 name = str(input("用户名不存在，请输入正确用户名："))
    
50. 
    
51.     elif order == "Q" or order == "q":
    
52.         print("程序关闭")
    
53.         break
    
54.    
    
55.     else:
    
56.         order = str(input("输入指令错误,重新输入："))
    

复制代码

最佳答案

月排行榜 / 总排行榜

冬雪雪冬

2016-11-28 20:33:34

python的一般文件只是读写字符串，对一定的结构数据可以用pickle模块。见小甲鱼视频31课。
我简单写了一下，除了文件不存在外，其他的错误处理都没有做，你可以再完善。

1. import os.path, pickle
   
2. while True:
   
3.     print("\n| --- 新建用户：N/n --- |")
   
4.     print("| --- 登录帐号：E/e --- |")
   
5.     print("| --- 退出程序：Q/q --- |")
   
6.     order = input("| ---   请输入指令  --- |\n")
   
7.     if order.upper() == 'N':
   
8.         if os.path.exists('帐号密码.pkl'):
   
9.             with open('帐号密码.pkl', 'rb') as f:
   
10.                 count = pickle.load(f)
    
11.         else:
    
12.             count = {}
    
13.         name = input("请输入用户名：")
    
14.         password = input("请输入密码：")
    
15.         count[name] = password
    
16.         with open('帐号密码.pkl', 'wb') as f:
    
17.             pickle.dump(count, f)
    
18.     elif order.upper() == "E":
    
19.         if os.path.exists('帐号密码.pkl'):
    
20.             with open('帐号密码.pkl', 'rb') as f:
    
21.                 count = pickle.load(f)
    
22.         else:
    
23.             print("尚无帐号密码存在！！！")
    
24.             continue
    
25.         name = input("请输入用户名：")
    
26.         if name in count:               
    
27.             password = input("请输入密码：")
    
28.             if password == count[name]:
    
29.                 print("欢迎进入xxoo系统！")
    
30.                 break
    
31.     elif order.upper() == "Q":
    
32.         print("程序关闭")
    
33.         break
    
34.         
    
35.         
    
36.         
    

复制代码

跳转到最佳答案楼层

zua

冬雪雪冬

python的一般文件只是读写字符串，对一定的结构数据可以用pickle模块。见小甲鱼视频31课。
我简单写了一下，除了文件不存在外，其他的错误处理都没有做，你可以再完善。

1. import os.path, pickle
   
2. while True:
   
3.     print("\n| --- 新建用户：N/n --- |")
   
4.     print("| --- 登录帐号：E/e --- |")
   
5.     print("| --- 退出程序：Q/q --- |")
   
6.     order = input("| ---   请输入指令  --- |\n")
   
7.     if order.upper() == 'N':
   
8.         if os.path.exists('帐号密码.pkl'):
   
9.             with open('帐号密码.pkl', 'rb') as f:
   
10.                 count = pickle.load(f)
    
11.         else:
    
12.             count = {}
    
13.         name = input("请输入用户名：")
    
14.         password = input("请输入密码：")
    
15.         count[name] = password
    
16.         with open('帐号密码.pkl', 'wb') as f:
    
17.             pickle.dump(count, f)
    
18.     elif order.upper() == "E":
    
19.         if os.path.exists('帐号密码.pkl'):
    
20.             with open('帐号密码.pkl', 'rb') as f:
    
21.                 count = pickle.load(f)
    
22.         else:
    
23.             print("尚无帐号密码存在！！！")
    
24.             continue
    
25.         name = input("请输入用户名：")
    
26.         if name in count:               
    
27.             password = input("请输入密码：")
    
28.             if password == count[name]:
    
29.                 print("欢迎进入xxoo系统！")
    
30.                 break
    
31.     elif order.upper() == "Q":
    
32.         print("程序关闭")
    
33.         break
    
34.         
    
35.         
    
36.         
    

复制代码

zua

谢谢，刚看到31课，这个问题可以解决了。因为还有些函数没学到，所以写的这么繁琐。

zua

改成完成啦，看起来略乱，但判断功能完备，打成当时的想法了。
感谢@冬雪雪冬

1. import pickle
   
2. import os
   
3. count = {}
   
4. #作为首次创建一个空字典文件，多次运行只会倒入一个空字典
   
5. f_new = open("C:\\帐号密码.pkl","ab")
   
6. pickle.dump(count,f_new)
   
7. f_new.close()
   
8. flag = 2
   
9. while flag:
   
10.     flag = 2
    
11.     print("\n| --- 新建用户：N/n --- |")
    
12.     print("| --- 登录帐号：E/e --- |")
    
13.     print("| --- 退出程序：Q/q --- |")
    
14.     order = str(input("| ---   请输入指令  --- |\n"))
    
15. 
    
16.     if order == "N" or order == "n":
    
17.         #第一次读取本地文件，判断该帐号是否已经注册过
    
18.         f_r = open("C:\\帐号密码.pkl","rb")
    
19.         f = pickle.load(f_r)
    
20.         f_r.close()
    
21.         while (flag -1):
    
22.             name = str(input("请输入用户名："))
    
23.             while (flag-1):
    
24.                 if name not in f:
    
25.                     password = str(input("请输入密码："))
    
26.                     count[name] = password
    
27.                     print("注册成功，赶紧试试登录吧 ~")
    
28.                     flag = flag -1
    
29.                     #以下是写入本地文件过程
    
30.                     f_r = open("C:\\帐号密码.pkl","rb")
    
31.                     f = pickle.load(f_r)#转换二进制文件
    
32.                     f_r.close()
    
33.                     f.update(count)#更新字典
    
34.                     f_r = open("C:\\帐号密码.pkl","wb")
    
35.                     pickle.dump(f,f_r)#覆盖式写入pkl
    
36.                     f_r.close()
    
37.                     break
    
38.                 else:
    
39.                     name = str(input("该帐号已存在，请重新输入："))
    
40. 
    
41. 
    
42.     elif order == "E" or order == "e" :
    
43.         f_r = open("C:\\帐号密码.pkl","rb")
    
44.         f = pickle.load(f_r)#读取pkl文件
    
45.         name_index = f.keys()#读取所有键也就是帐户
    
46.         name = str(input("请输入用户名："))
    
47.         while flag:
    
48.             if name in name_index:               
    
49.                 password = str(input("请输入密码："))
    
50.                 while True:
    
51.                     if password == f[name]:
    
52.                         print("欢迎进入xxoo系统！")
    
53.                         flag = flag - 2
    
54.                         break
    
55. 
    
56.                     else:
    
57.                         password = str(input("输入错误，请重新输入密码："))
    
58. 
    
59.             else:
    
60.                 name = str(input("用户名不存在，请输入正确用户名："))
    
61. 
    
62.     elif order == "Q" or order == "q":
    
63.         print("程序关闭")
    
64.         break
    
65.    
    
66.     else:
    
67.         order = str(input("输入指令错误,重新输入："))
    

复制代码