AES算法进行文件加解密

import base64
import sys

from Crypto.Cipher import AES
import os


class EncryptDate:
    def __init__(self, key):
        self.key = key.encode('utf-8')  # 初始化密钥
        self.length = AES.block_size  # 初始化数据块大小
        self.aes = AES.new(self.key, AES.MODE_ECB)  # 初始化AES,ECB模式的实例
        # 截断函数，去除填充的字符
        self.unpad = lambda date: date[0:-ord(date[-1])]

    def pad(self, text):
        """
        #填充函数，使被加密数据的字节码长度是block_size的整数倍
        """
        count = len(text.encode('utf-8'))
        add = self.length - (count % self.length)
        entext = text + (chr(add) * add)
        return entext

    # 判断文件是否存在
    def is_exist(self, path):
        while True:
            if not os.path.exists(path):
                path = input('文件路径有误，请重新输入:\n')
            else:
                break
        return path

    def encrypt(self, encrData):  # 加密函数
        res = self.aes.encrypt(self.pad(encrData).encode("utf8"))
        msg = str(base64.b64encode(res), encoding="utf8")
        return msg

    def decrypt(self, decrData):  # 解密函数
        res = base64.decodebytes(decrData.encode("utf8"))
        # print(res)
        msg = self.aes.decrypt(res).decode("utf8")
        # print(msg)
        return self.unpad(msg)


key = '1234567890123456'  # 密钥

eg = EncryptDate(key)  # 这里密钥的长度必须是16的倍数

# while 1:
choose = input('选择文件处理方式（1.加密，2.解密）：')
if choose == '1':
    path = input('输入待加密文件路径：')
    res = eg.is_exist(path)
    if len(res) > 1:
        file = open(path, encoding='utf-8')
        result = eg.encrypt(file.read())
        print('加密内容：', result)
        print('加密格式：', type(result))
        enpath = input('文件加密完成，请输入加密文件存放位置：')
        with open(enpath, 'w') as f:
            f.write(result)
    else:
        print('结束程序')
        sys.exit(0)

if choose == '2':
    # 读取文件
    path = input('请输入待解密文件路径：')
    # 保证文件路径正常输入
    res = eg.is_exist(path)
    if len(res) > 0:
        f = open(path, 'r')
        file = f.read()
        # 解密处理
        r = eg.decrypt(file)
        # 保存解密文件
        depath = input('解密处理完成，请输入解密文件存放位置：')
        with open(depath, 'w+') as f:
            f.write(r)
    else:
        print('结束程序')
        sys.exit(0)

 效果图