实现要求:1. 支持新用户注册,新用户名和密码注册到字典中 2.支持老用户登录,用户名和密码提示登录成功 3. 主程序通过循环询问进行何种操作,根据用户的选择,执行注册或是登录操作。 有待后续优化!
dictory = {} #字典
def new_user(): #定义新用户函数
nu = input('请输入新用户名:').strip()
if nu:
if nu in dictory:
print('用户名已存在,请重新输入!')
else:
np = input('请输入注册密码:').strip()
dictory[nu] = np
print('注册成功!')
else:
print('用户名不能为空')
def old_user_login(): #定义老用户登录函数
while 1:
ou = input('请输入用户名:').strip()
if ou not in dictory:
print('用户名输入错误,请重新输入!')
op = input('请输入密码:').strip()
if dictory.get(ou) == op:
print('\033[31m登陆成功\033[0m')
break
else:
print('\033[31m密码错误,重新输入\033[0m')
continue
def select(): #定义用户菜单选择
funcs = {'1': new_user, '2': old_user_login, '3': quit}
prompt = '''(1) 注册新用户:
(2) 老用户登录:
(3) 退出
请选择(1/2/3):'''
while 1: #主程序循环
choice = input(prompt).strip()
if choice not in ['1','2','3']:
print('无效的输入,请重试。')
continue
if choice == '3':
print('Bye-bye')
break
funcs[choice]()
if __name__ == '__main__':
select()