Implement swarm node removalremove_node

Signed-off-by: Joffrey F <joffrey@docker.com>

2 files changed, 47 insertions, 0 deletions

diff --git a/docker/api/swarm.py b/docker/api/swarm.py
index 6a1b752..9a240a9 100644
--- a/docker/api/swarm.py
+++ b/docker/api/swarm.py
@@ -224,6 +224,33 @@ class SwarmApiMixin(object):
 
         return self._result(self._get(url, params=params), True)
 
+    @utils.check_resource
+    @utils.minimum_version('1.24')
+    def remove_node(self, node_id, force=False):
+        """
+        Remove a node from the swarm.
+
+        Args:
+            node_id (string): ID of the node to be removed.
+            force (bool): Force remove an active node. Default: `False`
+
+        Raises:
+            :py:class:`docker.errors.NotFound`
+                If the node referenced doesn't exist in the swarm.
+
+            :py:class:`docker.errors.APIError`
+                If the server returns an error.
+        Returns:
+            `True` if the request was successful.
+        """
+        url = self._url('/nodes/{0}', node_id)
+        params = {
+            'force': force
+        }
+        res = self._delete(url, params=params)
+        self._raise_for_status(res)
+        return True
+
     @utils.minimum_version('1.24')
     def update_node(self, node_id, version, node_spec=None):
         """
@@ -231,6 +258,7 @@ class SwarmApiMixin(object):
 
         Args:
 
+            node_id (string): ID of the node to be updated.
             version (int): The version number of the node object being
                 updated. This is required to avoid conflicting writes.
             node_spec (dict): Configuration settings to update. Any values
diff --git a/docker/models/nodes.py b/docker/models/nodes.py
index 0887f99..8dd9350 100644
--- a/docker/models/nodes.py
+++ b/docker/models/nodes.py
@@ -41,6 +41,25 @@ class Node(Model):
         """
         return self.client.api.update_node(self.id, self.version, node_spec)
 
+    def remove(self, force=False):
+        """
+        Remove this node from the swarm.
+
+        Args:
+            force (bool): Force remove an active node. Default: `False`
+
+        Returns:
+            `True` if the request was successful.
+
+        Raises:
+            :py:class:`docker.errors.NotFound`
+                If the node doesn't exist in the swarm.
+
+            :py:class:`docker.errors.APIError`
+                If the server returns an error.
+        """
+        return self.client.api.remove_node(self.id, force=force)
+
 
 class NodeCollection(Collection):
     """Nodes on the Docker server."""