python - 压缩并服务器多个内存中文件

Question

我目前正在使用下面的代码生成一个Word文档，然后使用cherrypy在网络上提供该文档。

tpl.get_docx().save(iostream)
cherrypy.response.headers['Content-Type'] = (
            'application/vnd.openxmlformats-officedocument'
            '.wordprocessingml.document'
            )
cherrypy.response.headers['Content-Disposition'] = (
            'attachment; filename={fname}.docx'.format(
                fname='SP' + kwargs['sp'] + '-'+ kwargs['WO'] + ' ' + kwargs['site'] + ' - ' + 'RPC Report'  +'.docx'
            )
            )
iostream.seek(0)
return file_generator(iostream)

我计划创建更多文档，然后将它们压缩到内存中，然后在网络上提供它们。这是如何实现的，我尝试过使用 zipfile 库，压缩内存文件似乎很复杂。

我用谷歌搜索的以下示例可能会解决我的问题，但不确定如何使用它。

import zipfile
import StringIO

zipped_file = StringIO.StringIO()
with zipfile.ZipFile(zipped_file, 'w') as zip:
    for i, file in enumerate(files):
        file.seek(0)
        zip.writestr("{}.csv".format(i), file.read())

zipped_file.seek(0)

Answer 1

经过几个小时的坚持，我成功了，是啊

iostream = BytesIO()
tpl.get_docx().save(iostream)

iostream1 = BytesIO()
tpl1.get_docx().save(iostream1)

zip_output = StringIO.StringIO()
file = zipfile.ZipFile(zip_output, "w")
file.writestr("test.docx", iostream.getvalue())
file.writestr("test1.docx", iostream1.getvalue())
file.close()

cherrypy.response.headers['Content-Type'] = 'application/zip'
cherrypy.response.headers['Content-Disposition'] = 'attachment; filename="test.zip"'

return zip_output.getvalue()