import hashlib

db = {}   # 先建立一个空字典

# 根据用户输入的口令，转化为MD5口令
def calc_md5(password):
    md5 = hashlib.md5()
    md5.update(password.encode('utf-8'))
    return md5.hexdigest()

# 注册用户，计算加盐MD5,返回用户名和MD5字典
def register(username, password):
    db[username] = calc_md5(password + username + 'the-Salt')
    return db  # 此时字典里有 {username: password}

# test1
print(register("agan","vvaganvv"))
#>>>{'agan': 'e2a718adb7d3efc56a7f5d8a9f589a21'}

# 验证用户密码是否正确
def login(username, password):
    md5 = calc_md5(password + username + 'the-Salt')
    if db[username] == md5:
        return True
    else:
        return False

# test2
print(login("agan","vvaganvv"))
#>>>True