Added support for extra args to get_resources

copitux · copitux · commit c614dde5e58b · 2011-11-09T00:29:48.000+01:00
Paginate coupled? :S
diff --git a/github3/core.py b/github3/core.py
@@ -8,11 +8,13 @@ class Paginate:
 
     :param resource: URL resource
     :param requester: Bound method to request. See `GithubCore.get`
+    :param kwargs: Args to request (params)
     """
 
-    def __init__(self, resource, requester):
+    def __init__(self, resource, requester, **kwargs):
         self.resource = resource
         self.requester = requester
+        self.kwargs = kwargs
         self.page = 1
 
     def _last_page(self, link):
@@ -27,12 +29,14 @@ def _last_page(self, link):
 
         return self.last
 
+    # TODO: reset iterators... multiple?
     def __iter__(self):
         return self
 
     def initial(self):
         """ First request. Force requester to paginate returning link header """
-        link, content = self.requester(self.resource, paginate=True, page=1)
+        link, content = self.requester(self.resource, paginate=True,
+                                       page=1, **self.kwargs)
         self.last = self._last_page(link) if link else 1
         return content
 
@@ -45,7 +49,8 @@ def next(self):
             if self.page > self.last:
                 raise StopIteration
             else:
-                content = self.requester(self.resource, page=self.page)
+                content = self.requester(self.resource, page=self.page,
+                                         **self.kwargs)
                 self.page += 1
                 return content
 
diff --git a/github3/handlers/base.py b/github3/handlers/base.py
@@ -42,7 +42,7 @@ def _get_resources(self, resource, model=None, limit=None, **kwargs):
         """ Hander request to multiple resources """
 
         resource = self._prefix_resource(resource)
-        page_resources = Paginate(resource, self._gh.get)
+        page_resources = Paginate(resource, self._gh.get, kwargs)
         counter = 1
         for page in page_resources:
             for raw_resource in page:
