import base64
from Crypto.Cipher import AES
class EncryptDate:
def __init__(self, key):
self.key = key # 初始化密鑰
self.length = AES.block_size # 初始化數據塊大小
self.aes = AES.new(self.key.encode("utf8"), AES.MODE_ECB) # 初始化AES,ECB模式的實例
# 截斷函數,去除填充的字符
self.unpad = lambda date: date[0:-ord(date[-1])]
def pad(self, text):
"""
#填充函數,使被加密數據的字節碼長度是block_size的整數倍
"""
count = len(text.encode('utf-8'))
add = self.length - (count % self.length)
entext = text + (chr(add) * add)
return entext
def encrypt(self, encrData): # 加密函數
res = self.aes.encrypt(self.pad(encrData).encode("utf8"))
msg = str(base64.b64encode(res), encoding="utf8")
return msg
def decrypt(self, decrData): # 解密函數
res = base64.decodebytes(decrData.encode("utf8"))
msg = self.aes.decrypt(res).decode("utf8")
return self.unpad(msg)
eg = EncryptDate("m1C0s3oRBtxeYy3K") # 這里密鑰的長度必須是16的倍數
encrypt_text = eg.encrypt("0990e3dac6448c6c583bd74992cc8a42")
print('encrypt_text : ', encrypt_text)
decrypt_text = eg.decrypt(encrypt_text)
print('encrypt_text :', decrypt_text)
encrypt_text = eg.encrypt("AES 怎么樣")
print('encrypt_text : ', encrypt_text)
decrypt_text = eg.decrypt(encrypt_text)
print('encrypt_text :', decrypt_text)