[python] 字典运用：注册登陆系统

user_data = {}

def new_user():
      prompt = '请输入用户名：'
      while 1:
            name = input(prompt)
            if name in user_data:
                  prompt = '此用户名已经被使用，请重新输入：'
                  continue
            else:
                  break
      passwd = input('请输入密码：')
      user_data[name] = passwd
      print('注册成功！')
def old_user():
      prompt = '请输入用户名：'
      while 1:
            name = input(prompt)
            if name not in user_data:
                  prompt = '此用户不存在，请重新输入：'
                  continue
            else:
                  break
      passwd = input('请输入密码：')
      pwd = user_data.get(name)
      if passwd == pwd:
            print('欢迎进入系统！')
      else:
            print('密码错误！')
def showmenu():
      prompt = '''
|---新建用户：N---|
|---登陆账号：E---|
|---退出程序：Q---|
|---输入指令代码---|'''
      while 1:
            chosen = False
            while not chosen:
                  choice = input(prompt)
                  if choice not in 'NEQ':
                        print('指令错误重新输入：')
                  else:
                        chosen = True
            if choice =='Q':
                  break
            if choice == 'N':
                  new_user()
            if choice == 'E':
                  old_user()

showmenu()