l.behm/bumblebee-status
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[module/github] Use 'json' from requests and 'filter' instead of a loop

Browse source
This commit is contained in:
Frederic Junod 2017-07-07 06:46:10 +02:00
parent 8fb100c237
commit 0ce2a25d4e
1 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
bumblebee/modules/github.py
View file

@ -11,7 +11,6 @@ Parameters:
"""
import time
import json
import bumblebee.input
import bumblebee.output
import bumblebee.engine
@ -43,12 +42,8 @@ class Module(bumblebee.engine.Module):
return
try:
notifications = requests.get("https://api.github.com/notifications", headers={"Authorization":"token {}".format(token)}).text
unread = 0
for notification in json.loads(notifications):
if "unread" in notification and notification["unread"]:
unread += 1
self._count = unread
notifications = requests.get("https://api.github.com/notifications", headers={"Authorization":"token {}".format(token)}).json()
self._count = len(filter(lambda notification: notification.get("unread", False), notifications))
except Exception:
self._count = "n/a"