AES加解密工具类

#!/usr/bin/env python
# -*- coding=utf-8 -*-
"""
AES加密解密工具类
此工具类加密解密结果与 http://tool.chacuo.net/cryptaes 结果一致
数据块128位
key 为16位
iv 为16位，且与key相等
字符集utf-8
输出为base64
AES加密模式 为cbc
填充 pkcs7padding
"""

import base64
from Crypto.Cipher import AES


class AESTOOLS:

    def encrypt(self, key, data):
        """
        aes加密
        """
        key = aes.add_16(self, key)
        cipher = AES.new(key, AES.MODE_ECB)
        # 处理明文
        content_padding = aes.pkcs7padding(self, data)
        # 加密
        encrypt_bytes = cipher.encrypt(bytes(content_padding, encoding='utf-8'))
        # 重新编码
        result = str(base64.b64encode(encrypt_bytes), encoding='utf-8')
        return result

    def decrypt(self,key, data):
        '''
        aes解密
        :param key: 密钥
        :param data: 加密后的数据（密文）
        :return:明文
        '''

        unpad = lambda s: s[:-ord(s[len(s) - 1:])]
        key = key.encode('utf8')
        data = base64.b64decode(data)
        cipher = AES.new(key, AES.MODE_ECB)

        # 去补位
        text_decrypted = unpad(cipher.decrypt(data))
        text_decrypted = text_decrypted.decode('utf8')
        return text_decrypted
	
    def add_16(self, par):
        par = par.encode('gbk')
        while len(par) % 16 != 0:
            par += b'\x00'
        return par
		
	def pkcs7padding(self, data):
        """
        明文使用PKCS7填充
        最终调用AES加密方法时，传入的是一个byte数组，要求是16的整数倍，因此需要对明文进行处理
        :param text: 待加密内容(明文)
        :return:
        """
        bs = AES.block_size  # 16
        length = len(data)
        bytes_length = len(bytes(data, encoding='utf-8'))
        # tips：utf-8编码时，英文占1个byte，而中文占3个byte
        padding_size = length if(bytes_length == length) else bytes_length
        padding = bs - padding_size % bs
        # tips：chr(padding)看与其它语言的约定，有的会使用'\0'
        padding_text = chr(padding) * padding
        return data + padding_text