Merge pull request #118 from IvicaArsov/issues_subscribe_unsubscribe

Add support for subscribe and unsubscribe in issues
python-gitlab · May 28, 2016 · 7bbbfbd · 7bbbfbd
2 parents b8f19ca + d42687d
commit 7bbbfbd
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/gitlab/exceptions.py b/gitlab/exceptions.py
@@ -91,6 +91,14 @@ class GitlabUnblockError(GitlabOperationError):
     pass
 
 
+class GitlabSubscribeError(GitlabOperationError):
+    pass
+
+
+class GitlabUnsubscribeError(GitlabOperationError):
+    pass
+
+
 class GitlabMRForbiddenError(GitlabOperationError):
     pass
 

diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -998,6 +998,22 @@ def Note(self, id=None, **kwargs):
                                                     issue_id=self.id,
                                                     **kwargs)
 
+    def subscribe(self, **kwargs):
+        url = ('/projects/%(project_id)s/issues/%(issue_id)s/subscription' %
+               {'project_id': self.project_id, 'issue_id': self.id})
+
+        r = self.gitlab._raw_post(url, **kwargs)
+        raise_error_from_response(r, GitlabSubscribeError)
+        self._set_from_dict(r.json())
+
+    def unsubscribe(self, **kwargs):
+        url = ('/projects/%(project_id)s/issues/%(issue_id)s/subscription' %
+               {'project_id': self.project_id, 'issue_id': self.id})
+
+        r = self.gitlab._raw_delete(url, **kwargs)
+        raise_error_from_response(r, GitlabUnsubscribeError)
+        self._set_from_dict(r.json())
+
 
 class ProjectIssueManager(BaseManager):
     obj_cls = ProjectIssue