Add support for user block/unblock

Gauvain Pocentek · Gauvain Pocentek · commit e387de528ad2 · 2016-02-04T22:39:21.000+01:00
diff --git a/gitlab/exceptions.py b/gitlab/exceptions.py
@@ -83,6 +83,14 @@ class GitlabBuildRetryError(GitlabOperationError):
     pass
 
 
+class GitlabBlockError(GitlabOperationError):
+    pass
+
+
+class GitlabUnblockError(GitlabOperationError):
+    pass
+
+
 def raise_error_from_response(response, error, expected_code=200):
     """Tries to parse gitlab error message from response and raises error.
 
diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -527,6 +527,20 @@ def Key(self, id=None, **kwargs):
                                            user_id=self.id,
                                            **kwargs)
 
+    def block(self, **kwargs):
+        """Blocks the user."""
+        url = '/users/%s/block' % self.id
+        r = self.gitlab._raw_put(url, **kwargs)
+        raise_error_from_response(r, GitlabBlockError)
+        self.state = 'blocked'
+
+    def unblock(self, **kwargs):
+        """Unblocks the user."""
+        url = '/users/%s/unblock' % self.id
+        r = self.gitlab._raw_put(url, **kwargs)
+        raise_error_from_response(r, GitlabUnblockError)
+        self.state = 'active'
+
 
 class UserManager(BaseManager):
     obj_cls = User
diff --git a/tools/python_test.py b/tools/python_test.py
@@ -40,6 +40,9 @@
 assert(new_user.username == user.username)
 assert(new_user.email == user.email)
 
+new_user.block()
+new_user.unblock()
+
 # SSH keys
 key = new_user.keys.create({'title': 'testkey', 'key': SSH_KEY})
 assert(len(new_user.keys.list()) == 1)
