up

controller/templateController.js

@@ -77,96 +77,87 @@ async function getXmtzzjllTem({ startYear, endYear, tampName, projectId }) {
     return buffer;
 }
 
-async function getExcelTemplate(req, res, next) {
-    try {
-        let { tampName, projectId, startYear, endYear } = req.query;
-        startYear = startYear || new Date().getFullYear();
-        endYear = endYear || Number(startYear + 33);
-        let keyName = {
-            xmtzzjll: "项目投资资金流量表"
-        }
-        const workbook = new ExcelJS.Workbook();
-        let buffer;
-        if (tampName === 'xmtzzjll') { // 项目投资资金流量表
-            buffer = await getXmtzzjllTem({ startYear, endYear, tampName, projectId });
-        } else if (tampName === 'tjjh') { // 投决计划模板{
-            if (!startYear || !endYear) {
-                errorMessage.paramsError.message = "当前模板开始年份和结束年份不能为空";
-                return res.sendError(errorMessage.paramsError);
-            }
-            // 1. 读取空模板文件
-            const inputFilePath = path.join(__dirname, '../public/templates/投决计划.xlsx');
-            // 2. 处理excel 文件，动态生成表格
-            await workbook.xlsx.readFile(inputFilePath);
-            const worksheet = await workbook.getWorksheet(1);
-            // 2.1 处理表头
-            let count = endYear - startYear;
-            let columns = [];
-            for (let index = 0; index <= count; index++) {
-                columns.push({ name: Number(startYear + index) });
-            }
+async function getTjjh({ startYear, endYear, projectId, tampName }) {
+    const workbook = new ExcelJS.Workbook();
 
-            // 2.2 处理数据
-            let rows = Array(42).fill([]); // 5 是模板有5行
-            if (!projectId) {
-                let tzzjlls = await ProjectTzzjll.findAll({
-                    where: { projectId: projectId, del: 0 },
-                    raw: true,
-                });
-                tzzjlls = _.sortBy(tzzjlls, 'year');
-                rows = thvc(tzzjlls);
-            }
-            console.log(rows, "=============")
+    const inputFilePath = path.join(__dirname, '../public/templates/投决计划.xlsx');
+    // 2. 处理excel 文件，动态生成表格
+    await workbook.xlsx.readFile(inputFilePath);
+    const worksheet = await workbook.getWorksheet(1);
+    // 2.1 处理表头
+    let count = endYear - startYear;
+    let columns = [];
+    for (let index = 0; index <= count; index++) {
+        columns.push({ name: Number(startYear + index) });
+    }
 
-            // 使用addTable创建表格
-            worksheet.addTable({
-                name: 'FinancialData',
-                ref: 'D3',
-                headerRow: true,
-                totalsRow: false,
-                style: {
-                    theme: null,
-                    showRowStripes: false,
-                    showColumnStripes: false,
-                },
-                columns: columns,
-                rows: rows
-            });
+    // 2.2 处理数据
+    let rows = Array(42).fill([]); // 5 是模板有5行
+    if (!projectId) {
+        let tzzjlls = await ProjectTzzjll.findAll({
+            where: { projectId: projectId, del: 0 },
+            raw: true,
+        });
+        tzzjlls = _.sortBy(tzzjlls, 'year');
+        rows = thvc(tzzjlls);
+    }
+    console.log(rows, "=============")
 
-            worksheet.mergeCells(2, 3, 2, (3 + columns.length))
-            worksheet.mergeCells(1, 1, 1, (3 + columns.length))
-            const BLACK_BORDER = {
-                top: { style: 'thin', color: { argb: 'FF000000' } },
-                left: { style: 'thin', color: { argb: 'FF000000' } },
-                bottom: { style: 'thin', color: { argb: 'FF000000' } },
-                right: { style: 'thin', color: { argb: 'FF000000' } }
-            };
+    // 使用addTable创建表格
+    worksheet.addTable({
+        name: 'FinancialData',
+        ref: 'D3',
+        headerRow: true,
+        totalsRow: false,
+        style: {
+            theme: null,
+            showRowStripes: false,
+            showColumnStripes: false,
+        },
+        columns: columns,
+        rows: rows
+    });
 
-            // 遍历所有单元格添加边框
-            worksheet.eachRow((row, rowNumber) => {
-                row.eachCell((cell, colNumber) => {
-                    if (rowNumber >= 4 && rowNumber <= 8 && colNumber >= 4 && colNumber <= (3 + columns.length)) {
-                        cell.numFmt = '#,##0.00'; //数字格式
-                    }
-                    cell.alignment = {
-                        horizontal: 'center',
-                        vertical: 'middle', //垂直居中
-                        wrapText: true
-                    };
-                    cell.border = BLACK_BORDER; //边框
-                });
-            });
-            // 3. 将处理后的文件发送给前端
-            res.setHeader('Content-Type', 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet');
-            res.setHeader('Content-Disposition', 'attachment; filename=' + encodeURIComponent('项目投资资金流量表.xlsx'));
-            const buffer = await workbook.xlsx.writeBuffer();
-            // 发送文件并结束响应
-            res.end(buffer);
-        }
+    worksheet.mergeCells(2, 3, 2, (3 + columns.length))
+    worksheet.mergeCells(1, 1, 1, (3 + columns.length))
+    const BLACK_BORDER = {
+        top: { style: 'thin', color: { argb: 'FF000000' } },
+        left: { style: 'thin', color: { argb: 'FF000000' } },
+        bottom: { style: 'thin', color: { argb: 'FF000000' } },
+        right: { style: 'thin', color: { argb: 'FF000000' } }
+    };
+
+    // 遍历所有单元格添加边框
+    worksheet.eachRow((row, rowNumber) => {
+        row.eachCell((cell, colNumber) => {
+            if (rowNumber >= 4 && rowNumber <= 8 && colNumber >= 4 && colNumber <= (3 + columns.length)) {
+                cell.numFmt = '#,##0.00'; //数字格式
+            }
+            cell.alignment = {
+                horizontal: 'center',
+                vertical: 'middle', //垂直居中
+                wrapText: true
+            };
+            cell.border = BLACK_BORDER; //边框
+        });
+    });
+    const buffer = await workbook.xlsx.writeBuffer();
+    return buffer;
+}
 
-        else {
-            res.status(400).json({ message: '未知的模板名称' });
+async function getExcelTemplate(req, res, next) {
+    try {
+        let { tampName, projectId, startYear, endYear } = req.query;
+        startYear = Number(startYear) || new Date().getFullYear();
+        endYear = endYear || Number(startYear + 33);
+        let keyName = {
+            tjjh: "投决计划",
+            xmtzzjll: "项目投资资金流量表",
+            xmzbjxjll: "项目资本金现金流量表",
+            njfxjll: "能建方现金流量表",
+            lrb: "利润表"
         }
+        let buffer = await getXmtzzjllTem({ startYear, endYear, tampName, projectId, keyName });
         // 3. 将处理后的文件发送给前端
         res.setHeader('Content-Type', 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet');
         res.setHeader('Content-Disposition', 'attachment; filename=' + encodeURIComponent(`${keyName[tampName]}.xlsx`));