2

与这个苦苦挣扎...我正在尝试用来StreamingHttpResponse输出数据的csv。该文件本身构造良好，但由于某种原因，文件名在不同浏览器中的设置不正确。我正在使用带有自定义 csv 渲染器的 Django Rest Framework。从CSVRendererDjango Rest Framework 获取标准数据ReadOnlyModelViewset并使用StreamingHttpResponse. 这CSVRenderer是基于Django 文档编写的，解释了如何输出 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',
)
4

2 回答 2

1

我也遇到了这个问题，但是我在您的代码中看到您错误地覆盖了渲染方法。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)

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

于 2018-10-05T16:37:18.463 回答
0

您也可以覆盖渲染方法并在那里设置文件名

 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
于 2018-10-10T05:55:34.563 回答