前端React后端Django 导出Excel

2021-07-08 10:35:46 浏览数 (1)

代码语言:javascript复制
Dajngo查询数据,查询出来之后生成Excel保存本地


class ExportExcel(APIView):
    def post(self, request, *args, **kwargs):
        export_time = request.data.get('startEndTime')
        user_id = request.data.get('user_id')
        if len(user_id) == 1 and 0 in user_id:
            message_content = StatisticsMessageCount.objects.filter(
            creat_time__range=(export_time[0]   " 00:00:00", export_time[1]   " 23:59:59"))
        elif len(user_id) > 1 and 0 in user_id:
            return JsonResponse({"code": 201, "message": "导出失败,请把全部选项去除!"})
        else:
            message_content = StatisticsMessageCount.objects.filter(user_id__in=user_id,
                                                                    creat_time__range=(export_time[0]   " 00:00:00",
                                                                                       export_time[1]   " 23:59:59"))
        wb = xlwt.Workbook(encoding='utf8')
        sheet = wb.add_sheet('sheet', cell_overwrite_ok=True)
        style_heading = xlwt.easyxf("""
                font:
                    height 220;
                align:
                    vert center,
                    horiz center;
                pattern:
                    pattern solid,
                    fore-colour 0x16;
                borders:
                    left thin, 
                    right thin, 
                    top thin, 
                    bottom thin
                """)
        for i in range(0, len(message_content)):
            sheet.col(i).width = 256 * 30
            sheet.row(i).height = 20 * 80
        sheet.write(0, 0, '序号', style_heading)
        sheet.write(0, 1, '销售', style_heading)
        sheet.write(0, 2, '地区', style_heading)
        sheet.write(0, 3, 'Qustions', style_heading)
        sheet.write(0, 4, 'Answer', style_heading)
        sheet.write(0, 5, '解答人', style_heading)
        data_row = 0
        file_name = None
        for i in message_content:
            # 格式化datetime
            data_row  = 1
            if len(user_id) > 1 or len(user_id) == 1 and 0 in user_id:
                file_name = "KPI统计"
            else:
                file_name = i.answerer
            sheet.write(data_row, 0, data_row)
            sheet.write(data_row, 1, i.sales)
            sheet.write(data_row, 2, i.area, )
            sheet.write(data_row, 3, i.problem)
            sheet.write(data_row, 4, i.answer)
            sheet.write(data_row, 5, i.answerer, )
        try:
            import os
            test_url = "http://127.0.0.1:8081"
            test_path = str('/medias/weekly/{}.xlsx'.format(file_name))
            ret = os.getcwd()
            wb.save(os.getcwd()   pord_path)
            return JsonResponse({"code": 200,
                                 "fileName": "{}.xlsx".format(file_name),
                                 "filePath": pord_url   pord_path })
        except Exception as e:
            print("异常: {}".format(e))
            return JsonResponse({"code": 201, "message": "导出失败,请关闭当前本地电脑打开的相同Excel重新导出!"})
代码语言:javascript复制
前端 react 

   exportExcel = () => {
        const {startEndTime, selectedItems} = this.state
        let currentUser = JSON.parse(localStorage.getItem('userInfo'));
        const {dispatch} = this.props
        if (startEndTime.length === 2) {
            dispatch({
                type: 'GetStaticsCount/exportExcelData',
                payload: {
                    apiPath: '/wx/kpi_export/',
                    user_id:  currentUser.weights > 0? selectedItems:[currentUser.id],
                    startEndTime
                },
                callback: response => {

                    // 这块是关键, 根据后台api返回的文件路径,在本地可以正常a标签下载,在服务器上不可以,直接通过链接打开是文件流形式
                    axios.post(response.filePath, '', {
                        headers: {
                            'Content-Type': 'application/x-www-form-urlencoded', //请求的数据类型为form data格式
                        },
                        'responseType': 'blob'  //设置响应的数据类型为一个包含二进制数据的 Blob 对象,必须设置!!!
                    }).then(function (response) {
                        console.log(`数据流: ${response.data}`)
                        const blob = new Blob([response.data]);
                        const fileName = 'KPI统计.xlsx';
                        const linkNode = document.createElement('a');
                        linkNode.download = fileName; //a标签的download属性规定下载文件的名称
                        linkNode.style.display = 'none';
                        linkNode.href = URL.createObjectURL(blob); //生成一个Blob URL
                        document.body.appendChild(linkNode);
                        linkNode.click();  //模拟在按钮上的一次鼠标单击
                        URL.revokeObjectURL(linkNode.href); // 释放URL 对象
                        document.body.removeChild(linkNode);
                    }).catch(function (error) {
                        console.log(error);
                    });
                }
            })
        } else {
            message.error("请选择导出时间")
        }
    }

0 人点赞