Django实现web页面数据导出为CSV文件

一. 需求说明

在实际项目中,有一种需求是这样的:对于web页面显示的数据,业务人员需要将其下载到本地使用。针对这一需求,这里结合Django进行实现。

二. 后台实现的主要代码

# python3.6+Django1.6

import csv
import time
from django.http import HttpResponse

def students(request):
    # 根据条件查询相关数据
    stu_lis = models.Student.objects.all()
    # stus_lis = models.Student.objects.all().value_list('name', 'age', 'addr')
    response = HttpResponse(content_type='text/csv', charset='GBK')
    # 自定义文件名
    time_now = time.strftime('%Y%m%d')
    filename = 'students_' + time_now
    # 添加header,attachment表示以附件方式下载
    response['Content-Disposition'] = f'attachment; filename="{filename}.csv"'
    # 生成一个对象
    writer = csv.writer(response)
    # 定义表头
    writer.writerow(['姓名', '年龄', '地址'])
    
    # 添加数据
    # writer.writerows(stus_lis)
    for stu in stu_lis:
        writer.writerow([stu.name, stu.age, stu.addr])

    return response

上述代码中,数据查询数据添加时,分别使用了两种方法(注释掉了一种),两者的区别主要是添加数据时,是一条一条添加还是一次全部添加,使用writer.writerow()方法,每次添加一条数据,writer.writerows()可一次添加多条数据,但是若数据量过大,可能会导致内存溢出。因此,可根据实际情况选择使用。

三. 前台实现

前台页面中使用button按钮,并为其添加click事件,主要代码如下

# js实现点击下载
<button class="export">导出</button>

<script>
    $('.export').click(function () {
        location.href='http://127.0.0.1:8000/student/'
    })

</script>

刚开始想使用ajax异步请求,实现数据的下载,但是响应数据的类型,一直确定不了,所以后来直接使用location.href实现链接跳转,完成下载。如果哪位大佬有什么好的处理方法,还请不吝赐教

猜你喜欢

转载自www.cnblogs.com/joker-cai/p/12031771.html