710342d7d4
* Use requests library instead of curl * Exactly parse the github API responses * Return a string value in order to pass unit tests see #86
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
# 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
|