更新

src/views/PurchasingCenter/CreateInquirySheet/Export2Excel.js

@@ -1,7 +1,7 @@
 /* eslint-disable */
 require('script-loader!file-saver');
-// import XLSX from 'xlsx'
-import XLSX from 'xlsx-style'
+import XLSX from 'xlsx'
+
 function generateArray(table) {
   var out = [];
   var rows = table.querySelectorAll('tr');
@@ -145,74 +145,70 @@ export function export_table_to_excel(id) {
 }
 
 export function export_json_to_excel({
-  headerGroup,
-  dataGroup,
-  sheetGroup,
+  multiHeader = [],
+  header,
+  data,
   filename,
+  merges = [],
   autoWidth = true,
   bookType=  'xlsx'
 } = {}) {
-  var wb = new Workbook()
-  for (let i = 0; i < dataGroup.length; i++) {
-    /* original data */
-    let data = dataGroup[i]
-    filename = filename || 'excel-list'
-    data = [...data]
-    data.unshift(headerGroup[i]);
-    var ws_name = sheetGroup[i];
-    var ws = sheet_from_array_of_arrays(data);
+  /* original data */
+  filename = filename || 'excel-list'
+  data = [...data]
+  data.unshift(header);
 
-    if (autoWidth) {
-      /*设置worksheet每列的最大宽度*/
-      const colWidth = data.map(row => row.map(val => {
-        /*先判断是否为null/undefined*/
-        if (val == null) {
-          return {
-            'wch': 10
-          };
-        }
-        /*再判断是否为中文*/
-        else if (val.toString().charCodeAt(0) > 255) {
-          return {
-            'wch': val.toString().length * 2
-          };
-        } else {
-          return {
-            'wch': val.toString().length
-          };
-        }
-      }))
-      /*以第一行为初始值*/
-      let result = colWidth[0];
-      for (let i = 1; i < colWidth.length; i++) {
-        for (let j = 0; j < colWidth[i].length; j++) {
-          if (result[j]['wch'] < colWidth[i][j]['wch']) {
-            result[j]['wch'] = colWidth[i][j]['wch'];
-          }
+  for (let i = multiHeader.length-1; i > -1; i--) {
+    data.unshift(multiHeader[i])
+  }
+
+  var ws_name = "SheetJS";
+  var wb = new Workbook(),
+    ws = sheet_from_array_of_arrays(data);
+
+  if (merges.length > 0) {
+    if (!ws['!merges']) ws['!merges'] = [];
+    merges.forEach(item => {
+      ws['!merges'].push(XLSX.utils.decode_range(item))
+    })
+  }
+
+  if (autoWidth) {
+    /*设置worksheet每列的最大宽度*/
+    const colWidth = data.map(row => row.map(val => {
+      /*先判断是否为null/undefined*/
+      if (val == null) {
+        return {
+          'wch': 10
+        };
+      }
+      /*再判断是否为中文*/
+      else if (val.toString().charCodeAt(0) > 255) {
+        return {
+          'wch': val.toString().length * 2
+        };
+      } else {
+        return {
+          'wch': val.toString().length
+        };
+      }
+    }))
+    /*以第一行为初始值*/
+    let result = colWidth[0];
+    for (let i = 1; i < colWidth.length; i++) {
+      for (let j = 0; j < colWidth[i].length; j++) {
+        if (result[j]['wch'] < colWidth[i][j]['wch']) {
+          result[j]['wch'] = colWidth[i][j]['wch'];
         }
       }
-      ws['!cols'] = result;
     }
-    let R = 0;
-    data.map(item => {
-      let C = 0;
-      item.map(itm => {
-        var cell_ref = XLSX.utils.encode_cell({c:C,r:R});
-        var cell = {v: itm }
-        cell.s= {
-          alignment: {
-            horizontal: "center"
-          }
-        }
-        ws[cell_ref] = cell;
-        C++;
-      })
-      R++;
-    })
-    /* add worksheet to workbook */
-    wb.SheetNames.push(ws_name);
-    wb.Sheets[ws_name] = ws;
+    ws['!cols'] = result;
   }
+
+  /* add worksheet to workbook */
+  wb.SheetNames.push(ws_name);
+  wb.Sheets[ws_name] = ws;
+
   var wbout = XLSX.write(wb, {
     bookType: bookType,
     bookSST: false,