Django下载S3文件并包含在压缩包中

import os
import zipfile
import boto3
from django.http import HttpResponse

def download_files(request):
    # 获取要下载的文件列表
    file_list = ['file1.txt', 'file2.txt', 'file3.txt']

    # 创建一个临时目录来存储下载的文件
    temp_dir = '/path/to/temp/dir'
    os.makedirs(temp_dir, exist_ok=True)

    # 创建一个压缩文件
    zip_file_path = os.path.join(temp_dir, 'download.zip')
    with zipfile.ZipFile(zip_file_path, 'w') as zip_file:
        # 遍历文件列表，将每个文件下载并添加到压缩文件中
        for file_name in file_list:
            # 下载文件
            download_path = download_file_from_s3(file_name)

            # 将下载的文件添加到压缩文件中
            zip_file.write(download_path, file_name)

    # 构建HTTP响应，将压缩文件作为附件进行下载
    response = HttpResponse(content_type='application/zip')
    response['Content-Disposition'] = 'attachment; filename="download.zip"'

    # 将压缩文件内容写入HTTP响应
    with open(zip_file_path, 'rb') as zip_file:
        response.write(zip_file.read())

    # 删除临时目录和文件
    os.remove(zip_file_path)
    os.rmdir(temp_dir)

    return response

def download_file_from_s3(file_name):
    # 创建S3客户端
    s3 = boto3.client('s3')

    # 指定要下载的S3文件的存储桶和键
    bucket_name = 'your-bucket-name'
    key = 'path/to/' + file_name

    # 下载文件到本地临时目录
    download_path = os.path.join(temp_dir, file_name)
    s3.download_file(bucket_name, key, download_path)

    return download_path