字典-用户登录系统

#!/user/bin/env python
db={}
def newuser():
    prompt='login desired:'
    while True:
        name =raw_input(prompt)
        if db.has_key(name):
            prompt='name taken,try another:'
            continue
        else:
            break
    pwd=raw_input('passwd:')
    db[name]=pwd
def olduser():
    name=raw_input('login:')
    pwd=raw_input('passwd:')
    passwd=db.get(name)
    if passwd==pwd:
        print 'welcome back',name
    else:
        print 'login incorrect'
def showmenu():
    prompt="""
    (N)ew User Login
    (E)existing User Login
    (Q)uit
    Enter choice:"""
    done=False
    while not done:
              chosen=False
              while not chosen:
                  try:
                      choice=raw_input(prompt).strip()[0].lower()
                  except (EOFError,KeyboardInterrupt):
                      choice='q'
                  print '/nYou picked:[%s]'%choice
                  if choice not in 'neq':
                      print 'invalid option,try again'
                  else:
                      chosen=True
              if choice=='q':done=True
              if choice=='n':newuser()
              if choice=='e':olduser()
if __name__=='__main__':
    showmenu()