python AES加密與解密

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)