[module/github] Add pagination support
The GitHub API returns 50 items per page.
This commit is contained in:
parent
443a1098ca
commit
78adbc0b4a
1 changed files with 13 additions and 7 deletions
|
@ -29,6 +29,8 @@ class Module(bumblebee.engine.Module):
|
||||||
self._count = 0
|
self._count = 0
|
||||||
self._interval = int(self.parameter("interval", "5"))
|
self._interval = int(self.parameter("interval", "5"))
|
||||||
self._nextcheck = 0
|
self._nextcheck = 0
|
||||||
|
self._requests = requests.Session()
|
||||||
|
self._requests.headers.update({"Authorization":"token {}".format(self.parameter("token", ""))})
|
||||||
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
|
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
|
||||||
cmd="x-www-browser https://github.com/notifications")
|
cmd="x-www-browser https://github.com/notifications")
|
||||||
immediate_update = functools.partial(self.update, immediate=True)
|
immediate_update = functools.partial(self.update, immediate=True)
|
||||||
|
@ -41,15 +43,19 @@ class Module(bumblebee.engine.Module):
|
||||||
def update(self, _, immediate=False):
|
def update(self, _, immediate=False):
|
||||||
if immediate or self._nextcheck < int(time.time()):
|
if immediate or self._nextcheck < int(time.time()):
|
||||||
self._nextcheck = int(time.time()) + self._interval * 60
|
self._nextcheck = int(time.time()) + self._interval * 60
|
||||||
token = self.parameter("token", "")
|
|
||||||
|
|
||||||
if not token:
|
|
||||||
self._count = "n/a"
|
|
||||||
return
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
notifications = requests.get("https://api.github.com/notifications", headers={"Authorization":"token {}".format(token)}).json()
|
self._count = 0
|
||||||
self._count = len(filter(lambda notification: notification.get("unread", False), notifications))
|
url = "https://api.github.com/notifications"
|
||||||
|
while True:
|
||||||
|
notifications = self._requests.get(url)
|
||||||
|
self._count += len(filter(lambda notification: notification.get("unread", False), notifications.json()))
|
||||||
|
next_link = notifications.links.get('next')
|
||||||
|
if next_link is not None:
|
||||||
|
url = next_link.get('url')
|
||||||
|
else:
|
||||||
|
break
|
||||||
|
|
||||||
except Exception:
|
except Exception:
|
||||||
self._count = "n/a"
|
self._count = "n/a"
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue