python 8行代码搞定 AES加解密

python 实现AES加解密相关的知识，可以参考以下文章：
python实现AES加密解密

但该文章中，对于加密前数据的补全、及解密后去掉多余数据，由作者自己进行了封装。导致代码较为复杂。
实际可以使用库中pad和unpad来解决该问题，而使代码非常简洁。

代码如下：

import string
import base64
import random
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad


def get_random_digits(size=16):
    """Generate random number by size"""
    return "".join(random.choice(string.digits) for i in range(size))


def get_aes_key_iv():
    """Generate AES cryptor key and iv"""
    return get_random_digits(16).encode(), get_random_digits(16).encode()


def aes_encrypt(plain_text, aes_key, aes_iv):
    """
    AES encrypt
    :param plain_text: bytes
    :param aes_key: bytes
    :param aes_iv: bytes
    :return: bytes
    """
    return AES.new(aes_key, AES.MODE_CBC, aes_iv).encrypt(pad(plain_text, AES.block_size))


def aes_decrypt(plain_text, aes_key, aes_iv):
    """
    AES decrypt
    :param plain_text: bytes
    :param aes_key: bytes, aes_key
    :param aes_iv: bytes, aes_iv
    :return: bytes
    """
    return unpad(AES.new(aes_key, AES.MODE_CBC, aes_iv).decrypt(plain_text), AES.block_size)


def test_aes_cryptor(data, my_key, my_iv):
    rData = aes_encrypt(data, my_key, my_iv)
    print("密文：", base64.b64encode(rData))  # 转换成base64
    rData = aes_decrypt(rData, my_key, my_iv)
    print("明文：", rData)


if __name__ == "__main__":
    my_key1, my_iv1 = get_aes_key_iv()
    data1 = "hello word"
    test_aes_cryptor(data1.encode(), my_key1, my_iv1)

运行结果如下：

密文： b'mBtSmX/dSUa3nKTXvP1Prg=='
明文： b'hello word'