Create lock() and unlock() for issues

adrianmoisey · adrianmoisey · commit 01f722516034 · 2016-02-16T11:02:52.000+02:00
diff --git a/github3/issues/issue.py b/github3/issues/issue.py
@@ -31,6 +31,12 @@ class Issue(GitHubCore):
 
     """
 
+    # The Accept header will likely be removable once the feature is out of
+    # preview mode. See: https://git.io/vgXmB
+    LOCKING_PREVIEW_HEADERS = {
+        'Accept': 'application/vnd.github.the-key-preview+json'
+    }
+
     def _update_attributes(self, issue):
         self._api = issue.get('url', '')
         #: :class:`User <github3.users.User>` representing the user the issue
@@ -254,6 +260,17 @@ def labels(self, number=-1, etag=None):
         url = self._build_url('labels', base_url=self._api)
         return self._iter(int(number), url, Label, etag=etag)
 
+    @requires_auth
+    def lock(self):
+        """Lock an issue.
+
+        :returns: bool
+        """
+        headers = Issue.LOCKING_PREVIEW_HEADERS
+
+        url = self._build_url('lock', base_url=self._api)
+        return self._boolean(self._put(url, headers=headers), 204, 404)
+
     def pull_request(self):
         """Retrieve the pull request associated with this issue.
 
@@ -309,3 +326,14 @@ def reopen(self):
         labels = [str(l) for l in self.original_labels]
         return self.edit(self.title, self.body, assignee, 'open',
                          number, labels)
+
+    @requires_auth
+    def unlock(self):
+        """Unlock an issue.
+
+        :returns: bool
+        """
+        headers = Issue.LOCKING_PREVIEW_HEADERS
+
+        url = self._build_url('lock', base_url=self._api)
+        return self._boolean(self._delete(url, headers=headers), 204, 404)
