Add methods to delete infra/node on infra

Author: Agathe Guillemot

dataikuapi/dss/apideployer.py

@@ -166,6 +166,15 @@ def get_settings(self):
 
         return DSSAPIDeployerInfraSettings(self.client, self.infra_id, settings)
 
+    def delete(self):
+        """
+        Deletes this infra
+        You may only delete an infra if it has no deployments on it anymore.
+        """
+        return self.client._perform_empty(
+            "DELETE", "/api-deployer/infras/%s" % (self.infra_id))
+
+
 class DSSAPIDeployerInfraSettings(object):
     """The settings of an API Deployer Infra. 
 
@@ -191,6 +200,18 @@ def add_apinode(self, url, api_key, graphite_prefix=None):
         }
         self.settings["apiNodes"].append(new_node)
 
+    def remove_apinode(self, node_url):
+        """
+        Removes a node from the list of nodes of this infra.
+        Only applicable to STATIC infrastructures
+
+        :param str node_url: URL of the node to remove
+        """
+        for node in self.settings["apiNodes"]:
+            if node["url"] == node_url:
+                self.settings["apiNodes"].remove(node)
+                break
+
     def get_raw(self):
         """
         Gets the raw settings of this infra. This returns a reference to the raw settings, not a copy,