挣扎于此...我正在尝试使用 StreamingHttpResponse 输出数据的 csv。文件本身构造良好,但由于某种原因,文件名在不同浏览器中设置不正确。我正在使用带有自定义 csv 渲染器的 Django Rest Framework。 CSVRenderer 从 Django Rest Framework ReadOnlyModelViewset 中获取标准数据,并使用 StreamingHttpResponse 呈现它。 CSVRenderer 是基于 Django docs explaining how to output a csv 编写的
import csv
from rest_framework.renderers import BaseRenderer
from django.http import StreamingHttpResponse
class Echo(object):
'''An object that implements just the write method of the file-like
interface.
'''
def write(self, value):
'''Write the value by returning it, instead of storing in a buffer.
'''
return value
class CSVRenderer(BaseRenderer):
'''Renderer class to generate basic csvs of the data supplied by the
serializer
'''
media_type = 'text/csv'
format = 'csv'
def render(self, data, media_type=None, renderer_context=None):
rows = []
first_data_row = next(iter(data['results']), None)
# Create header rows if data present
if first_data_row:
rows.append(list(first_data_row.keys()))
# print the data to rows
for row in data['results']:
rows.append(list(row.values()))
pseudo_buffer = Echo()
writer = csv.writer(pseudo_buffer)
response = StreamingHttpResponse((writer.writerow(row) for row in rows), content_type='text/csv')
response['Content-Type'] = 'text/csv; name="export.csv"'
response['Content-Disposition'] = 'attachment; filename="export.csv"'
return response
所以当我对 url 发出 GET 请求时,例如http://example.com/api/mydata?format=csv 在 IE 中我得到一个包含正确数据的文件 mydata.txt 这是错误的文件名和扩展名!在 Firefox 中,我得到一个包含名为 XuKsgJ.csv 的正确数据的文件,它是正确的扩展名,但只是一个随机文件名!
我想我没有在某处设置某些东西,但我不知道在哪里。有什么建议么?运行 Django 1.10.8,Django Rest Framework 3.5.3。我也尝试使用 djangorestframework-csv代替我的自定义 CSVRenderer 但得到相同的输出。与我的中间件有关吗?这就是我的 settings.py
MIDDLEWARE_CLASSES = (
'django.contrib.sessions.middleware.SessionMiddleware',
'django.middleware.common.CommonMiddleware',
'django.middleware.csrf.CsrfViewMiddleware',
'django.contrib.auth.middleware.AuthenticationMiddleware',
'django.contrib.auth.middleware.SessionAuthenticationMiddleware',
'django.contrib.messages.middleware.MessageMiddleware',
'django.middleware.clickjacking.XFrameOptionsMiddleware',
'django.middleware.security.SecurityMiddleware',
)
请您参考如下方法:
我也遇到过这个问题,但我在您的代码中看到您错误地覆盖了 render 方法。 CSVRenderer 类具有返回字符串而不是响应的渲染函数。
def render(self, data, media_type=None, renderer_context=None): ...
"""
Renders serialized *data* into CSV. For a dictionary:
"""
因为这个函数应该返回字符串,而不是响应对象。
实际上有一种方法可以更改文件名:
只需将 finaliez_response 方法(这是返回响应之前的最后一个方法)添加到您的 API View 中
def finalize_response(self, request, response, *args, **kwargs):
response['Content-Disposition'] = 'attachment; filename="export.csv"'
return super().finalize_response(request, response, *args, **kwargs)
希望对遇到这个问题的人有所帮助!
