用户登录与注册

用户登录与注册：用函数封装功能
 1 def register(user):
 2     if user in info:
 3         user = input('此用户名已被使用，请重新输入：')
 4     password = input('请输入密码：')
 5     info[user] = password
 6     print('注册成功，赶紧试试登陆吧^_^')
 7 
 8 
 9 def login():
10     user = input('请输入登陆名：')
11     if not(user in info):
12         user = input('登陆名不存在，请重新输入：')
13     password = input('请输入密码：')
14     if info[user] == password:
15         print('欢迎进入XXOO系统，点击右上角的x结束程序')
16     else:
17         print('密码错误')
18 
19 
20 welcome = """
21 |---新建用户：N/n---|
22 |---登录账号：E/e---|
23 |---退出程序：Q/q---|
24 |---请输入指令代码：
25 """
26 
27 
28 info = {}
29 while True:  # 循环用来多次操作直到退出程序
30     choice = input(welcome)
31     if (choice == 'N') or (choice == 'n'):
32         user1 = input('请输入用户名：')
33         register(user1)
34     elif (choice == 'E') or (choice == 'e'):
35         login()
36     elif (choice == 'Q') or (choice == 'q'):
37         print('感谢使用，再见！')
38         break