116 lines
3.2 KiB
Python
116 lines
3.2 KiB
Python
|
# pylint: disable=C0111,R0903
|
||
|
|
|
||
|
|
"""
|
||
|
|
Displays the unread GitHub notifications count for a GitHub user using the following reasons:
|
||
|
|
|
||
|
|
* https://developer.github.com/v3/activity/notifications/#notification-reasons
|
||
|
|
|
||
|
|
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.
|
||
|
|
* github.reasons: Comma separated reasons to be parsed (e.g.: github.reasons=mention,team_mention,review_requested)
|
||
|
|
|
||
|
|
contributed by:
|
||
|
|
* v1 - `yvesh <https://github.com/yvesh>`_ - many thanks!
|
||
|
|
* v2 - `cristianmiranda <https://github.com/cristianmiranda>`_ - many thanks!
|
||
|
|
"""
|
||
|
|
|
||
|
|
import shutil
|
||
|
|
import requests
|
||
|
|
|
||
|
|
import core.module
|
||
|
|
import core.widget
|
||
|
|
import core.decorators
|
||
|
|
import core.input
|
||
|
|
|
||
|
|
import util.format
|
||
|
|
|
||
|
|
|
||
|
|
class Module(core.module.Module):
|
||
|
|
@core.decorators.every(minutes=5)
|
||
|
|
def __init__(self, config, theme):
|
||
|
|
super().__init__(config, theme, core.widget.Widget(self.github))
|
||
|
|
|
||
|
|
self.__count = 0
|
||
|
|
self.__label = ""
|
||
|
|
self.__requests = requests.Session()
|
||
|
|
self.__requests.headers.update(
|
||
|
|
{"Authorization": "token {}".format(self.parameter("token", ""))}
|
||
|
|
)
|
||
|
|
|
||
|
|
self.__reasons = []
|
||
|
|
reasons = self.parameter("reasons", "")
|
||
|
|
if reasons:
|
||
|
|
self.__reasons = util.format.aslist(reasons)
|
||
|
|
|
||
|
|
cmd = "xdg-open"
|
||
|
|
if not shutil.which(cmd):
|
||
|
|
cmd = "x-www-browser"
|
||
|
|
|
||
|
|
core.input.register(
|
||
|
|
self,
|
||
|
|
button=core.input.LEFT_MOUSE,
|
||
|
|
cmd="{} https://github.com/notifications".format(cmd),
|
||
|
|
)
|
||
|
|
|
||
|
|
def github(self, _):
|
||
|
|
return str(self.__label)
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
try:
|
||
|
|
url = "https://api.github.com/notifications"
|
||
|
|
notifications = self.__requests.get(url)
|
||
|
|
|
||
|
|
total = self.__getTotalUnreadNotificationsCount(notifications)
|
||
|
|
self.__count = total
|
||
|
|
self.__label = str(total)
|
||
|
|
|
||
|
|
counts = []
|
||
|
|
if len(self.__reasons) > 0:
|
||
|
|
for reason in self.__reasons:
|
||
|
|
unread = self.__getUnreadNotificationsCountByReason(
|
||
|
|
notifications, reason
|
||
|
|
)
|
||
|
|
counts.append(str(unread))
|
||
|
|
|
||
|
|
self.__label += " - "
|
||
|
|
self.__label += "/".join(counts)
|
||
|
|
|
||
|
|
except Exception as err:
|
||
|
|
print(err)
|
||
|
|
self.__label = "n/a"
|
||
|
|
|
||
|
|
def __getUnreadNotificationsCountByReason(self, notifications, reason):
|
||
|
|
return len(
|
||
|
|
list(
|
||
|
|
filter(
|
||
|
|
lambda notification: notification["unread"]
|
||
|
|
and notification["reason"] == reason,
|
||
|
|
notifications.json(),
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
def __getTotalUnreadNotificationsCount(self, notifications):
|
||
|
|
return len(
|
||
|
|
list(
|
||
|
|
filter(
|
||
|
|
lambda notification: notification["unread"], notifications.json()
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
def state(self, widget):
|
||
|
|
state = []
|
||
|
|
|
||
|
|
if self.__count > 0:
|
||
|
|
state.append("warning")
|
||
|
|
|
||
|
|
return state
|
||
|
|
|
||
|
|
|
||
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|