CSV 流响应 Django

Question

我正在通过查询数据库生成一个 csv 文件，并希望将该文件返回给用户以供下载。文件已损坏，好像下载在完成之前被中断。使用的代码如下：

class Echo(object):
    def write(self, value):
        return value
pseudo_buffer = Echo()
csv_data = []
for d in pesquisa:
    csv_data.append([d[h] for h in data])
writer = csv.writer(pseudo_buffer)
response = StreamingHttpResponse((writer.writerow(row) for row in csv_data),
content_type="text/csv")
response['Content-Disposition'] = 'attachment; filename="report.csv"'
return response

我正在使用 Django 1.8.9 + Gunicorn + Nginx

如果我使用 django 测试服务器得到以下错误：

ConnectionResetError: [Errno 104] Connection reset by peer

Answer 1

不需要StreamingHttpResponse：

csv_data = []
for d in pesquisa:
    csv_data.append([d[h] for h in data])

response = HttpResponse(content_type='text/csv')
writer = csv.writer(response)
writer.writerows(csv_data)
response['Content-Disposition'] = 'attachment; filename="report.csv"'

return response