#coding=utf-8
importrsaimportbase64def create_keys(): #生成公钥和私钥
(pubkey, privkey) = rsa.newkeys(1024)
pub=pubkey.save_pkcs1()
with open('public.pem', 'wb+')as f:
f.write(pub)
pri=privkey.save_pkcs1()
with open('private.pem', 'wb+')as f:
f.write(pri)defencrypt(content):'''公钥加密
:param content:
:return:'''with open('public.pem', 'rb') as publickfile:
p=publickfile.read()
pubkey=rsa.PublicKey.load_pkcs1(p)
original_text=str(content).encode()#加密后的密文
crypt_text =rsa.encrypt(original_text, pubkey)return crypt_text #返回的是字节。
defdecrypt(crypt_text):'''私钥解密
:param crypt_text:
:return:'''with open('private.pem', 'rb') as privatefile:
p=privatefile.read()
privkey=rsa.PrivateKey.load_pkcs1(p)
lase_text= rsa.decrypt(crypt_text, privkey).decode() #注意,这里如果结果是bytes类型,就需要进行decode()转化为str
returnlase_textif __name__ == '__main__':#TODO 生成 .pem 证书
#create_keys() # 生成公钥和私钥,当前目录下会生成private.pem 和public.pem
#TODO 加密和解密 返回字节
encrypt_str = encrypt('你好中国')print(encrypt_str) #b"\tGu\t'\xbcx\xd4\xbb\xd7TI\xadS\x10\x1bPj\x12G\xd4z/\xdf\xa0$\xc2\xd1.\x14rI\xe5U\xb7\xef{\x8fX\xd5\x8d%H\x14\xc5\xaa\x15\xf1\xcd\x9aNCiqo\x99O@\x8b\xe2SsZ\xa9(5\xd3\xe5G!\x82\x1d~\x04\xc0(\xe5\xa7\x85\xb7P\xd6\x8a ?j#\xae\x17\x9a\xde\xb0\xa8\x90l\xb9\xd2\xe4\xf2G}\x8eH\x7f&\xe6\xd5\xcd\xd9\x02\xb6\x86\xc4.\xbc\xf7\xca\xd8\x10\xd3\x90,\xfe#}\xec\x0bL"
## base64编码后
print(str(base64.b64encode(encrypt_str), encoding="utf-8")) #CUd1CSe8eNS711RJrVMQG1BqEkfUei/foCTC0S4UcknlVbfve49Y1Y0lSBTFqhXxzZpOQ2lxb5lPQIviU3NaqSg10+VHIYIdfgTAKOWnhbdQ1oogP2ojrhea3rCokGy50uTyR32OSH8m5tXN2QK2hsQuvPfK2BDTkCz+I33sC0w=
#TODO 解密
decrypt_str =decrypt(encrypt_str)print(decrypt_str) #你好中国
#如果编码后解密的时候需要解码
print(bytes(base64.b64decode(str(base64.b64encode(encrypt_str), encoding="utf-8")))) #b"\tGu\t'\xbcx\xd4\xbb\xd7TI\xadS\x10\x1bPj\x12G\xd4z/\xdf\xa0$\xc2\xd1.\x14rI\xe5U\xb7\xef{\x8fX\xd5\x8d%H\x14\xc5\xaa\x15\xf1\xcd\x9aNCiqo\x99O@\x8b\xe2SsZ\xa9(5\xd3\xe5G!\x82\x1d~\x04\xc0(\xe5\xa7\x85\xb7P\xd6\x8a ?j#\xae\x17\x9a\xde\xb0\xa8\x90l\xb9\xd2\xe4\xf2G}\x8eH\x7f&\xe6\xd5\xcd\xd9\x02\xb6\x86\xc4.\xbc\xf7\xca\xd8\x10\xd3\x90,\xfe#}\xec\x0bL"