remaining_requests attr to Github and Services

copitux · copitux · commit 8345cf66caed · 2012-03-03T11:29:56.000+01:00
Updated with each request
diff --git a/pygithub3/core/client.py b/pygithub3/core/client.py
@@ -14,6 +14,8 @@
 class Client(object):
     """ Client to send configurated requests"""
 
+    remaining_requests = '~'
+
     def __init__(self, **kwargs):
         self.requester = requests.session()
         self.config = {
@@ -72,6 +74,8 @@ def wrapper(self, verb, request, **kwargs):
     def request(self, verb, request, **kwargs):
         request = "%s%s" % (self.config['base_url'], request)
         response = self.requester.request(verb, request, **kwargs)
+        Client.remaining_requests = response.headers.get(
+            'x-ratelimit-remaining', -1)
         GithubError(response).process()
         return response
 
diff --git a/pygithub3/github.py b/pygithub3/github.py
@@ -18,6 +18,12 @@ def __init__(self, **config):
         self._users = User(**config)
         self._repos = Repos(**config)
 
+    @property
+    def remaining_requests(self):
+        """ Limit of Github API v3 """
+        from pygithub3.core.client import Client
+        return Client.remaining_requests
+
     @property
     def users(self):
         """
diff --git a/pygithub3/services/base.py b/pygithub3/services/base.py
@@ -41,6 +41,10 @@ def __init__(self, **config):
         self._client = Client(**config)
         self.request_builder = Factory()
 
+    @property
+    def remaining_requests(self):
+        return Client.remaining_requests
+
     def get_user(self):
         return self._client.user
 
