diff options
| -rw-r--r-- | github3/core.py | 11 | ||||
| -rw-r--r-- | github3/handlers/base.py | 2 |
2 files changed, 9 insertions, 4 deletions
diff --git a/github3/core.py b/github3/core.py index 5c7343a..d7237af 100644 --- 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 @@ class Paginate: 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 @@ class Paginate: 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 index ab9801e..8bcabca 100644 --- a/github3/handlers/base.py +++ b/github3/handlers/base.py @@ -42,7 +42,7 @@ class Handler(object): """ 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: |