Add support for subscribe and unsubscribe in issues

IvicaArsovSculpteo · IvicaArsovSculpteo · commit d42687db9f0c · 2016-05-20T18:39:04.000+02:00
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
