Add methods to list stages/delete published items

Author: Agathe Guillemot

dataikuapi/dss/apideployer.py

@@ -56,6 +56,15 @@ def create_deployment(self, deployment_id, service_id, infra_id, version):
         self.client._perform_json("POST", "/api-deployer/deployments", body=settings)
         return self.get_deployment(deployment_id)
 
+    def list_stages(self):
+        """
+        Lists infrastructure stages of the API Deployer
+
+        :rtype: a list of dict. Each dict contains a field "id" for the stage identifier and "desc" for its description.
+        :rtype: list
+        """
+        return self.client._perform_json("GET", "/api-deployer/stages")
+
     def list_infras(self, as_objects=True):
         """
         Lists deployment infrastructures on the API Deployer
@@ -332,6 +341,7 @@ def save(self):
                 "PUT", "/api-deployer/deployments/%s/settings" % (self.deployment_id),
                 body = self.settings)
 
+
 class DSSAPIDeployerDeploymentStatus(object):
     """The status of an API Deployer deployment. 
 
@@ -387,7 +397,6 @@ def get_health_messages(self):
         return self.heavy_status["healthMessages"]
 
 
-
 ###############################################
 # Published Service
 ###############################################
@@ -440,6 +449,15 @@ def get_settings(self):
 
         return DSSAPIDeployerServiceSettings(self.client, self.service_id, settings)
 
+    def delete(self):
+        """
+        Deletes this service
+
+        You may only delete a service if it has no deployments on it anymore.
+        """
+        return self.client._perform_empty(
+            "DELETE", "/api-deployer/services/%s" % (self.service_id))
+
 
 class DSSAPIDeployerServiceSettings(object):
     """The settings of an API Deployer Service. 
@@ -466,6 +484,7 @@ def save(self):
                 "PUT", "/api-deployer/services/%s/settings" % (self.service_id),
                 body = self.settings)
 
+
 class DSSAPIDeployerServiceStatus(object):
     """The status of an API Deployer Service. 
 

dataikuapi/dss/projectdeployer.py

@@ -55,6 +55,15 @@ def create_deployment(self, deployment_id, project_key, infra_id, bundle_id):
         self.client._perform_json("POST", "/project-deployer/deployments", body=settings)
         return self.get_deployment(deployment_id)
 
+    def list_stages(self):
+        """
+        Lists infrastructure stages of the Project Deployer
+
+        :rtype: a list of dict. Each dict contains a field "id" for the stage identifier and "desc" for its description.
+        :rtype: list
+        """
+        return self.client._perform_json("GET", "/project-deployer/stages")
+
     def list_infras(self, as_objects=True):
         """
         Lists deployment infrastructures on the Project Deployer
@@ -393,6 +402,14 @@ def get_settings(self):
 
         return DSSProjectDeployerProjectSettings(self.client, self.project_key, settings)
 
+    def delete(self):
+        """
+        Deletes this project
+
+        You may only delete a project if it has no deployments on it anymore.
+        """
+        return self.client._perform_empty(
+            "DELETE", "/project-deployer/projects/%s" % (self.project_key))
 
 class DSSProjectDeployerProjectSettings(object):
     """The settings of a published project.