一: 前言

django中,如何将网站数据csv文件给用户下载?我们前面有学习过httpResponse,结合起来一起用。

二: 案例1-小文件导出

1. 案例描述
我们将数据库中的信息,直接提供给客户端下载;

新增数据库模型,插入数据;新建视图,利用HttpResponse,csv模块回去数据,生成csv格式返回;新建url映射;
这个案例中,这种方式仅可以解决数据量小的问题,如果数据量非常大的情况下,可能会因为服务器超时而无法导出。

2. 数据库代码

from django.db import models# Create your models here.class Article(models.Model): id = models.BigAutoField(primary_key=True) title = models.CharField(max_length=100) content = models.TextField() price = models.FloatField(default=0) create_time = models.DateTimeField(auto_now_add=True) class Meta: db_table = 'k1_method_article'#使用makegrations生成迁移脚本#python manage.py makemigrations# 将新生成的脚本文件映射到数据库中# python manage.py migrate


3. 视图代码

def csv_downloads1(request): data = [] #最终要返回的列表数据 db_result = Article.objects.all() for row in db_result: source_data = [] #将数据库的每行信息生成一个列表 source_data.append(row.title) source_data.append(row.content) source_data.append(row.price) data.append(source_data) #将每一行数据列表,加入到一个大列表中 response = HttpResponse(content_type='text/csv') #定义一个HttpResponse,类型是CSV response['Content-Disposition'] = "attachment;filename=huangjaijin.csv" #定义返回的信息,附件方式及文件名称; writer = csv.writer(response) #利用csv模块写入response writer.writerow(['标题', '内容', '价格']) #标题行 for row in data: writer.writerow(row) #循环写入数据库信息 return response


3. URL代码

urlpatterns = [ path('csv_downloads1/', views.csv_downloads1, name='csv_downloads1'),]

4. URL代码
数据库插入数据

5. 访问下载
http://127.0.0.1:8000/k02_httpresponse/csv_downloads1/


二: 案例2-生成csv文件

def csv_downloads2(request): data = [ ['Name', 'Height'], ['Keys', '176cm'], ['HongPing', '160cm'], ['WenChao', '176cm'] ] with open('templates/abc.csv', 'w', encoding='utf-8', newline='') as f: writer = csv.writer(f, dialect='excel') for row in data: writer.writerow(row) return HttpResponse("成功")三: 案例3-大文件导出

假设我们需要导出一份1000W行的数据csv文件,使用案例1的方法是生成数据,然后csv写入response(等待写入),写入完成后,网页返回下载;那么在写入过程,就有可能因为服务器超时异常。有没有边生成,边下载的方式呢?我们可以使用StreamingHttpResponse对象实现。代码如下:
(1)利用使用StreamingHttpResponse实现

def large_csv_downloads(request): # 使用StreamingHttpResponse response = StreamingHttpResponse(content_type='text/csv') response['Content-Disposition'] = "attachment;filename=huangjaijin.csv" result = ("Row {}, {} \n".format(row, row) for row in range(0, 10000000)) #生成器 ''' print(type(result)): <class 'generator'> print(result): <generator object large_csv_downloads.<locals>.<genexpr> at 0x0000000004512660> ''' response.streaming_content = result return response

注:生成器案例讲解

In [38]: L = [x*2 for x in range(5)]In [39]: LOut[39]: [0, 2, 4, 6, 8]In [40]: G = (x*2 for x in range(5)) #区别就是[]变成()In [41]: GOut[41]: <generator object <genexpr> at 0x7fad5268d910>In [42]: for num in G: ...: print(num) ...: 02468

下载地址:http://127.0.0.1:8000/k02_httpresponse/large_csv_downloads

我们可以看到,在打开网页在瞬间就开始下载,因为是边生成数据,边下载。


(2)普通csv和HttpResponse
这里的案例就是,我们使用普通的方式下载同样的打的文件,试试。

def large_csv_downloads(request): #使用HttpResponse response = HttpResponse(content_type='text/csv') response['Content-Disposition'] = "attachment;filename=huangjaijin.csv" writer = csv.writer(response) for row in range(0, 10000000): writer.writerow(['Row {}'.format(row), ''.format(row)]) return response

我们就可以发现,当我们打开网页的时候,就是一只链接等待