python - Django StreamingHttpResponse 没有设置正确的文件名

Question

挣扎于此...我正在尝试使用 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',
)

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 View 中

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

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