Python简要标准库（5）

hashlib

---

 Python的hashlib提供了常见的摘要算法，如MD5，SHA1等等。

基本的生成MD密匙的函数

import hashlib

md5 = hashlib.md5()
md5.update('This is my password'.encode('utf-8'))
print(md5.hexdigest())

 

若是把password改为了password2

所得到的MD5就完全不一样了

这体现了MD5的单向加密功能的强大

若要生成SHA1密匙的话，只需要将功能函数替换即可

 sha1 = hashlib.sha1() 



根据用户输入的登录名和口令模拟用户注册，计算更安全的MD5
　　使用了shelve模块 存储上次的用户数据信息

---

import hashlib
import shelve

#注册账号到数据库中   之前检测是否有重复
def Register(db,username,password):
    if username in db.keys():
        print ('find the same username : ')
        return
    
    #生成MD5密匙
    md5 = hashlib.md5()
    md5.update((username+password+'something').encode('utf-8'))
    md5_password = md5.hexdigest()
    
    #创建记录
    user = {}
    user['password'] = md5_password
    
    #保存记录
    db[username] = user
    
    #交互显示
    print ('Registered successfully!')
    print ('username : ',username)
    print ('password : ',password)
    


#显示数据库的信息
def showDataBase (db):
    print ('------This is database show------')
    for x in db.keys():
        d = db[x]
        print ('username : ',x)
        for key,value in d.items():
            print (key,':',value)
    print ('------over------')
#登录数据库
def SignIn(db,username,password):
    md5 = hashlib.md5()
    md5.update((username+password+'something').encode('utf-8'))
    md5_password = md5.hexdigest()
    #检测是否有用户账号
    if username in db.keys():
        if db[username]['password'] == md5_password:
            print ('welcome',username)
        else:
            print ('password is error')
    #不存在用户账号信息
    else:
        print ('no user call ',username)

#获取用户输入
def enter_command():
    cmd = input('Enter command (r,s,q,show) : ')
    cmd = cmd.strip().lower()
    return cmd

def get_user_input():
    username = input('pls input your username : ')
    password = input ('pls input your password : ')
    L = [username,password]
    return L
def main():
    database = shelve.open('userData.dat')

    try:
        while True:
            cmd = enter_command()
            if cmd == 'r':
                user_input = get_user_input()
                Register(database,user_input[0],user_input[1])
            elif cmd == 's':
                user_input = get_user_input()
                SignIn(database,user_input[0],user_input[1])
            elif cmd == 'show':
                showDataBase(database)
            elif cmd == 'q':
                print ('quit by user')
                break
            
    finally:
        database.close()

if (__name__) == '__main__':
    main()