Add download featureset functionality

Improve downloaded featureset formatting

Author: acrellin

cesium_app/app_server.py

@@ -56,7 +56,8 @@ def make_app(cfg, baselayer_handlers, baselayer_settings):
     handlers = baselayer_handlers + [
         (r'/project(/.*)?', ProjectHandler),
         (r'/dataset(/.*)?', DatasetHandler),
-        (r'/features(/.*)?', FeatureHandler),
+        (r'/features(/[0-9]+)?', FeatureHandler),
+        (r'/features/([0-9]+)/(download)', FeatureHandler),
         (r'/models(/[0-9]+)?', ModelHandler),
         (r'/models/([0-9]+)/(download)', ModelHandler),
         (r'/predictions(/[0-9]+)?', PredictionHandler),

cesium_app/handlers/feature.py

@@ -13,20 +13,34 @@
 from os.path import join as pjoin
 import uuid
 import datetime
+import pandas as pd
 
 
 class FeatureHandler(BaseHandler):
     @auth_or_token
-    def get(self, featureset_id=None):
-        if featureset_id is not None:
-            featureset_info = Featureset.get_if_owned_by(featureset_id,
-                                                         self.current_user)
+    def get(self, featureset_id=None, action=None):
+        if action == 'download':
+            fset_path = Featureset.get_if_owned_by(featureset_id,
+                                                   self.current_user).file_uri
+            print(fset_path)
+            fset, data = featurize.load_featureset(fset_path)
+            fset.index.name = 'ts_name'
+            fset.columns = fset.columns.get_level_values(0)
+            fset.columns.name = None
+            self.set_header("Content-Type", 'text/csv; charset="utf-8"')
+            self.set_header("Content-Disposition", "attachment; "
+                            "filename=cesium_featureset.csv")
+            self.write(fset.to_csv(index=True))
         else:
-            featureset_info = [f for p in self.current_user.projects
-                               for f in p.featuresets]
-            featureset_info.sort(key=lambda f: f.created_at, reverse=True)
-
-        self.success(featureset_info)
+            if featureset_id is not None:
+                featureset_info = Featureset.get_if_owned_by(featureset_id,
+                                                             self.current_user)
+            else:
+                featureset_info = [f for p in self.current_user.projects
+                                   for f in p.featuresets]
+                featureset_info.sort(key=lambda f: f.created_at, reverse=True)
+
+            self.success(featureset_info)
 
     @auth_or_token
     async def _await_featurization(self, future, fset):

static/js/components/Features.jsx

@@ -13,6 +13,7 @@ import Plot from './Plot';
 import FoldableRow from './FoldableRow';
 import { reformatDatetime, contains } from '../utils';
 import Delete from './Delete';
+import Download from './Download';
 
 const { Tab, Tabs, TabList, TabPanel } = { ...ReactTabs };
 
@@ -257,7 +258,11 @@ export let FeatureTable = props => (
                 <td>{featureset.name}</td>
                 <td>{reformatDatetime(featureset.created_at)}</td>
                 {status}
-                <td><DeleteFeatureset ID={featureset.id} /></td>
+                <td>
+                  <Download url={`/features/${featureset.id}/download`} />
+                  &nbsp;&nbsp;
+                  <DeleteFeatureset ID={featureset.id} />
+                </td>
               </tr>
               {foldedContent}
             </FoldableRow>
@@ -290,7 +295,6 @@ FeatureTable = connect(ftMapStateToProps)(FeatureTable);
 const mapDispatchToProps = dispatch => (
   { delete: id => dispatch(Action.deleteFeatureset(id)) }
 );
-
 const DeleteFeatureset = connect(null, mapDispatchToProps)(Delete);
 
 export default FeaturesTab;