python3.8 AES-ECB-PKCS5/PKCS7 加解密(支持中文)

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)
posted @ 2020-05-11 11:56  sfdevs  阅读(1191)  评论(0)    收藏  举报