bumblebee-status/modules/contrib/getcrypto.py

# pylint: disable=C0111,R0903

"""Displays the price of a cryptocurrency.

Requires the following python packages:
    * requests

Parameters:
    * getcrypto.interval: Interval in seconds for updating the price, default is 120, less than that will probably get your IP banned.
    * getcrypto.getbtc: 0 for not getting price of BTC, 1 for getting it (default).
    * getcrypto.geteth: 0 for not getting price of ETH, 1 for getting it (default).
    * getcrypto.getltc: 0 for not getting price of LTC, 1 for getting it (default).
    * getcrypto.getcur: Set the currency to display the price in, usd is the default.
"""

import requests
from requests.exceptions import RequestException
import time

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

import util.format


def getfromkrak(coin, currency):
    abbrev = {
        "Btc": ["xbt", "XXBTZ"],
        "Eth": ["eth", "XETHZ"],
        "Ltc": ["ltc", "XLTCZ"],
    }
    data = abbrev.get(coin, None)
    if not data:
        return
    epair = "{}{}".format(data[0], currency)
    tickname = "{}{}".format(data[1], currency.upper())
    try:
        krakenget = requests.get(
            "https://api.kraken.com/0/public/Ticker?pair=" + epair
        ).json()
    except (RequestException, Exception):
        return "No connection"
    if not "result" in krakenget:
        return "No data"
    kethusdask = float(krakenget["result"][tickname]["a"][0])
    kethusdbid = float(krakenget["result"][tickname]["b"][0])
    return coin + ": " + str((kethusdask + kethusdbid) / 2)[0:6]


class Module(core.module.Module):
    @core.decorators.every(minutes=30)
    def __init__(self, config, theme):
        super().__init__(config, theme, core.widget.Widget(self.curprice))

        self.__curprice = ""
        self.__getbtc = util.format.asbool(self.parameter("getbtc", True))
        self.__geteth = util.format.asbool(self.parameter("geteth", True))
        self.__getltc = util.format.asbool(self.parameter("getltc", True))
        self.__getcur = self.parameter("getcur", "usd")
        core.input.register(
            self, button=core.input.LEFT_MOUSE, cmd="xdg-open https://cryptowat.ch/"
        )

    def curprice(self, widget):
        return self.__curprice

    def update(self):
        currency = self.__getcur
        btcprice, ethprice, ltcprice = "", "", ""
        if self.__getbtc:
            btcprice = getfromkrak("Btc", currency)
        if self.__geteth:
            ethprice = getfromkrak("Eth", currency)
        if self.__getltc:
            ltcprice = getfromkrak("Ltc", currency)
        self.__curprice = (
            btcprice
            + " " * (self.__getbtc * self.__geteth)
            + ethprice
            + " " * (self.__getltc * max(self.__getbtc, self.__geteth))
            + ltcprice
        )


# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4