需求：electron环境下想要实现根据多个表生成多个Excel文件，打包存入文件夹内并压缩下载到本地。（实际场景描述：界面中有软件工程一班学生信息、软件工程二班学生信息、软件工程三班学生信息，上方有“一键生成”的按钮，点击时弹出文件存储位置选择框选择压缩包所要保存的位置，选择完成后点击保存后生成压缩包中存放一、二、三班三张对应班级学生信息Excel的文件。）注：生成的Excel文件需要按照对应模板进行显示（合并单元格、修改excel单元格的高度和宽度...）
思路：使用electron中的dialog选择保存文件的位置，判断所选地址是否存在；如果存在使用XLSX、XLSXStyle进行文件的生成；然后使用adm-zip将所有文件存放在zip文件夹中保存到本地。
1.下载xlsx xlsx-style adm-zip
npm i xlsx
npm i xlsx-style
npm i adm-zip
或
yarn add xlsx
yarn add xlsx-style
yarn add adm-zip
注意：使用xlsx-style插件的时候需要简单修改下源码，第一种情况下会出现报错，第二种是为了可以调整excel单元格的行高。
1）修改node_modules文件夹下的xlsx-style下的dist下的cpexcel.js文件807行：var cpt = cptable;
2）修改xlsx-style文件夹下面的xlsx.js文件 替换write_ws_xml_data以下方法
// xlsx-style版本0.8.13
// xlsx版本0.14.1
//这是xlsx-style文件中的xlsx.js的需要修改的代码，是从xlsx文件夹中的xlsx.js中复制出来的
// write_ws_xml_data找到找个方法名字，全部替换
// 把xlsx中能修改高度的代码复制到xlsx-style中
var DEF_PPI = 96, PPI = DEF_PPI;
function px2pt(px) { return px * 96 / PPI; }
function pt2px(pt) { return pt * PPI / 96; }
function write_ws_xml_data(ws, opts, idx, wb) {
var o = [], r = [], range = safe_decode_range(ws['!ref']), cell="", ref, rr = "", cols = [], R=0, C=0, rows = ws['!rows'];
var dense = Array.isArray(ws);
var params = ({r:rr}), row, height = -1;
for(C = range.s.c; C <= range.e.c; ++C) cols[C] = encode_col(C);
for(R = range.s.r; R <= range.e.r; ++R) {
r = [];
rr = encode_row(R);
for(C = range.s.c; C <= range.e.c; ++C) {
ref = cols[C] + rr;
var _cell = dense ? (ws[R]||[])[C]: ws[ref];
if(_cell === undefined) continue;
if((cell = write_ws_xml_cell(_cell, ref, ws, opts, idx, wb)) != null) r.push(cell);
}
if(r.length > 0 || (rows && rows[R])) {
params = ({r:rr});
if(rows && rows[R]) {
row = rows[R];
if(row.hidden) params.hidden = 1;
height = -1;
if (row.hpx) height = px2pt(row.hpx);
else if (row.hpt) height = row.hpt;
if (height > -1) { params.ht = height; params.customHeight = 1; }
if (row.level) { params.outlineLevel = row.level; }
}
o[o.length] = (writextag('row', r.join(""), params));
}
}
if(rows) for(; R < rows.length; ++R) {
if(rows && rows[R]) {
params = ({r:R+1});
row = rows[R];
if(row.hidden) params.hidden = 1;
height = -1;
if (row.hpx) height = px2pt(row.hpx);
else if (row.hpt) height = row.hpt;
if (height > -1) { params.ht = height; params.customHeight = 1; }
if (row.level) { params.outlineLevel = row.level; }
o[o.length] = (writextag('row', "", params));
}
}
return o.join("");
}
2.可以在utils下新建一个js文件用于写入xlsx以及xlsx-style相关方法
// 导出为excel
import XLSX from 'xlsx'
import XLSXStyle from 'xlsx-style'
export function exportExcel(filename, data) {
var sheet_name = 'Sheet1'
var work_book = XLSX.utils.book_new()
var sheet = XLSX.utils.aoa_to_sheet(data)
XLSX.utils.book_append_sheet(work_book, sheet, sheet_name)
XLSXStyle.writeFile(work_book, filename) //导出Excel
}
// 将workbook装化成ArrayBuffer对象
export function workbook2ArrayBuffer(workbook) {
var wopts = {
bookType: 'xlsx',
bookSST: false,
type: 'binary',
}
var wbout = XLSXStyle.write(workbook, wopts)
// 将字符串转ArrayBuffer
function s2ab(s) {
var buf = new ArrayBuffer(s.length)
var view = new Uint8Array(buf)
for (var i = 0; i != s.length; ++i) view[i] = s.charCodeAt(i) & 0xff
return buf
}
return s2ab(wbout)
}
3.在要进行一键生成的界面进行相关逻辑的编写。
<button @click='ClickGenerate'>一键生成</button>
...
...
methods: {
// 一键生成按钮方法
ClickGenerate(){
const fs = require('fs')
let zipName = '软件工程学生信息.zip'
// 弹出dialog选择保存文件的位置
const { dialog } = require("electron").remote
let options = {
title: "软件工程学生信息", //下载文件名称
defaultPath: zipName //下载文件title
}
dialog.showSaveDialog(options, (result) => {
if (!result) {
this.$message.warning('下载任务已取消')
return
}
let path = getDirectory(result)
if (!fs.existsSync(path)) {
this.$message.warning('所选路径不存在！')
return
}
// 准备下载zip文件
this.downloadZip(result)
})
},
// 导出涉密人员zip
async downloadZip(savePath) {
// 准备生成zip
let AdmZip = require("adm-zip")
let zip = new AdmZip()
zip.writeZip(savePath)
//#region
// 生成软件工程一班学生信息并放入zip包中
let soft1buffer = this.creatSoftbuffer('1')
if (soft1buffer) {
zip.addFile('软件工程一班学生信息' + ".xlsx", soft1buffer, "")
}
// #endregion
zip.writeZip(savePath, (error) => {
if (error) {
this.$message.error('[下载异常]' + error.message)
return
}
this.$message.success('下载已完成')
})
},
// 生成excel文件流
creatSoftbuffer(name) {
let list = []
let date = new Date()
let Y = date.getFullYear() + '-';
let M = (date.getMonth() + 1 < 10 ? '0' + (date.getMonth() + 1) : date.getMonth() + 1) + '-';
let D = date.getDate() + ' ';
let dqsj = Y + M + D // 获取当前时间
let soft1Datas = [
{ id: 1, name: "张三", age: '24', address: '哈尔滨市南岗区', grade: '软件一班' },
{ id: 2, name: "里斯", age: '22', address: '哈尔滨市呼兰区', grade: '软件一班' },
{ id: 3, name: "王二", age: '21', address: '哈尔滨市松北区', grade: '软件一班' },
...
] // 此处数据可从后台获取
switch (name) {
case '1':
list.push(["软件工程一班学生信息"]) //确定列名
list.push(["统计时间：", dqsj, "", "", ""]) //确定列名
list.push(["序号", "姓名", "年龄", "地址", "年级"]) //确定列名
let lock1 = 0
for (let i in soft1Datas) { //每一行的值
lock1 = lock1 + 1
let item = soft1Datas[i]
let column = [lock1, item["name"], item["age"], item["address"], item["grade"]]
list.push(column)
}
//Excel sheet页的名称
var sheet_name = "软件工程一班学生信息"
var sum = 5 // 列的数量，根据自身项目进行数据的更改
break
default:
console.log(0)
}
var sheet = XLSX.utils.aoa_to_sheet(list)
// 设置列宽（这里用到列的数量是用来设置不同列的不同宽度的）
let counts = soft1Datas.length + 5
sheet['!cols'] = []
sheet['!rows'] = []
for (let i = 1; i < sum; i++) {
sheet['!cols'].push({ wpx: 150 }) // 设置列宽
}
for (let j = 1; j < counts; j++) {
sheet['!rows'].push({ hpx: 14 }) // 设置列高
}
sheet['!rows'][0].hpx = 35
sheet['!rows'][1].hpx = 20
sheet['!rows'][2].hpx = 20
sheet['!rows'][3].hpx = 20
// 所有设置边框字体水平居中等样式
for (let key in sheet) {
if (sheet[key] instanceof Object) {
sheet[key].s = {
alignment: {
horizontal: 'center', // 水平居中
vertical: 'center' // 垂直居中
},
font: {
sz: 11, // 字号
name: '宋体' // 字体
},
border: { // 边框
top: {
style: 'thin'
},
bottom: {
style: 'thin'
},
left: {
style: 'thin'
},
right: {
style: 'thin'
}
}
}
}
}
// 标题样式修改
sheet.A1.s = {
font: {
name: '宋体',
sz: 16, // 字号
bold: true,
},
alignment: {
horizontal: 'center', // 水平居中
vertical: 'center' // 垂直居中
}
}
sheet['!merges'] = [
{ s: { r: 0, c: 0 }, e: { r: 0, c: 5 } }
];
//新建book
var work_book = XLSX.utils.book_new()
//将数据添加到工作薄
XLSX.utils.book_append_sheet(work_book, sheet, sheet_name)
const workbookArrayBuffer = workbook2ArrayBuffer(work_book)
//arrayBuffer转Buffer
var buf = new Buffer(workbookArrayBuffer.byteLength)
var view = new Uint8Array(workbookArrayBuffer)
for (var i = 0; i < buf.length; ++i) {
buf[i] = view[i]
}
return buf
}
}