python rsa实现数据加密和解密、签名加密和验签功能实现数据加密和解密、签名加密和验签功能
本篇文章主要说明python库rsa生成密钥对,数据的加密解密,api接口的签名和验签,如有抄袭,请留言联系我。
先安装先安装
pip install rsa
安装好后,请看代码
"""
注意:
api签名时:签名用私钥,验签用公钥
数据加密时:加密用公钥,解密用私钥
密钥和公钥保存的图片会在下边展示出来
"""
import rsa
# 对api的签名机制进行验证:签名用私钥,验签用公钥
class RsaEncrypt:
def __init__(self, sign_str):
self.sign_str = sign_str
def rsa_generate(self):
"""
生成私钥和公钥并保存
:return:
"""
# 生成公钥和私钥
pubkey, privkey = rsa.newkeys(1024)
pub = pubkey.save_pkcs1()
# 公钥
with open('public.pem', 'wb') as w_pub:
w_pub.write(pub)
# 私钥
pri = privkey.save_pkcs1()
with open('private.pem', 'wb') as w_pri:
w_pri.write(pri)
return "保存成功"
@classmethod
def read_rsa(self):
"""
读取公钥和私钥
:return:
"""
with open('public.pem', 'rb') as publickfile:
pub = publickfile.read()
pubkey = rsa.PublicKey.load_pkcs1(pub)
with open('private.pem', 'rb') as privatefile:
priv = privatefile.read()
# print(pub)
privkey = rsa.PrivateKey.load_pkcs1(priv)
return pubkey, privkey
def str_sign(self):
privkey = self.read_rsa()[1] # 先将要加密的数据转成二进制
str_encode = self.sign_str.encode()
# 用私钥进行加密,并设置加密算法
signature = rsa.sign(str_encode, privkey, 'SHA-1') # 签名加密算法可以更换比如:SHA-256
# print(signature)
return signature
def sign_verify(self, signature):
"""
验证签名是否正确,如果正确,则返回签名算法,否则返回验证失败
:param signature:
:return:
"""
pubkey = self.read_rsa()[0] try:
agl = rsa.verify(self.sign_str.encode(), signature, pubkey)
# print(type(agl))
print(agl) # 返回加密算法代表验签成功
return True
except rsa.VerificationError:
print("验证失败")
return False
# 对数据进行加密:加密用公钥,解密用私钥
class DataEncrypt:
def __init__(self, data_str):
self.data_str = data_str
评论0