python3中Crypto的AES和RSA

# -*- coding: utf-8 -*-
import rsa

# 先生成一对密钥，然后保存.pem格式文件，当然也可以直接使用
(pubkey, privkey) = rsa.newkeys(1024)

pub = pubkey.save_pkcs1()
pubfile = open('public.pem','w+')
pubfile.write(pub)
pubfile.close()

pri = privkey.save_pkcs1()
prifile = open('privateKey.pem','w+')
prifile.write(pri)
prifile.close()

# rsa加密
def rsa_encrypt(message):
    with open('pubkey.pem') as f:
        key = f.read()
        rsakey = RSA.importKey(key)
        cipher = Cipher_pkcs1_v1_5.new(rsakey)
        cipher_text = base64.b64encode(cipher.encrypt(message))
        return cipher_text

# rsa解密
def rsa_decrypt(message):
    with open('privateKey.pem') as f:
        key = f.read()
        rsakey = RSA.importKey(key)
        cipher = Cipher_pkcs1_v1_5.new(rsakey)
        message = message.encode("ascii")
        byte_msg = base64.b64decode(message)
        random_generator = Random.new().read
        return cipher.decrypt(byte_msg, random_generator)

BLOCK_SIZE = 16  # Bytes
# 填充补码，不足16*N个字节，填充字符为chr(填充个数)
pad = lambda s: s + (BLOCK_SIZE - len(s) % BLOCK_SIZE) * \
                chr(BLOCK_SIZE - len(s) % BLOCK_SIZE)
# 去掉填充的补码
unpad = lambda s: s[:-ord(s[len(s) - 1:])]


def aes_encrpty(key, message):
    message = pad(message)
    cipher = AES.new(key, AES.MODE_ECB)
    aessource = cipher.encrypt(message)
    return base64.b64encode(aessource)


def aes_decrpty(key, message):
    cipher = AES.new(key, AES.MODE_ECB)
    aessource = cipher.decrypt(base64.b64decode(message))
    return unpad(aessource).decode('utf8')