import getpass
userdb = {}
def register():
username = input('username: ')
if username in userdb:
print('%s already exits.' % username)
else:
password = input('password: ')
userdb[username] = password
def login():
username = input('username: ')
password = getpass.getpass('password: ')
if userdb.get(username) != password:
print('login failed')
else:
print('login successful')
def show_menu():
cmds = {'0':register, '1':login}
prompt = '''
(0) register
(1) login
(2) exit
Please input your choice(0/1/2):
'''
while True:
choice = input(prompt).strip()[0]
if choice not in '012':
print('Tnvalid input, Try again.')
continue
if choice == '2':
break
cmds[choice]()
if __name__ == '__main__':
show_menu()
结果输出: