创建内存中zip文件并以http响应返回的功能

Question

我正在避免在磁盘上创建文件，这是到目前为止的结果：

def get_zip(request):
    import zipfile, StringIO
    i = open('picture.jpg', 'rb').read()
    o = StringIO.StringIO()
    zf = zipfile.ZipFile(o, mode='w')
    zf.writestr('picture.jpg', i)
    zf.close()
    o.seek(0)
    response = HttpResponse(o.read())
    o.close()
    response['Content-Type'] = 'application/octet-stream'
    response['Content-Disposition'] = "attachment; filename=\"picture.zip\""
    return response

您认为正确，优雅，Python够吗？还有更好的方法吗？

谢谢！

Answer 1

对于StringIO您通常应该使用o.getvalue()以获得结果。另外，如果要将普通文件添加到zip文件中，可以使用zf.write('picture.jpg')。您不需要手动阅读它。