Skip to main content
 首页 » 编程设计

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

2024年12月31日35lori

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

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