Discuss / Python / 交作业

交作业

Topic source

-- coding:utf-8 --

import hashlib import random

def get_md5(s): return hashlib.md5(s.encode('utf-8')).hexdigest()

class User(object): def init(self, username, password): self.username = username self.salt = ''.join([chr(random.randint(48, 122)) for i in range(20)]) self.password = get_md5(password + self.salt)

db = { 'michael': User('michael', '123456'), 'bob': User('bob', 'abc999'), 'alice': User('alice', 'alice2008') }

def login(username, password): user = db[username] return user.password == get_md5(password + user.salt)

def register(username, password): db[username] = User(username, password)

测试:

register('ManiaJack', 'Python') assert login('ManiaJack', 'Python') assert not login('ManiaJack', 'python') 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')


  • 1

Reply