一、生成小型的csv文件
1、直接处理数据
from django.http import HttpResponse
import csv
# 简单生成下载csv文件
def down_csv_1(request):
# 指定返回的类型为csv
response = HttpResponse(content_type='text/csv')
# 添加返回头说明如何处理这个返回对象,并指明文件名。attachment:作为附件的形式进行下载
response['Content-Disposition'] = "attachment;filename='test.csv'"
# 创建写的对象
writer = csv.writer(response)
# 写入一行数据
writer.writerrow(['username','age'])
# 写入下一行数据
writer.writerrow(['Xsan',20])
return response
2、使用模版处理要下载的数据:
(1)在templates文件夹下面创建处理数据的模版文件比如test.txt。
test.txt文件
{# 一定要将循环体和for语句写在一行,否则会出现空行的情况 #}
{% for row in rows %}{{ row.0 }},{{ row.1 }}
{% endfor %}
(2)在视图函数中引用
views.py
from django.http import HttpResponse
from django.template import loader
import csv
#使用模版处理要下载的数据
def down_csv_2(request):
# 指定返回的类型为csv
response = HttpResponse(content_type='text/csv')
# 添加返回头说明如何处理这个返回对象,并指明文件名。attachment:作为附件的形式进行下载
response['Content-Disposition'] = "attachment;filename='test.csv'"
# 创建要下载的数据,此时要和模版结合起来
context = {
'rows':[
['username','age'],
['Xsan',20]]
}
# 引用自定义的模版
template = loader.get_template('test.txt')
# 将要下载的数据使用模版格式化
csv_template = template.render(context)
# 将处理好的数据渲染至上下文
response.content = csv_template
return response
二、生成大型的csv文件
上面我们讲了生成一个小的csv文件,如果要生成大型的csv文件该如何做呢?上面的代码可能会发生超时的情况(服务器要生成一个大型的csv文件,需要的时间可能会超过浏览器默认的时间)。这时候我们需要借助一个类StreamingHttpResponse对象,这个对象能将响应的数据作为一个流返回给客户端,而不是作为一个整体返回。
关于StreamingHttpResponse:
这个类时专门用来处理流数据的。使得在处理大型文件的时候,不会因为服务器处理时间过长而导致连接超时。这个类不是继承自HttpResponse,并且跟HttpResponse对比有以下几点区别:
1、这个类没有属性content,相反是streaming_content。
2、这个类的streaming_content必须是一个可迭代对象。
3、这个类没有write方法,如果给这个类的对象写入数据将会报错。
注意:StreamingHttpResponse会启动一个进程来和客户端保持长连接,所以会很消耗资源。所以如果不是特殊要求,尽量少用这种方法。
views.py
from django.http import HttpResponse,StreamingHttpResponse
from django.template import loader
import csv
def large_csv_view(request):
response = StreamingHttpResponse(content_type='text/csv')
response['Content-Disposition'] = "attachment;filename='large.csv'"
rows = ("Row {},{}\n".format(row,row) for row in range(0,10000))
response.streaming_content = ("username,age\n","Xsan,20\n")
return response