import hashlib

db = {}

def get_md5(password): md5 = hashlib.md5() md5.update(password.encode('utf-8')) return md5.hexdigest()

def register(): print('开始注册') username = input('请输入用户名：') password = input('请输入密码:') db[username] = get_md5(password + username + 'the-Salt') print('注册成功\n')

def login(): print('开始登录') name = input('请输入用户名：') word = input('请输入密码:') try: if db[name] == get_md5(word + name + 'the-Salt'): print('登录成功\n') else: print('密码错误\n') except KeyError: print('用户名不存在\n')

while True: print('1.注册 2.登录 3.退出') a = input('请输入操作编号：') try: if int(a) == 1: register() elif int(a) == 2: login() elif int(a) == 3: exit() else: print('输入非法，请重输\n') except ValueError: print('输入非法，请重输\n')