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

import hashlib

db = {}

def get_md5(s):

    md5 = hashlib.md5()

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

    return md5.hexdigest()

def register(username, password):

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

    print('register success，please login')

def login(username, password):

    if not username in db:

        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)