GeezerVison is showing github.com — python_test/test43.py at master · sinferwu/python_test

Expand file tree

test43.py

Copy path

More file actions

Latest commit

History

46 lines (34 loc) · 1.36 KB

test43.py

Copy path

File metadata and controls

46 lines (34 loc) · 1.36 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

#!/usr/bin/env python3

# -*- conding: utf-8 -*-

'练习内建模块之hmac'

__author__ = 'sergiojune'

import random, hmac # 这个模块是可以根据口令进行加密保存数据，相当于md5加slat的效果，还比他强

message = b'hello world'

key = b'key'

# 第一个和第二个参数都必须是bytes类型

p = hmac.new(key, message, digestmod='md5') # 指定md5算法，一个参数为口令，第二个为加密的信息

print(p.hexdigest())

# 作业：将上一节的salt改为标准的hmac算法，验证用户口令

def hmac_md5(key, s):

return hmac.new(key.encode('utf-8'), s.encode('utf-8'), 'MD5').hexdigest()

class User(object):

def __init__(self, username, password):

self.username = username

self.key = ''.join([chr(random.randint(48, 122)) for i in range(20)])

self.password = hmac_md5(self.key, password)

db = {

'michael': User('michael', '123456'),

'bob': User('bob', 'abc999'),

'alice': User('alice', 'alice2008')

}

def login(username, password):

user = db[username]

return user.password == hmac_md5(user.key, password)

# 测试:

assert login('michael', '123456')

assert login('bob', 'abc999')

assert login('alice', 'alice2008')

assert not login('michael', '1234567')

assert not login('bob', '123456')

assert not login('alice', 'Alice2008')

print('ok')