对在Django中将CSV文件转成ZIP文件感到困惑

Question

我有一个视图，它从我的网站获取数据，然后把这些数据制作成一个压缩的csv文件。下面是我能正常运行的代码，没有压缩的部分：

def backup_to_csv(request):
    response = HttpResponse(mimetype='text/csv')
    response['Content-Disposition'] = 'attachment; filename=backup.csv'

    writer = csv.writer(response, dialect='excel')

    #code for writing csv file go here...

    return response

这个代码运行得很好。现在我想在发送这个文件之前，把它压缩一下。这就是我遇到问题的地方。

def backup_to_csv(request):

    output = StringIO.StringIO() ## temp output file
    writer = csv.writer(output, dialect='excel')

    #code for writing csv file go here...

    response = HttpResponse(mimetype='application/zip')
    response['Content-Disposition'] = 'attachment; filename=backup.csv.zip'

    z = zipfile.ZipFile(response,'w')   ## write zip to response
    z.writestr("filename.csv", output)  ## write csv file to zip

    return response

但这还不是全部，我完全不知道该怎么做。

Answer 1

zipfile.ZipFile(response,'w') 

在python 2.7.9中似乎不太好使。response是一个django.HttpResponse对象（据说它像文件一样），但是它报错了，提示"HttpResponse对象没有'seek'这个属性"。当在python 2.7.0或2.7.6中运行相同的代码时，一切正常……所以你最好在python 2.7.9中测试一下，看看是否会出现同样的问题。

Answer 2

注意，在正常工作的情况下，你是 return response... 而在不工作的情况下，你返回的是 z，而 z 当然不是一个 HttpResponse（而它应该是的！）。

所以：使用你的 csv_writer 不是在 response 上，而是在一个临时文件上；把临时文件压缩成一个 zip 文件；然后把这个压缩后的字节流写入 response！

Answer 3

好的，我明白了。这是我新的函数：

def backup_to_csv(request):

    output = StringIO.StringIO() ## temp output file
    writer = csv.writer(output, dialect='excel')

    #code for writing csv file go here...

    response = HttpResponse(mimetype='application/zip')
    response['Content-Disposition'] = 'attachment; filename=backup.csv.zip'

    z = zipfile.ZipFile(response,'w')   ## write zip to response
    z.writestr("filename.csv", output.getvalue())  ## write csv file to zip

    return response