网站各注册用户的密码的摘要存储

import hashlib

db = {}

def get_md5(s):

    md5 =hashlib.md5()

   md5.update(s.encode('utf-8'))

    returnmd5.hexdigest()

def register(username, password):

    db[username] =get_md5(password + username + 'the-sAlt')

    print('registersuccess，please login')

def login(username, password):

    if not username indb:

       print('User is not exists!')

       return

    if db[username] ==get_md5(password + username + 'the-sAlt'):

       print('Login successfully.')

    else:

       print('Incorrect password!')

if __name__ == '__main__':

    u1 = 'zhangsan'

    p1 = '112233'

    register(u1, p1)

    #测试成功登录

    login(u1, p1)

    

    #测试错误密码

    login(u1, p1 + '')

    #测试错误用户名

    login(u1 + ' ',p1)