add api for continuous activities

Author: FChataigner

dataikuapi/dss/continuousactivity.py

@@ -0,0 +1,43 @@
+import time
+import sys
+from dataikuapi.utils import DataikuException
+
+class DSSContinuousActivity(object):
+    """
+    A continuous activity on the DSS instance
+    """
+    def __init__(self, client, project_key, recipe_id):
+        self.client = client
+        self.recipe_id = recipe_id
+        self.project_key = project_key
+
+    def start(self, loop_params={}):
+        """
+        Start the activity
+        """
+        return self.client._perform_json(
+            "POST", "/projects/%s/continuous-activities/%s/start" % (self.project_key, self.recipe_id), body=loop_params)
+
+    def stop(self):
+        """
+        Stop the activity
+        """
+        self.client._perform_empty(
+            "POST", "/projects/%s/continuous-activities/%s/stop" % (self.project_key, self.recipe_id))
+
+    def get_status(self):
+        """
+        Get the current status of the continuous activity
+        
+        Returns:
+            the state of the continuous activity, as a JSON object
+        """
+        return self.client._perform_json(
+            "GET", "/projects/%s/continuous-activities/%s/" % (self.project_key, self.recipe_id))
+
+    def get_recipe(self):
+    	"""
+    	Return a handle on the associated recipe
+    	"""
+    	from .recipe import DSSRecipe
+    	return DSSRecipe(self.client, self.project_key, self.recipe_id)

dataikuapi/dss/project.py

@@ -6,6 +6,7 @@
 from .managedfolder import DSSManagedFolder
 from .savedmodel import DSSSavedModel
 from .job import DSSJob, DSSJobWaiter
+from .continuousactivity import DSSContinuousActivity
 from .scenario import DSSScenario
 from .apiservice import DSSAPIService
 from .future import DSSFuture
@@ -727,6 +728,32 @@ def new_job_definition_builder(self, job_type='NON_RECURSIVE_FORCED_BUILD'):
         warnings.warn("new_job_definition_builder is deprecated, please use new_job", DeprecationWarning)
         return JobDefinitionBuilder(self, job_type)
 
+    ########################################################
+    # Continuous activities
+    ########################################################
+
+    def list_continuous_activities(self, as_objects=True):
+        """
+        List the continuous activities in this project
+        
+        Returns:
+            a list of the continuous activities, each one as a JSON object, containing both the definition and the state
+        """
+        list = self.client._perform_json("GET", "/projects/%s/continuous-activities/" % self.project_key)
+        if as_objects:
+            return [DSSContinuousActivity(self.client, a['projectKey'], a['recipeId']) for a in list]
+        else:
+            return list
+
+    def get_continuous_activity(self, recipe_id):
+        """
+        Get a handler to interact with a specific continuous activities
+        
+        Returns:
+            A :class:`dataikuapi.dss.continuousactivity.DSSContinuousActivity` job handle
+        """
+        return DSSContinuousActivity(self.client, self.project_key, recipe_id)
+
     ########################################################
     # Variables
     ########################################################

dataikuapi/dss/recipe.py

@@ -190,6 +190,12 @@ def get_object_discussions(self):
         """
         return DSSObjectDiscussions(self.client, self.project_key, "RECIPE", self.recipe_name)
 
+    def get_continuous_activity(self):
+        """
+        Return a handle on the associated recipe
+        """
+        from .continuousactivity import DSSContinuousActivity
+        return DSSContinuousActivity(self.client, self.project_key, self.recipe_name)
 
 class DSSRecipeStatus(object):
     """Status of a recipce.