django返回excel文件给前端下载

古城微笑少年丶 2022-08-28 06:49 234阅读 0赞

### Django使用openpyxl返回excel文件给前端下载 ###

虚拟环境导入

pip install openpyxl

views.py代码：

import datetime
    from openpyxl import Workbook
    
    def export_data(request, *args, **kwargs):
        # 使用get方法
        res = { }
        if request.method != 'GET':
            res['retcode'] = 400
            res['status'] = '只支持Get方法'
            return JsonResponse(res)
        
        param = request.GET
        start_time = param.get('start_time') # 开始时间
        end_time = param.get('end_time') # 结束时间
        if not start_time or not end_time or start_time > end_time:
            res['retcode'] = 500
            res['status'] = '请输入正确的开始时间与结束时间'
            return JsonResponse(res)
        # 获取数据
        alert_messages = gsm.AlertMessage.objects.filter(addtime__gt=start_time, addtime__lt=end_time)
        datas = gss.AlertMessageSeirializer(alert_messages, many=True).data
        if not alert_messages:
            res['retcode'] = 500
            res['status'] = '当前时间段没有数据'
            return JsonResponse(res)
    	# 创建Excel文件对象
        xls_am = Workbook()
        sht1_am = xls_am.active
        # 把字段名写进Excel表格第一行
        sht1_am.append(['序号', '告警id', '设备编号', '线路编号','方位', '告警位置', '经度', '纬度', '具体位置',
                        '告警类型', '告警等级', '开始时间', '结束时间'])
        for data in datas:
            # 按行添加进Excel表格
            sht1_am.append([
                data['id'],
                data['alert_id'],
                data['device_code'],
                data['channel_code'],
                data['channel_name'],
                data['position'],
                data['longitude'],
                data['latitude'],
                data['address'],
                data['alarm_type'],
                data['alarm_level'],
                data['start_time'],
                data['end_time']
            ])
    
        response = HttpResponse(content_type="application/vnd.ms-excel")
        now = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
        response['Content-Disposition'] = 'attachment;filename={0}-{1}.xlsx'.format(urllib.parse.quote("告警数据"), now)
        output = io.BytesIO()  # 创建在内存中处理对象
        xls_am.save(output)
        xls_am.close()
        response.write(output.getvalue())
        return response

## 后记 ##

【后记】为了让大家能够轻松学编程，我创建了一个公众号【轻松学编程】，里面有让你快速学会编程的文章，当然也有一些干货提高你的编程水平，也有一些编程项目适合做一些课程设计等课题。

也可加我微信【1257309054】，拉你进群，大家一起交流学习。  
如果文章对您有帮助，请我喝杯咖啡吧！

公众号

![公众号][20200317124808234.jpg]

关注我，我们一起成长~~

[20200317124808234.jpg]: /images/20220828/995e9ba28d874c699a3e35202acc5f1c.png