More helpers

Author: cstenac

dataikuapi/dss/dataset.py

@@ -526,6 +526,16 @@ def get_zone(self):
         """
         return self.project.get_flow().get_zone_of_object(self)
 
+    def move_to_zone(self, zone):
+        """
+        Moves this object to a flow zone
+
+        :param object zone: a :class:`dataikuapi.dss.flow.DSSFlowZone` where to move the object
+        """
+        if isinstance(zone, basestring):
+           zone = self.project.get_flow().get_zone(zone)
+        zone.add_item(self)
+
     def get_usages(self):
         """
         Get the recipes or analyses referencing this dataset

dataikuapi/dss/flow.py

@@ -15,6 +15,19 @@ def get_graph(self):
         data = self.client._perform_json("GET", "/projects/%s/flow/graph/" % (self.project.project_key))
         return DSSProjectFlowGraph(self, data)
 
+    def create_zone(self, name, color="#2ab1ac"):
+        """
+        Creates a new flow zone
+
+        :returns the newly created zone
+        :rtype: :class:`DSSFlowZone`
+        """
+        data = self.client._perform_json("POST", "/projects/%s/flow/zones" % (self.project.project_key), body={
+            "name": name,
+            "color":color
+        })
+        return DSSFlowZone(self, data)
+
     def get_zone(self, id):
         """
         Gets a single Flow zone by id

dataikuapi/dss/managedfolder.py

@@ -167,6 +167,15 @@ def get_zone(self):
         """
         return self.project.get_flow().get_zone_of_object(self)
 
+    def move_to_zone(self, zone):
+        """
+        Moves this object to a flow zone
+
+        :param object zone: a :class:`dataikuapi.dss.flow.DSSFlowZone` where to move the object
+        """
+        if isinstance(zone, basestring):
+           zone = self.project.get_flow().get_zone(zone)
+        zone.add_item(self)
 
     def get_usages(self):
         """

dataikuapi/dss/savedmodel.py

@@ -117,6 +117,16 @@ def get_zone(self):
         """
         return self.project.get_flow().get_zone_of_object(self)
 
+    def move_to_zone(self, zone):
+        """
+        Moves this object to a flow zone
+
+        :param object zone: a :class:`dataikuapi.dss.flow.DSSFlowZone` where to move the object
+        """
+        if isinstance(zone, basestring):
+           zone = self.project.get_flow().get_zone(zone)
+        zone.add_item(self)
+
     def get_usages(self):
         """
         Get the recipes referencing this model