Django利用csv，HttpResponse，StreamingHttpResponse导出

一： 前言

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

二： 案例1-小文件导出

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

1. 新增数据库模型，插入数据；
2. 新建视图，利用HttpResponse，csv模块回去数据，生成csv格式返回；
3. 新建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]: L
Out[39]: [0, 2, 4, 6, 8]

In [40]: G = (x*2 for x in range(5))  #区别就是[]变成()

In [41]: G
Out[41]: <generator object <genexpr> at 0x7fad5268d910>

In [42]: for num in G:
    ...:     print(num)
    ...:     
0
2
4
6
8

下载地址：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

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