|
|
@ -7,76 +7,76 @@ const moment = require('moment') |
|
|
|
|
|
|
|
|
module.exports = function (app, opts) { |
|
|
module.exports = function (app, opts) { |
|
|
|
|
|
|
|
|
//递归创建目录 同步方法
|
|
|
//递归创建目录 同步方法
|
|
|
async function makeDir (dir) { |
|
|
async function makeDir(dir) { |
|
|
if (!fs.existsSync(dir)) { |
|
|
if (!fs.existsSync(dir)) { |
|
|
makeDir(path.dirname(dir)) |
|
|
makeDir(path.dirname(dir)) |
|
|
fs.mkdirSync(dir, function (err) { |
|
|
fs.mkdirSync(dir, function (err) { |
|
|
if (err) { |
|
|
if (err) { |
|
|
throw err |
|
|
throw err |
|
|
} |
|
|
} |
|
|
}); |
|
|
}); |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
async function simpleExcelDown ({ data = [], header = [], fileName = moment().format('YYYY-MM-DD HH:mm:ss') } = {}) { |
|
|
async function simpleExcelDown({ data = [], header = [], fileName = moment().format('YYYY-MM-DD HH:mm:ss') } = {}) { |
|
|
const fileDirPath = path.join(__dirname, `../../downloadFiles`) |
|
|
const fileDirPath = path.join(__dirname, `../../downloadFiles`) |
|
|
makeDir(fileDirPath) |
|
|
makeDir(fileDirPath) |
|
|
const file = new xlsx.File(); |
|
|
const file = new xlsx.File(); |
|
|
const sheet_1 = file.addSheet('sheet_1'); |
|
|
const sheet_1 = file.addSheet('sheet_1'); |
|
|
|
|
|
|
|
|
// header
|
|
|
// header
|
|
|
const headerStyle = new xlsx.Style(); |
|
|
const headerStyle = new xlsx.Style(); |
|
|
headerStyle.align.h = 'center'; |
|
|
headerStyle.align.h = 'center'; |
|
|
headerStyle.align.v = 'center'; |
|
|
headerStyle.align.v = 'center'; |
|
|
headerStyle.border.right = 'thin'; |
|
|
headerStyle.border.right = 'thin'; |
|
|
headerStyle.border.rightColor = '#000000'; |
|
|
headerStyle.border.rightColor = '#000000'; |
|
|
headerStyle.border.bottom = 'thin'; |
|
|
headerStyle.border.bottom = 'thin'; |
|
|
headerStyle.border.bottomColor = '#000000'; |
|
|
headerStyle.border.bottomColor = '#000000'; |
|
|
|
|
|
|
|
|
const headerRow = sheet_1.addRow(); |
|
|
const headerRow = sheet_1.addRow(); |
|
|
const indexCell = headerRow.addCell(); |
|
|
const indexCell = headerRow.addCell(); |
|
|
indexCell.value = '序号' |
|
|
indexCell.value = '序号' |
|
|
indexCell.style = headerStyle |
|
|
indexCell.style = headerStyle |
|
|
for (let h of header) { |
|
|
for (let h of header) { |
|
|
const cell = headerRow.addCell(); |
|
|
const cell = headerRow.addCell(); |
|
|
cell.value = h.title; |
|
|
cell.value = h.title; |
|
|
cell.style = headerStyle |
|
|
cell.style = headerStyle |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// data
|
|
|
// data
|
|
|
const style = new xlsx.Style(); |
|
|
const style = new xlsx.Style(); |
|
|
style.align.h = 'left'; |
|
|
style.align.h = 'left'; |
|
|
style.align.v = 'center'; |
|
|
style.align.v = 'center'; |
|
|
style.border.right = 'thin'; |
|
|
style.border.right = 'thin'; |
|
|
style.border.rightColor = '#000000'; |
|
|
style.border.rightColor = '#000000'; |
|
|
style.border.bottom = 'thin'; |
|
|
style.border.bottom = 'thin'; |
|
|
style.border.bottomColor = '#000000'; |
|
|
style.border.bottomColor = '#000000'; |
|
|
for (let i = 0; i < data.length; i++) { |
|
|
for (let i = 0; i < data.length; i++) { |
|
|
const row = sheet_1.addRow(); |
|
|
const row = sheet_1.addRow(); |
|
|
const indexCell = row.addCell(); |
|
|
const indexCell = row.addCell(); |
|
|
indexCell.value = i + 1 |
|
|
indexCell.value = i + 1 |
|
|
indexCell.style = headerStyle |
|
|
indexCell.style = headerStyle |
|
|
for (let h of header) { |
|
|
for (let h of header) { |
|
|
const cell = row.addCell(); |
|
|
const cell = row.addCell(); |
|
|
cell.value = data[i][h.key] || h.defaultValue || ''; |
|
|
cell.value = data[i][h.key] || h.defaultValue || '-'; |
|
|
cell.style = style |
|
|
cell.style = style |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
const savePath = path.join(fileDirPath, fileName) |
|
|
const savePath = path.join(fileDirPath, fileName) |
|
|
await new Promise(function (resolve, reject) { |
|
|
await new Promise(function (resolve, reject) { |
|
|
file.saveAs() |
|
|
file.saveAs() |
|
|
.pipe(fs.createWriteStream(savePath)) |
|
|
.pipe(fs.createWriteStream(savePath)) |
|
|
.on('finish', () => { |
|
|
.on('finish', () => { |
|
|
resolve() |
|
|
resolve() |
|
|
}); |
|
|
}); |
|
|
}) |
|
|
}) |
|
|
return savePath |
|
|
return savePath |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
return { |
|
|
return { |
|
|
simpleExcelDown, |
|
|
simpleExcelDown, |
|
|
makeDir |
|
|
makeDir |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
