containermgr: Implement remove_all_nodes() for Kubernetes.

projectatomic · Feb 24, 2017 · 9da6c1a · 9da6c1a
1 parent 4689cb3
commit 9da6c1a
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/commissaire/containermgr/kubernetes/__init__.py b/src/commissaire/containermgr/kubernetes/__init__.py
@@ -243,12 +243,27 @@ def remove_node(self, name):
         resp = self._delete(part)
         if resp.status_code != 200:
             error_msg = (
-                'Unexpected response when trying to remove the node {}.'
+                'Unexpected response when trying to remove the node {}. '
                 'Status: {}, Data: {}'.format(
                     name, resp.status_code, resp.text))
             self.logger.error(error_msg)
             raise ContainerManagerError(error_msg, resp.status_code)
 
+    def remove_all_nodes(self):
+        """
+        Removes all nodes from the Kubernetes Container Manager.
+
+        :raises: commissaire.containermgr.ContainerManagerError
+        """
+        resp = self._delete('/nodes')
+        if resp.status_code != 200:
+            error_msg = (
+                'Unexpected response when trying to remove all nodes. '
+                'Status: {}, Data: {}'.format(
+                    resp.status_code, resp.text))
+            self.logger.error(error_msg)
+            raise ContainerManagerError(error_msg, resp.status_code)
+
     def node_registered(self, name):
         """
         Checks is a node was registered.