bumblebee-status/modules/contrib/github.py

# 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 shutil
import requests

import core.module
import core.widget
import core.decorators
import core.input


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.__requests = requests.Session()
        self.__requests.headers.update(
            {"Authorization": "token {}".format(self.parameter("token", ""))}
        )

        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),
        )
        core.input.register(self, button=core.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