from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex
class prpcrypt():
def __init__(self, key):
self.key = key
self.mode = AES.MODE_CBC
# 加密函数,如果text不是16的倍数【加密文本text必须为16的倍数!】,那就补足为16的倍数
def encrypt(self, text):
cryptor = AES.new(self.key, self.mode, self.key)
# 这里密钥key 长度必须为16(AES-128)、24(AES-192)、或32(AES-256)Bytes 长度.目前AES-128足够用
length = 16
count = len(text)
if (count % length != 0):
add = length - (count % length)
else:
add = 0
text = text + ('\0' * add)
self.ciphertext = cryptor.encrypt(text)
# 因为AES加密时候得到的字符串不一定是ascii字符集的,输出到终端或者保存时候可能存在问题
# 所以这里统一把加密后的字符串转化为16进制字符串
return b2a_hex(self.ciphertext)
# 解密后,去掉补足的空格用strip() 去掉
def decrypt(self, text):
cryptor = AES.new(self.key, self.mode, self.key)
plain_text = cryptor.decrypt(a2b_hex(text))
plain_text = plain_text.decode('ascii')
return plain_text.rstrip('\0')
使用
加密:
请注意保存生成的AES密匙
# 生成AES密匙
def random_password(num):
result = ''
choice = '0123456789' + string.ascii_lowercase
result += random.choice('0123456789')
result += random.choice(string.ascii_lowercase)
list = []
for i in range(num - 2):
a = random.choice(choice)
list.append(a)
result += a
return result
setKey = random_password(16)
cipher = encry.prpcrypt(str(setKey))
secretKey = cipher.encrypt(token)
print(secretKey.decode('utf-8'))
解密:
Key就是AESkey,str是上一步加密生成的密文
cipher = encry.prpcrypt(Key)
text = cipher.decrypt(str)
print(text)