Python实现内存生成带密码保护Zip并通过Django HttpResponse返回

完全可以在内存中完成加密压缩操作，不需要将任何中间文件保存到文件系统，以下是两种可直接落地的实现方案：

方案1：基于7-Zip子进程管道实现 #

核心逻辑是利用7-Zip支持的标准输入/输出能力，直接在内存中完成数据传递，无需落地文件。

前置依赖 #

服务器需要提前安装7-Zip工具：

• Debian/Ubuntu系统执行：apt install p7zip-full
• CentOS/RHEL系统执行：yum install p7zip p7zip-plugins
• Windows系统可直接下载7-Zip并配置环境变量

代码实现（基于原有逻辑修改） #

from django.http import HttpResponse
import io, csv, subprocess

# 配置参数
ZIP_PASSWORD = "你的自定义密码"  # 可根据业务逻辑动态生成
EXPORT_FILE_NAME = "foobar.csv"
EXPORT_ZIP_NAME = "foobar.zip"

# 1. 内存生成CSV内容
buffer_string_csv = io.StringIO()
writer = csv.writer(buffer_string_csv)
writer.writerow(['foo','bar'])
# 可继续写入更多行业务数据
csv_content = buffer_string_csv.getvalue().encode('utf-8')  # 转字节类型供7z读取

# 2. 调用7z在内存中完成加密压缩
proc = subprocess.Popen(
    [
        '7z', 'a',  # 新增文件到压缩包命令
        '-tzip',  # 指定压缩格式为zip
        f'-p{ZIP_PASSWORD}',  # 设置压缩密码
        '-y',  # 自动确认所有提示
        f'-si{EXPORT_FILE_NAME}',  # 从标准输入读取内容，作为压缩包内的指定文件
        '-so'  # 将压缩后的结果输出到标准输出
    ],
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE
)
# 传入CSV内容，获取加密后的zip字节
encrypted_zip_content, stderr = proc.communicate(input=csv_content)

# 3. 直接返回加密压缩包给前端
response = HttpResponse(encrypted_zip_content)
response['Content-Type'] = 'application/x-zip-compressed'
response['Content-Disposition'] = f'attachment; filename="{EXPORT_ZIP_NAME}"'
return response

方案2：基于pyminizip纯Python实现 #

如果不想依赖系统安装的7-Zip，可以使用纯Python库pyminizip完成内存加密压缩，兼容性覆盖绝大多数解压工具。

前置依赖 #

安装Python库：pip install pyminizip

代码实现 #

from django.http import HttpResponse
import io, csv
import pyminizip

# 配置参数
ZIP_PASSWORD = "你的自定义密码"
EXPORT_FILE_NAME = "foobar.csv"
EXPORT_ZIP_NAME = "foobar.zip"
COMPRESS_LEVEL = 5  # 压缩等级1-9，数字越高压缩比越高、速度越慢

# 1. 内存生成CSV内容
buffer_string_csv = io.StringIO()
writer = csv.writer(buffer_string_csv)
writer.writerow(['foo','bar'])
csv_content = buffer_string_csv.getvalue().encode('utf-8')

# 2. 内存中生成加密zip
encrypted_zip_content = pyminizip.compress_to_buffer(
    [csv_content],  # 要压缩的文件内容字节列表
    [EXPORT_FILE_NAME],  # 压缩包内对应的文件名列表
    ZIP_PASSWORD,  # 压缩密码
    COMPRESS_LEVEL  # 压缩等级
)

# 3. 返回响应
response = HttpResponse(encrypted_zip_content)
response['Content-Type'] = 'application/x-zip-compressed'
response['Content-Disposition'] = f'attachment; filename="{EXPORT_ZIP_NAME}"'
return response

• 两种方案全程在内存中处理数据，不会产生任何临时磁盘文件，完全符合需求
• 密码不要硬编码在代码中，可根据业务逻辑结合用户身份、临时密钥等动态生成
• Zip 2.0加密算法安全性中等，如果涉及极高敏感数据，建议使用AES-256加密的7z格式，只需修改方案1中-tzip为-t7z、调整响应Content-Type即可

内容的提问来源于stack exchange，提问作者yhd.leung