django下载csv文件笔记

下载小的csv文件

这里将用一个生成小的CSV文件为例，来把生成CSV文件的技术要点讲到位。我们用Python内置的csv模块来处理csv文件，并且使用HttpResponse来将csv文件返回回去。示例代码如下：

---

1.在templates目录下新建一个abc.txt的文件模板，内容如下

{% for row in rows%}{{row.0}},{{row.1}}
{% endfor %}

2.在view视图中编写如下代码

from django.shortcuts import render,loader
from django.http import HttpResponse,StreamingHttpResponse
import csv
def csvtest(request):
    #设置类型为csv
    response = HttpResponse(content_type='text/csv')
    #设置Content-Disposition属性，文件名为abc.csv
    response['Content-Disposition'] = "attachment;filename='abc.csv'"
    context = {
        'rows':[['username', 'age'],
                ['age', 18]]
    }
    render_csv = render(request, 'abc.txt',context=context)
    response.content = render_csv

    #或者使用csv写入内容
    # writer = csv.writer(response)
    # writer.writerow(['username', 'age'])
    # writer.writerow(['zhiliao', '18'])
    
    return response

这里再来对每个部分的代码进行解释：

1. 我们在初始化HttpResponse的时候，指定了Content-Type为text/csv，这将告诉浏览器，这是一个csv格式的文件而不是一个HTML格式的文件，如果用默认值，默认值就是html，那么浏览器将把csv格式的文件按照html格式输出，这肯定不是我们想要的。
2. 第二个我们还在response中添加一个Content-Disposition头，这个东西是用来告诉浏览器该如何处理这个文件，我们给这个头的值设置为attachment;，那么浏览器将不会对这个文件进行显示，而是作为附件的形式下载，第二个filename="abc.csv"是用来指定这个csv文件的名字。
3. 我们使用response.content设置response的内容

 

---

下载大的csv文件

以上的例子是生成的一个小的csv文件，如果想要生成大型的csv文件，那么以上方式将有可能会发生超时的情况（服务器要生成一个大型csv文件，需要的时间可能会超过浏览器默认的超时时间）。这时候我们可以借助另外一个类，叫做StreamingHttpResponse对象，这个对象是将响应的数据作为一个流返回给客户端，而不是作为一个整体返回。示例代码如下：

from django.shortcuts import render,loader
from django.http import HttpResponse,StreamingHttpResponse
def large_csv(request):
    response = StreamingHttpResponse(content_type='text/csv')
    response['content-Disposition']= "attachment;filename='lange.csv'"
    rows = ('Row{},{}/n'.format(row, row) for row in range(0,1000000))
    response.streaming_content = rows
    return response

这个类是专门用来处理流数据的。使得在处理一些大型文件的时候，不会因为服务器处理时间过长而到时连接超时。这个类不是继承自HttpResponse，并且跟HttpResponse对比有以下几点区别：

1. 这个类没有属性content，相反是streaming_content。
2. 这个类的streaming_content必须是一个可以迭代的对象。
3. 这个类没有write方法，如果给这个类的对象写入数据将会报错。

注意：StreamingHttpResponse会启动一个进程来和客户端保持长连接，所以会很消耗资源。所以如果不是特殊要求，尽量少用这种方法。