621564d247
As correctly pointed out by @rrhuffy, restricting update intervals to minutes is pretty arbitrary. Therefore, change logic to specify updates intervals in *seconds*. Also, to maintain backwards compatibility for users that already have custom intervals in their config, allow a module to specify their own "factor" for intervals. So a module that expects the interval to be in minutes can set the factor to 60.
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
"""Displays the unread GitHub notifications for a GitHub user
|
|
|
|
Requires the following library:
|
|
* requests
|
|
|
|
Parameters:
|
|
* github.token: GitHub user access token, the token needs to have the 'notifications' scope.
|
|
* github.interval: Interval in minutes between updates, default is 5.
|
|
"""
|
|
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
|
|
try:
|
|
import requests
|
|
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_factor(60)
|
|
self.interval(5)
|
|
self._requests = requests.Session()
|
|
self._requests.headers.update({"Authorization":"token {}".format(self.parameter("token", ""))})
|
|
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
|
|
cmd="x-www-browser https://github.com/notifications")
|
|
engine.input.register_callback(self, button=bumblebee.input.RIGHT_MOUSE, cmd=self.update)
|
|
|
|
def github(self, _):
|
|
return str(self._count)
|
|
|
|
def update(self, _):
|
|
try:
|
|
self._count = 0
|
|
url = "https://api.github.com/notifications"
|
|
while True:
|
|
notifications = self._requests.get(url)
|
|
self._count += len(list(filter(lambda notification: notification['unread'], notifications.json())))
|
|
next_link = notifications.links.get('next')
|
|
if next_link is not None:
|
|
url = next_link.get('url')
|
|
else:
|
|
break
|
|
|
|
except Exception:
|
|
self._count = "n/a"
|
|
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|