Django StreamingHttpResponse未设置正确的文件名

Question

在这个问题上苦苦挣扎。我正在尝试使用StreamingHttpResponse输出csv数据。文件本身构造良好，但由于某些原因，文件名在不同的浏览器中设置不正确。我正在使用带有自定义csv渲染器的Django Rest框架。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，例如IE中的http://example.com/api/mydata?format=csv发出GET请求时，我得到了一个包含名为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',
)

Answer 1

我也遇到过这个问题，但我在您的代码中看到，覆盖render方法是错误的。类CSVRenderer的渲染函数返回字符串，而不是响应。

def render(self, data, media_type=None, renderer_context=None): ...
"""
   Renders serialized *data* into CSV. For a dictionary:
"""

原因这个函数应该返回字符串，而不是响应对象。

实际上，有一种方法可以更改文件名：

只需将finaliez_response方法(它是返回响应前的最后一个方法)添加到您的API视图中

  def finalize_response(self, request, response, *args, **kwargs):
    response['Content-Disposition'] = 'attachment; filename="export.csv"'
    return super().finalize_response(request, response, *args, **kwargs)

希望对遇到这个问题的人有所帮助！

Answer 2

您还可以覆盖render方法并在其中设置文件名

 def render(self, data, media_type=None, renderer_context={}, writer_opts=None):
        rendered_data = super(MyRenderer, self).render(data, media_type, renderer_context, writer_opts)
        renderer_context['response']['Content-Disposition'] = "attachment; filename=%s" % ('ab.csv')
        return rendered_data