Add quick accessors to zone of dataset/managed_folder/saved_model

Author: cstenac

dataikuapi/dss/dataset.py

@@ -515,9 +515,17 @@ def get_metric_history(self, metric, partition=''):
                 params={'metricLookup' : metric if isinstance(metric, str) or isinstance(metric, unicode) else json.dumps(metric)})
 
     ########################################################
-    # Usages
+    # Misc
     ########################################################
 
+    def get_zone(self):
+        """
+        Gets the flow zone of this dataset
+
+        :rtype: :class:`dataikuapi.dss.flow.DSSFlowZone`
+        """
+        return self.project.get_flow().get_zone_of_object(self)
+
     def get_usages(self):
         """
         Get the recipes or analyses referencing this dataset
@@ -527,9 +535,6 @@ def get_usages(self):
         """
         return self.client._perform_json("GET", "/projects/%s/datasets/%s/usages" % (self.project_key, self.dataset_name))
 
-    ########################################################
-    # Discussions
-    ########################################################
     def get_object_discussions(self):
         """
         Get a handle to manage discussions on the dataset

dataikuapi/dss/flow.py

@@ -260,6 +260,7 @@ def get_graph(self):
         data = self.client._perform_json("GET", "/projects/%s/flow/zones/%s/graph" % (self.flow.project.project_key, self.id))
         return DSSProjectFlowGraph(self.flow, data)
 
+
 class DSSFlowZoneSettings(object):
     """The settings of a flow zone. Do not create this directly, use :meth:`DSSFlowZone.get_settings`"""
     def __init__(self, zone):

dataikuapi/dss/managedfolder.py

@@ -12,6 +12,7 @@ class DSSManagedFolder(object):
     """
     def __init__(self, client, project_key, odb_id):
         self.client = client
+        self.project = client.get_project(project_key)
         self.project_key = project_key
         self.odb_id = odb_id
 
@@ -155,9 +156,18 @@ def get_metric_history(self, metric):
 
 
     ########################################################
-    # Usages
+    # Misc
     ########################################################
 
+    def get_zone(self):
+        """
+        Gets the flow zone of this managed folder
+
+        :rtype: :class:`dataikuapi.dss.flow.DSSFlowZone`
+        """
+        return self.project.get_flow().get_zone_of_object(self)
+
+
     def get_usages(self):
         """
         Get the recipes referencing this folder
@@ -167,9 +177,6 @@ def get_usages(self):
         """
         return self.client._perform_json("GET", "/projects/%s/managedfolders/%s/usages" % (self.project_key, self.odb_id))
 
-    ########################################################
-    # Discussions
-    ########################################################
     def get_object_discussions(self):
         """
         Get a handle to manage discussions on the managed folder

dataikuapi/dss/savedmodel.py

@@ -12,6 +12,7 @@ class DSSSavedModel(object):
     """
     def __init__(self, client, project_key, sm_id):
         self.client = client
+        self.project = client.get_project(project_key)
         self.project_key = project_key
         self.sm_id = sm_id
 
@@ -105,9 +106,17 @@ def get_metric_values(self, version_id):
 
 
     ########################################################
-    # Usages
+    # Misc
     ########################################################
 
+    def get_zone(self):
+        """
+        Gets the flow zone of this saved model
+
+        :rtype: :class:`dataikuapi.dss.flow.DSSFlowZone`
+        """
+        return self.project.get_flow().get_zone_of_object(self)
+
     def get_usages(self):
         """
         Get the recipes referencing this model
@@ -117,10 +126,6 @@ def get_usages(self):
         """
         return self.client._perform_json("GET", "/projects/%s/savedmodels/%s/usages" % (self.project_key, self.sm_id))
 
-
-    ########################################################
-    # Discussions
-    ########################################################
     def get_object_discussions(self):
         """
         Get a handle to manage discussions on the saved model