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

[modules/github] Update to latest API

Browse source
This commit is contained in:
tobi-wan-kenobi 2020-04-18 17:05:19 +02:00
parent 2c4af853a5
commit 0a6f914232
1 changed files with 22 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
modules/contrib/github.py
View file

@ -10,39 +10,36 @@ Parameters:
* github.interval: Interval in minutes between updates, default is 5. * github.interval: Interval in minutes between updates, default is 5.
""" """
import bumblebee.input
import bumblebee.output
import bumblebee.engine
try:
import requests import requests
except ImportError:
pass
class Module(bumblebee.engine.Module): import core.module
def __init__(self, engine, config): import core.widget
super(Module, self).__init__(engine, config, import core.decorators
bumblebee.output.Widget(full_text=self.github) import core.input
)
self._count = 0 class Module(core.module.Module):
self.interval_factor(60) @core.decorators.every(minutes=5)
self.interval(5) def __init__(self, config):
self._requests = requests.Session() super().__init__(config, core.widget.Widget(self.github))
self._requests.headers.update({'Authorization':'token {}'.format(self.parameter('token', ''))})
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE, self.__count = 0
self.__requests = requests.Session()
self.__requests.headers.update({'Authorization':'token {}'.format(self.parameter('token', ''))})
core.input.register(self, button=core.input.LEFT_MOUSE,
cmd='x-www-browser https://github.com/notifications') cmd='x-www-browser https://github.com/notifications')
engine.input.register_callback(self, button=bumblebee.input.RIGHT_MOUSE, cmd=self.update) core.input.register(self, button=core.input.RIGHT_MOUSE, cmd=self.update)
def github(self, _): def github(self, _):
return str(self._count) return str(self.__count)
def update(self, _): def update(self):
try: try:
self._count = 0 self.__count = 0
url = 'https://api.github.com/notifications' url = 'https://api.github.com/notifications'
while True: while True:
notifications = self._requests.get(url) notifications = self.__requests.get(url)
self._count += len(list(filter(lambda notification: notification['unread'], notifications.json()))) self.__count += len(list(filter(lambda notification: notification['unread'], notifications.json())))
next_link = notifications.links.get('next') next_link = notifications.links.get('next')
if next_link is not None: if next_link is not None:
url = next_link.get('url') url = next_link.get('url')
@ -50,7 +47,6 @@ class Module(bumblebee.engine.Module):
break break
except Exception: except Exception:
self._count = 'n/a' self.__count = 'n/a'
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4