python3----練習題(....)

 1 import re
 2 import hashlib
 3 import json
 4 
 5 def get_md5(pwd):     # md5加密
 6     m = hashlib.md5()
 7     m.update(pwd.encode('utf-8'))
 8     return m.hexdigest()
 9 
10 def get_name_md5():    # ‘加鹽’加密
11     hash = hashlib.md5('python'.encode('utf-8'))
12     hash.update('admin'.encode('utf-8'))
13     return hash.hexdigest()
14 
15 def write_to_file(content):
16     with open('dic.txt', 'a', encoding='utf-8') as f:
17         f.write(json.dumps(content, ensure_ascii=False) + '\n')
18         f.close()
19 
20 
21 def regist():
22     dd = {}
23     print('{0:*^50}'.format('用戶註冊界面'))
24     name = input("請輸入你的用戶名(以字母開頭):")
25     if re.match(r'^[a-zA-Z].{2,9}', name):
26         pwd = input("請設置密碼:")
27         pwd = get_md5(pwd)
28         dd[name] = pwd
29         write_to_file(dd)
30         print(dd)
31     else:
32         print("您輸入的用戶名格式不正確,請小於10位")
33         regist()
34 
35 
36 def login():
37     print('{0:*^50}'.format('用戶登陸界面'))
38     name = input('賬號:')
39     with open('dic.txt') as f:
40         for line in iter(f):
41             dic1 = json.loads(line)
42             if name in dic1:
43                 pwd = input('密碼:')
44                 pwd = get_md5(pwd)
45                 if dic1[name] == pwd:
46                     print('登陸成功')
47                     return True
48         print("賬號或密碼錯誤!")
49         login()
50 
51 def main():
52      login()
53 
54 if __name__ == '__main__':
55     main()