使用Python将多个.CSV文件发送到.ZIP而无需在磁盘中存储

Question

我正在为我的Django-powered网站开发一个报告应用程序。我想运行几个报表，并让每个报表在内存中生成一个.csv文件，该文件可以作为.zip文件批量下载。我想这样做，而不存储任何文件到磁盘。到目前为止，要生成一个.csv文件，我要执行以下常见操作：

mem_file = StringIO.StringIO()
writer = csv.writer(mem_file)
writer.writerow(["My content", my_value])
mem_file.seek(0)
response = HttpResponse(mem_file, content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename=my_file.csv'

这很好，但只适用于一个解压缩的.csv文件。例如，如果我有一个使用StringIO流创建的.csv文件列表：

^{pr2}$

我怎样才能返回一个包含myFiles中所有对象的压缩文件，并能正确地解压缩以显示三个.csv文件？在

Answer 1

def zipFiles(files):
    outfile = StringIO() # io.BytesIO() for python 3
    with zipfile.ZipFile(outfile, 'w') as zf:
        for n, f in enumarate(files):
            zf.writestr("{}.csv".format(n), f.getvalue())
    return outfile.getvalue()

zipped_file = zip_files(myfiles)
response = HttpResponse(zipped_file, content_type='application/octet-stream')
response['Content-Disposition'] = 'attachment; filename=my_file.zip'

StringIO有返回整个内容的getvalue方法。你可以压缩拉链 通过zipfile.ZipFile(outfile, 'w', zipfile.ZIP_DEFLATED)。压缩的默认值是ZIP_STORED，它将创建不压缩的zip文件。在