Merge pull request #483 from unitaryfund/459_export_task_to_csv

#459: Export task to csv
unitaryfund · Mar 8, 2022 · b296e35 · b296e35
2 parents dc46052 + 2db1947
commit b296e35
Show file tree

Hide file tree

Showing 3 changed files with 100 additions and 12 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -15,6 +15,7 @@
     "chart.js": "^3.7.1",
     "chartjs-adapter-moment": "^1.0.0",
     "chartjs-plugin-datalabels": "^2.0.0",
+    "json2csv": "^5.0.7",
     "moment": "^2.29.1",
     "prop-types": "^15.7.2",
     "rc-table": "^7.17.1",

diff --git a/src/views/Task.js b/src/views/Task.js
@@ -15,6 +15,7 @@ import { FontAwesomeIcon } from '@fortawesome/react-fontawesome'
 import SotaChart from '../components/SotaChart'
 import { FacebookShareButton, TwitterShareButton, FacebookIcon, TwitterIcon } from 'react-share'
 import moment from 'moment'
+import { parse } from 'json2csv'
 
 library.add(faEdit)
 
@@ -29,6 +30,7 @@ class Task extends React.Component {
       item: { submissions: [], childTasks: [], parentTask: {} },
       allTaskNames: [],
       results: [],
+      resultsJson: [],
       chartData: {},
       chartKey: '',
       metricNames: [],
@@ -41,6 +43,7 @@ class Task extends React.Component {
     this.handleOnChange = this.handleOnChange.bind(this)
     this.handleTrimTasks = this.handleTrimTasks.bind(this)
     this.sliceChartData = this.sliceChartData.bind(this)
+    this.handleCsvExport = this.handleCsvExport.bind(this)
   }
 
   handleShowEditModal () {
@@ -150,7 +153,17 @@ class Task extends React.Component {
             return 0
           }
         })
-        this.setState({ results: results })
+        const resultsJson = results.map(row =>
+          ({
+            key: row.id,
+            submissionId: this.state.item.submissions.find(e => e.name === row.submissionName).id,
+            name: row.submissionName,
+            methodName: row.methodName,
+            metricName: row.metricName,
+            metricValue: row.metricValue,
+            tableDate: row.evaluatedAt ? new Date(row.evaluatedAt).toLocaleDateString() : new Date(row.createdAt).toLocaleDateString()
+          }))
+        this.setState({ results: results, resultsJson: resultsJson })
         this.sliceChartData(results)
       })
       .catch(err => {
@@ -225,6 +238,23 @@ class Task extends React.Component {
     this.setState({ metricNames: metricNames, chartKey: chartKey, chartData: chartData, isLowerBetterDict: isLowerBetterDict })
   }
 
+  handleCsvExport () {
+    const fields = Object.keys(this.state.resultsJson[0])
+    const opts = { fields }
+    const csv = parse(this.state.resultsJson, opts)
+
+    const element = document.createElement('a')
+    element.setAttribute('href', 'data:text/plain;charset=utf-8,' + encodeURIComponent(csv))
+    element.setAttribute('download', this.state.item.name)
+
+    element.style.display = 'none'
+    document.body.appendChild(element)
+
+    element.click()
+
+    document.body.removeChild(element)
+  }
+
   render () {
     return (
       <div id='metriq-main-content'>
@@ -357,7 +387,7 @@ class Task extends React.Component {
               </div>
               <br />
               {(this.state.results.length > 0) &&
-                <h2>Results</h2>}
+                <h2>Results <button className='btn btn-primary' onClick={this.handleCsvExport}>Export to CSV</button></h2>}
               {(this.state.results.length > 0) &&
                 <div className='row'>
                   <div className='col-md-12'>
@@ -393,16 +423,7 @@ class Task extends React.Component {
                         key: 'metricValue',
                         width: 300
                       }]}
-                      data={this.state.results.map(row =>
-                        ({
-                          key: row.id,
-                          submissionId: this.state.item.submissions.find(e => e.name === row.submissionName).id,
-                          name: row.submissionName,
-                          methodName: row.methodName,
-                          metricName: row.metricName,
-                          metricValue: row.metricValue,
-                          tableDate: row.evaluatedAt ? new Date(row.evaluatedAt).toLocaleDateString() : new Date(row.createdAt).toLocaleDateString()
-                        }))}
+                      data={this.state.resultsJson}
                       onRow={(record) => ({
                         onClick () { window.location.href = '/Submission/' + record.submissionId }
                       })}