python hash模块用key加密方法

#-*-coding:utf-8 -*-
#hashlib模块练习
import hashlib
def hash_lib(hash_type,hash_words,*key):  #传入加密类型，加密的字符串，密钥
    if key:   #判断是否有密钥
        if hash_type=='md5':
            m=hashlib.md5(bytes(key[0],encoding='utf8'))  #md5加密算法
        elif hash_type=='sha256':
            m=hashlib.sha256(bytes(key[0],encoding='utf8'))   #sha256加密算法
        else:
            return    #结束函数，返回None
    else:   #无密钥
        if hash_type=='md5':
            m=hashlib.md5()  #md5加密算法
        elif hash_type=='sha256':
            m=hashlib.sha256()   #sha256加密算法
        else:
            return    #结束函数，返回None
    m.update(hash_words.encode('utf-8'))   #对字符串进行加密
    return m.hexdigest()   #返回加密结果

ret=hash_lib('md5','abc','123')    #调用hash_lib函数进行加密运算
print(ret)

 相关链接：https://my.oschina.net/kefy/blog/1800202