python AES加密

因为原来的python aes 加密库Crypto已经很久没更新了，而且不支持python3，所以用了pyCryptodome这个库，它跟Crypto基本用法一样，但支持python2.4之后，和所有python3，安装：
pip install pycryptodome
直接上代码：#!usr/bin/env python3
# -*- coding: utf-8 -*-

"""
AES加密解密

需要使用pyCryptodome模块进行AES加密，安装：
pip install pycryptodome

pyCryptodome支持 python2.4 之后的版本和所有 python3 的版本
"""

__author__ = 'LiTian'

from Crypto.Cipher import AES
import base64

class MyAES:
def __init__(self, key, iv):
self.key = bytes(key, encoding='utf8')
self.iv = bytes(iv, encoding='utf8')
self.mode = AES.MODE_CBC

def my_encrypt(self, text):
"""加密函数"""
# 这里密钥key 长度必须为16（AES-128）、24（AES-192）、或32（AES-256）
my_aes = AES.new(self.key, self.mode, self.iv)
# 如果AES加密的text不是16的倍数，那就使用PKCS5Padding来为text填充（例如缺6位，则补6个6），如果刚好是16的倍数，则补16个bytes的16
length = 16
count = len(text)
self.padding = length - (count % length)
text = text + (chr(self.padding) * self.padding)
text = bytes(text, encoding='utf8') # 加密的文本必须是bytes
cipher_text = my_aes.encrypt(text)
# 统一把加密后的bytes转化为base64
return str(base64.b64encode(cipher_text), encoding='utf8')

def my_decrypt(self, text):
"""用base64解密后，用rstrip()去掉补足的字符"""
my_aes = AES.new(self.key, self.mode, self.iv)
plain_text = my_aes.decrypt(base64.b64decode(bytes(text, encoding='utf8')))
return str(plain_text, encoding='utf8').rstrip(chr(self.padding))因为是我这边加密，然后同事那边用java解密的，所以我配合着他那边的程序，自己实现pkcs5来做填充，并将结果编译成base64，并转为string格式。
解密方面还不能做到单独使用，欢迎补充