Python---AES加密-CBC模式-PKCS7填充

最近公司业务需求，看到别人的文章，感觉很有用，故转载保存。

from Crypto.Cipher import AES
import base64
import time
import json


class Encrypt:
    def __init__(self, key, iv):
        self.key = key.encode('utf-8')
        self.iv = iv.encode('utf-8')

    # @staticmethod
    def pkcs7padding(self, text):
        """明文使用PKCS7填充 """
        bs = 16
        length = len(text)
        bytes_length = len(text.encode('utf-8'))
        padding_size = length if (bytes_length == length) else bytes_length
        padding = bs - padding_size % bs
        padding_text = chr(padding) * padding
        self.coding = chr(padding)
        return text + padding_text

    def aes_encrypt(self, content):
        """ AES加密 """
        cipher = AES.new(self.key, AES.MODE_CBC, self.iv)
        # 处理明文
        content_padding = self.pkcs7padding(content)
        # 加密
        encrypt_bytes = cipher.encrypt(content_padding.encode('utf-8'))
        # 重新编码
        result = str(base64.b64encode(encrypt_bytes), encoding='utf-8')
        return result


    def aes_decrypt(self, content):
        """AES解密 """
        cipher = AES.new(self.key, AES.MODE_CBC, self.iv)
        content = base64.b64decode(content)
        text = cipher.decrypt(content).decode('utf-8')
        return text.rstrip(self.coding)

if __name__ == '__main__':
    key = 'ONxYDyNaCoyTzsp83JoQ3YYuMPHxk3j7'
    iv = 'yNaCoyTzsp83JoQ3'

    ts = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
    p_json = {
		"CompanyName": "testmall",
		"UserId": "test",
		"Password": "grasp@101",
		"TimeStamp": "2019-05-05 10:59:26"
	}
    a = Encrypt(key=key, iv=iv)
    e = a.aes_encrypt(json.dumps(p_json))
    d = a.aes_decrypt(e)
    print("加密:", e)
    print("解密:", d)

注意最后输出格式为base64格式，如果要使用HEX输出格式，改为：

import binascii
# 重新编码
        result = binascii.b2a_hex(encrypt_bytes).decode()