1 #-*- coding: utf-8 -*-
2 import hashlib
3 # 定义数据库(声明字典)
4 #注册登录的简单hash处理
5 db={}
6 def get_md5(password):
7 md5=hashlib.md5()
8 #此处密码hash加密处理
9 md5.update(password.encode('utf-8'))
10 return md5.hexdigest()
11
12 def register(username,password):
13 pwd=db.get(username,-1)
14 if pwd==-1:
15 db[username]=get_md5(username+password+'123456')
16 print('注册成功!')
17 print('user:%s'%username)
18 print('md5:%s'%db[username])
19 else:
20 print('用户已经存在!')
21 def login(username,password):
22 pwd=db.get(username,-1)#db.get()方法是获取value,若不存在返回-1
23 if pwd==-1:
24 print('用户不存在!')
25 elif get_md5(username+password+'123456')!=pwd:
26 print('用户名或密码不正确!')
27 else:
28 print('欢迎您,%s'%username)
29 print('开始注册!')
30 user=input('user:')
31 password=input('password:')
32 register(user,password)
33 print('开始登陆!')
34 user=input('user:')
35 password=input('password:')
36 login(user,password)