drip-public/iview
6
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve export to CSV functionality

Browse source 
- added`callback`option so we can get CSV data in case we do not want
to download
 - added options:
    - `quoted` for cases when we have line breaks in content
    - `separator` for cases when we have commas in content
This commit is contained in:
Sergio Crisostomo 2017-10-17 10:13:54 +02:00
parent 109465d3c5
commit fb8479988c
2 changed files with 32 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/components/table/table.vue
View file

@ -714,8 +714,9 @@
let noHeader = false;
if ('noHeader' in params) noHeader = params.noHeader;
const data = Csv(columns, datas, ',', noHeader);
ExportCsv.download(params.filename, data);
const data = Csv(columns, datas, params, noHeader);
if (params.callback) params.callback(data);
else ExportCsv.download(params.filename, data);
}
},
created () {

52
src/utils/csv.js
View file

@ -1,25 +1,39 @@
// https://github.com/Terminux/react-csv-downloader/blob/master/src/lib/csv.js
/*
inspired by https://www.npmjs.com/package/react-csv-downloader
now removed from Github
*/
const newLine = '\r\n';
const appendLine = (content, row, { separator, quoted }) => {
const line = row.map(data => {
if (!quoted) return data;
// quote data
data = typeof data === 'string' ? data.replace(/"/g, '"') : data;
return `"${data}"`;
});
content.push(line.join(separator));
};
export default function csv(columns, datas, separator = ',', noHeader = false) {
const defaults = {
separator: ',',
quoted: false
};
export default function csv(columns, datas, options, noHeader = false) {
options = Object.assign({}, defaults, options);
let columnOrder;
const content = [];
const column = [];
if (columns) {
columnOrder = columns.map(v => {
if (typeof v === 'string') {
return v;
}
if (typeof v === 'string') return v;
if (!noHeader) {
column.push((typeof v.title !== 'undefined') ? v.title : v.key);
column.push(typeof v.title !== 'undefined' ? v.title : v.key);
}
return v.key;
});
if (column.length > 0) {
content.push(column.join(separator));
}
if (column.length > 0) appendLine(content, column, options);
} else {
columnOrder = [];
datas.forEach(v => {
@ -29,26 +43,16 @@ export default function csv(columns, datas, separator = ',', noHeader = false) {
});
if (columnOrder.length > 0) {
columnOrder = columnOrder.filter((value, index, self) => self.indexOf(value) === index);
if (!noHeader) {
content.push(columnOrder.join(separator));
}
if (!noHeader) appendLine(content, columnOrder, options);
}
}
if (Array.isArray(datas)) {
datas.map(v => {
if (Array.isArray(v)) {
return v;
datas.forEach(row => {
if (!Array.isArray(row)) {
row = columnOrder.map(k => (typeof row[k] !== 'undefined' ? row[k] : ''));
}
return columnOrder.map(k => {
if (typeof v[k] !== 'undefined') {
return v[k];
}
return '';
});
}).forEach(v => {
content.push(v.join(separator));
appendLine(content, row, options);
});
}
return content.join(newLine);