support csv formatter

AllenFang · May 7, 2016 · acd12b1 · acd12b1
1 parent 636922b
commit acd12b1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/src/BootstrapTable.js b/src/BootstrapTable.js
@@ -622,7 +622,10 @@ class BootstrapTable extends Component {
     const keys = [];
     this.props.children.map(function(column) {
       if (column.props.hidden === false) {
-        keys.push(column.props.dataField);
+        keys.push({
+          field: column.props.dataField,
+          format: column.props.csvFormat
+        });
       }
     });
 

diff --git a/src/TableHeaderColumn.js b/src/TableHeaderColumn.js
@@ -120,6 +120,7 @@ TableHeaderColumn.propTypes = {
   dataSort: PropTypes.bool,
   onSort: PropTypes.func,
   dataFormat: PropTypes.func,
+  csvFormat: PropTypes.func,
   isKey: PropTypes.bool,
   editable: PropTypes.any,
   hidden: PropTypes.bool,
@@ -154,6 +155,7 @@ TableHeaderColumn.defaultProps = {
   dataAlign: 'left',
   dataSort: false,
   dataFormat: undefined,
+  csvFormat: undefined,
   isKey: false,
   editable: true,
   onSort: undefined,

diff --git a/src/csv_export_util.js b/src/csv_export_util.js
@@ -13,11 +13,13 @@ function toString(data, keys) {
   let dataString = '';
   if (data.length === 0) return dataString;
 
-  dataString += keys.join(',') + '\n';
+  dataString += keys.map(x => x.field).join(',') + '\n';
 
   data.map(function(row) {
     keys.map(function(col, i) {
-      const cell = typeof row[col] !== 'undefined' ? ('"' + row[col] + '"') : '';
+      const { field, format } = col;
+      const value = typeof format !== 'undefined' ? format(row[field]) : row[field];
+      const cell = typeof value !== 'undefined' ? ('"' + value + '"') : '';
       dataString += cell;
       if (i + 1 < keys.length) dataString += ',';
     });