Add requires_auth decorator to GitHubCore.

sigmavirus24 · sigmavirus24 · commit 087c28884da3 · 2012-07-16T14:33:05.000-04:00
GitHubCore is imported in just about every module so it only makes sense to
place it there.
diff --git a/github3/models.py b/github3/models.py
@@ -100,6 +100,22 @@ def to_json(self):
         """Return the json representing this object."""
         return self._json_data
 
+    @staticmethod
+    def requires_auth(func):
+        """Decorator to note which class methods require authorization."""
+        def auth_wrapper(self, *args, **kwargs):
+            auth = False
+            if hasattr(self, '_session'):
+                auth = self._session.auth or \
+                    self._session.headers['Authorization']
+
+            if auth:
+                return func(self, *args, **kwargs)
+            else:
+                raise Error(type('Faux Request', (object, ),
+                    {'status_code': 401, 'message': 'Requires authentication'}
+                    ))
+
 
 class BaseComment(GitHubCore):
     """The :class:`BaseComment <BaseComment>` object. A basic class for Gist,
@@ -150,13 +166,15 @@ def created_at(self):
         """datetime object representing when the comment was created."""
         return self._created
 
+    @GitHubCore.requires_auth
     def delete(self):
         """Delete this comment.
 
         :returns: bool
         """
-        return self._delete(self._api)
+        return self._boolean(self._delete(self._api), 204, 404)
 
+    @GitHubCore.requires_auth
     def edit(self, body):
         """Edit this comment.
 
