在Django中提供动态生成的ZIP档案

Question

如何在Django中为用户提供动态生成的ZIP压缩文件？

我正在制作一个网站，用户可以选择任意组合的可用书籍，并将它们下载为ZIP压缩文件。我担心每次请求都生成这样的压缩文件会让我的服务器变得非常慢。我还听说Django目前没有很好的方法来提供动态生成的文件。

Answer 1

这里有很多回答建议使用 StringIO 或 BytesIO 这种缓冲区。但是其实并不需要，因为 HttpResponse 本身就是一个像文件一样的对象：

response = HttpResponse(content_type='application/zip')
zip_file = zipfile.ZipFile(response, 'w')
for filename in filenames:
    zip_file.write(filename)
response['Content-Disposition'] = 'attachment; filename={}'.format(zipfile_name)
return response

请注意，你不要调用 zip_file.close()，因为打开的“文件”是 response，我们绝对不想关闭它。

Answer 2

这是一个用Django框架写的视图，用来实现这个功能：

import os
import zipfile
import StringIO

from django.http import HttpResponse


def getfiles(request):
    # Files (local path) to put in the .zip
    # FIXME: Change this (get paths from DB etc)
    filenames = ["/tmp/file1.txt", "/tmp/file2.txt"]

    # Folder name in ZIP archive which contains the above files
    # E.g [thearchive.zip]/somefiles/file2.txt
    # FIXME: Set this to something better
    zip_subdir = "somefiles"
    zip_filename = "%s.zip" % zip_subdir

    # Open StringIO to grab in-memory ZIP contents
    s = StringIO.StringIO()

    # The zip compressor
    zf = zipfile.ZipFile(s, "w")

    for fpath in filenames:
        # Calculate path for file in zip
        fdir, fname = os.path.split(fpath)
        zip_path = os.path.join(zip_subdir, fname)

        # Add file, at correct path
        zf.write(fpath, zip_path)

    # Must close zip for all contents to be written
    zf.close()

    # Grab ZIP file from in-memory, make response with correct MIME-type
    resp = HttpResponse(s.getvalue(), mimetype = "application/x-zip-compressed")
    # ..and correct content-disposition
    resp['Content-Disposition'] = 'attachment; filename=%s' % zip_filename

    return resp

Answer 3

解决方案如下。

可以使用Python的一个模块zipfile来创建zip压缩文件。不过，在指定文件时要用StringIO对象（因为ZipFile构造函数需要一个像文件一样的对象）。接着，把你想要压缩的文件添加进去。然后在你的Django应用中，将StringIO对象的内容通过HttpResponse返回，并把类型设置为application/x-zip-compressed（或者至少是application/octet-stream）。如果需要的话，你可以设置content-disposition头，但其实这并不是必须的。

不过要注意，每次请求都创建zip压缩文件并不是个好主意，这可能会让你的服务器崩溃（如果压缩文件很大，还可能导致超时）。从性能的角度来看，比较好的做法是把生成的输出缓存到文件系统中的某个地方，只有在源文件发生变化时再重新生成。更好的办法是提前准备好压缩文件（比如通过定时任务），然后让你的web服务器像处理静态文件一样来提供这些压缩文件。