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)