Python zipfile加密

在Python中，`zipfile`模块主要用于创建、读取和操作ZIP归档文件，但它本身并不支持加密功能。如果你需要对ZIP文件进行加密，可以结合使用`zlib`库（用于压缩）和第三方库，如`pycryptodome`（提供高级加密功能）。

例如，你可以使用`zipfile`创建一个包含加密文件的ZIP，然后用`pycryptodome`的`AES`模块对数据进行加密：

from zipfile import ZipFile
import os
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad

# 创建一个ZipFile实例
with ZipFile('encrypted.zip', 'w', compression=zipfile.ZIP_DEFLATED) as zipf:

    # 加密函数
    def encrypt_data(data, key):
        cipher = AES.new(key, AES.MODE_CBC)
        encrypted = cipher.encrypt(pad(data.encode(), AES.block_size))
        return cipher.iv + encrypted

    # 假设有一个名为'data.txt'的文件需要加密
    with open('data.txt', 'rb') as f:
        data = f.read()
        key = b'some_secret_key'  # 你需要一个秘钥来加密
        iv = os.urandom(16)  # CBC模式需要一个随机初始化向量

        # 添加到ZIP，并加密内容
        encrypted_data = encrypt_data(data, key)
        zipinfo = ZipInfo(name='data.txt', date_time=datetime.now())
        zipinfo.filename = os.path.basename(zipinfo.filename)  # 设置原文件名
        zipinfo.compress_type = zipfile.ZIP_DEFLATED
        zipinfo.file_size = len(encrypted_data)
        zipinfo.CRC = zlib.crc32(encrypted_data)
        zipf.writestr(zipinfo, encrypted_data)