Python 登录/注册 简单封装函数

770686119

1. # 定义指令主菜单
   
2. MenuNote = '''
   
3. =====================欢迎使用用户登录/注册小程序=================
   
4. 
   
5.                         |--- 新建用户：N/n ---|
   
6.                         |--- 登录账号：E/e ---|
   
7.                         |--- 退出程序：Q/q ---|
   
8. 
   
9. '''
   
10. # 存放用户名密码的字典
    
11. userInfo = {}
    
12. # 判断用户名唯一
    
13. def userNameIsUnique(userName):
    
14.     if userName in userInfo:
    
15.         return False
    
16.     else:
    
17.         return True
    
18. # 判断密码与确认密码是否一致
    
19. def userPassCheck(userPass,userPassSecond):
    
20.     if userPass == userPassSecond:
    
21.         print("密码设置成功")
    
22.         print(MenuNote)
    
23.         return True
    
24.     else:
    
25.         print("两次密码输入不一致,请重新设置")
    
26.         return False
    
27. #用户密码匹配校验
    
28. def passwordMatching(userName,userPass):
    
29.     if userInfo[userName] == userPass:
    
30.         print("密码验证通过,登录成功")
    
31.         return True
    
32.     else:
    
33.         print("用户名与密码不匹配")
    
34.         return False
    
35. #空字符串或空字符校验
    
36. def isEmpty(dataString):
    
37.     flag = 1 if  dataString.strip() is '' else 0
    
38.     return  flag
    
39. #设置密码
    
40. def setUserPass(userName):
    
41.     while True:
    
42.         userpassFirst = input("请为" + userName + "设置密码（第1次）")
    
43.         if isEmpty(userpassFirst):
    
44.             print("密码不能为空，请重设")
    
45.             continue
    
46.         userPassSecond = input("请为" + userName + "设置密码（第2次）")
    
47.         isSamePassFlag = userPassCheck(userpassFirst, userPassSecond)
    
48.         if isSamePassFlag:
    
49.             userInfo[userName] = userpassFirst
    
50.             print(userInfo)
    
51.             break
    
52. 
    
53. # 新建用户模块
    
54. def newUser(userName1):
    
55.     while True :
    
56.         userName = input("请输入注册用户名： ") if isEmpty(userName1)  else userName1
    
57.         while True :
    
58.             if isEmpty(userName):
    
59.                 userName = input("用户名不能为空,请重新输入")
    
60.             else:
    
61.                 break
    
62.         isUniqueFlag = userNameIsUnique(userName)
    
63.         #用户名是否唯一
    
64.         if isUniqueFlag:
    
65.             while True:
    
66.                 userpassFirst = input("请为"+userName+"设置密码（第1次）:")
    
67.                 if isEmpty(userpassFirst):
    
68.                     print("密码不能为空，请重设")
    
69.                     continue
    
70.                 userPassSecond = input("请为"+userName+"设置密码（第2次）:")
    
71.                 isSamePassFlag = userPassCheck(userpassFirst,userPassSecond)
    
72.                 if isSamePassFlag :
    
73.                     userInfo[userName] = userpassFirst
    
74.                     print(userInfo)
    
75.                     break
    
76. 
    
77. 
    
78.             break
    
79.         else:
    
80.             print("用户名重复")
    
81. # 登录模块
    
82. def userLogin():
    
83.     while True:
    
84.         userName = input("请输入登录账号： ")
    
85.         if isEmpty(userName) == 0:
    
86.             if userNameIsUnique(userName) == False:
    
87.                 userPass = input("请输入密码： ")
    
88.                 isMatch =  passwordMatching(userName, userPass)
    
89.                 if isMatch == True:
    
90.                     print(MenuNote)
    
91.                 else:
    
92.                     resetPasswordCode = input('是否重设密码？（Y/N）')
    
93.                     while True:
    
94.                         if resetPasswordCode == 'Y' or resetPasswordCode == 'y':
    
95.                             setUserPass(userName)
    
96.                             break
    
97.                         elif resetPasswordCode == 'N' or resetPasswordCode == 'n':
    
98.                             print('您选择了不重设密码，将回到主页面')
    
99.                             print(MenuNote)
    
100.                             break
     
101.                         else:
     
102.                             resetPasswordCode = input('指令不存在，请重新选择是否重设密码？（Y/N）')
     
103.             elif userNameIsUnique(userName) == True:
     
104.                 isNewUserCode = input("用户不存在，是否新建？（Y/N）")
     
105.                 while True:
     
106.                     if isNewUserCode == 'Y' or isNewUserCode == 'y':
     
107.                         newUser(userName)
     
108.                         break
     
109.                     elif isNewUserCode == 'N' or isNewUserCode == 'n':
     
110.                         print("你选择了否，将退回主菜单")
     
111.                         print(MenuNote)
     
112.                         break
     
113.                     else:
     
114.                         isNewUserCode = input('你输入的指令错误，请重新输入是否新建（Y:是 N：否）')
     
115.             break
     
116.         else:
     
117.             print("用户不能为空")
     
118. 
     
119. # 主函数部分
     
120. print(MenuNote)
     
121. username = ''
     
122. while True:
     
123.     code = input("请输入指令代码: ")
     
124.     if(code == 'N' or code == 'n'):
     
125.         newUser(username)
     
126.     elif(code == 'E' or code == 'e'):
     
127.         userLogin()
     
128.     elif(code == 'Q' or code == 'q'):
     
129.         print("感谢使用，再见！")
     
130.         break
     
131.     else:
     
132.         print("指令不存在，请重新输入")
     
133. 
     

复制代码

业务规则：
①新建用户，用户登录，退出的功能
②用户名，密码不能为空
③登录时用户不存在可直接新建
④用户密码匹配校验
⑤指令不存在的提示
⑥设置密码输2遍

----------------------测试场景--------
序号        场景覆盖测试
1        新建用户，输入唯一用户名，设置2次相同的密码，
2        新建用户，输入唯一用户名，设置2次不相同的密码，
3        新建用户，输入已存在的用户名
4        用户登录，输入不存在的用户名，是否新建选择‘是’，设置2次相同密码
5        用户登录，输入不存在的用户名，是否新建选择‘是’，设置2次不相同密码
6        用户登录，输入不唯一用户名，是否新建选择‘否’
7        用户登录，输入不唯一用户名，是否新建输入不存在的指令
8        用户登录，输入已存在的用户名，密码输入正确
9        用户登录，输入已存在的用户名，密码输入不正确
10        输入不存在的菜单指令
11        退出
12        新建用户，设置密码为空
13        新建用户，设置用户名为空
14        登录，输入密码不对，重设置密码选择‘是’
15        登录，输入密码不对，重设置密码选择‘否’
16        登录，重设用户名为空
17        登录，重设密码为空