diff --git a/bumblebee/modules/github.py b/bumblebee/modules/github.py new file mode 100644 index 0000000..187d266 --- /dev/null +++ b/bumblebee/modules/github.py @@ -0,0 +1,54 @@ +# pylint: disable=C0111,R0903 + +"""Displays the unread GitHub notifications for a GitHub user + +Requires the following executable: + * curl + +Parameters: + * github.token: GitHub user access token + * github.interval: Interval in minutes +""" + +import bumblebee.input +import bumblebee.output +import bumblebee.engine +import re +import time +import json + +try: + import requests + from requests.exceptions import RequestException +except ImportError: + pass + +class Module(bumblebee.engine.Module): + def __init__(self, engine, config): + super(Module, self).__init__(engine, config, + bumblebee.output.Widget(full_text=self.github) + ) + self._count = 0 + self._interval = int(self.parameter("interval", "5")) + self._nextcheck = 0 + + def github(self, widget): + return str(self._count) + + def update(self, widgets): + if self._nextcheck < int(time.time()): + self._nextcheck = int(time.time()) + self._interval * 60 + token = self.parameter("token", "") + + if not token: + self._count = 0 + return + + notifications = requests.get("https://api.github.com/notifications?access_token={}".format(token)).text + unread = 0 + for notification in json.loads(notifications): + if "unread" in notification and notification["unread"] == True: + unread += 1 + self._count = unread + +# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4 diff --git a/screenshots/github.png b/screenshots/github.png new file mode 100644 index 0000000..bbcf48b Binary files /dev/null and b/screenshots/github.png differ