public api for notebooks

Author: FChataigner

dataikuapi/dss/project.py

@@ -8,6 +8,7 @@
 import sys
 import os.path as osp
 from .future import DSSFuture
+from .notebook import DSSNotebook
 
 class DSSProject(object):
     """
@@ -493,4 +494,20 @@ def sync_datasets_acls(self):
             "POST", "/projects/%s/actions/sync" % (self.project_key))
         return DSSFuture(self.client, future_response.get('jobId', None), future_response)
 
-        
+    ########################################################
+    # Notebooks
+    ########################################################
+            
+    def list_running_notebooks(self, as_objects=False):
+        """
+        List the currently-running notebooks
+
+        Returns:
+            list of notebooks. Each object contains at least a 'name' field
+        """
+        list = self.client._perform_json("GET", "/projects/%s/notebooks/active" % self.project_key)
+        if as_objects:
+            return [DSSNotebook(self.client, notebook['projectKey'], notebook['name'], notebook) for notebook in list]
+        else:
+            return list
+

dataikuapi/dssclient.py

@@ -9,6 +9,7 @@
 from dss.admin import DSSUser, DSSGroup, DSSConnection, DSSGeneralSettings
 from dss.meaning import DSSMeaning
 from dss.sqlquery import DSSSQLQuery
+from dss.notebook import DSSNotebook
 import os.path as osp
 from .utils import DataikuException
 
@@ -76,6 +77,24 @@ def get_future(self, job_id):
         return DSSFuture(self, job_id)
 
 
+    ########################################################
+    # Notebooks
+    ########################################################
+            
+    def list_running_notebooks(self, as_objects=False):
+        """
+        List the currently-running notebooks
+
+        Returns:
+            list of notebooks. Each object contains at least a 'name' field
+        """
+        list = self._perform_json("GET", "/admin/notebooks/")
+        if as_objects:
+            return [DSSNotebook(self, notebook['projectKey'], notebook['name'], notebook) for notebook in list]
+        else:
+            return list
+
+
     ########################################################
     # Projects
     ########################################################