Add support for user block/unblock

2 files changed, 22 insertions, 0 deletions

diff --git a/gitlab/exceptions.py b/gitlab/exceptions.py
index 74e6137..1b5ec6a 100644
--- 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
index f8c102b..a781cb1 100644
--- a/gitlab/objects.py
+++ b/gitlab/objects.py
@@ -527,6 +527,20 @@ class User(GitlabObject):
                                            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