diff --git a/modules/contrib/stock.py b/modules/contrib/stock.py new file mode 100644 index 0000000..5333700 --- /dev/null +++ b/modules/contrib/stock.py @@ -0,0 +1,61 @@ +# -*- coding: UTF-8 -*- +# pylint: disable=C0111,R0903 + +"""Display a stock quote from worldtradingdata.com + +Requires the following python packages: + * requests + +Parameters: + * stock.symbols : Comma-separated list of symbols to fetch + * stock.change : Should we fetch change in stock value (defaults to True) +""" + +import bumblebee.input +import bumblebee.output +import bumblebee.engine +import bumblebee.util + +import json +import requests + +import logging + +class Module(bumblebee.engine.Module): + def __init__(self, engine, config): + super(Module, self).__init__(engine, config, + bumblebee.output.Widget(full_text=self.value) + ) + self._symbols = self.parameter('symbols', '') + self._change = bumblebee.util.asbool(self.parameter('change', True)) + self._value = None + self.interval_factor(60) + self.interval(60) + + def value(self, widget): + results = [] + if not self._value: + return 'n/a' + data = json.loads(self._value) + + for symbol in data['quoteResponse']['result']: + valkey = 'regularMarketChange' if self._change else 'regularMarketPrice' + sym = 'n/a' if not 'symbol' in symbol else symbol['symbol'] + currency = 'USD' if not 'currency' in symbol else symbol['currency'] + val = 'n/a' if not valkey in symbol else '{:.2f}'.format(symbol[valkey]) + results.append('{} {} {}'.format(sym, val, currency)) + return u' '.join(results) + + def fetch(self): + if self._symbols: + url = 'https://query1.finance.yahoo.com/v7/finance/quote?symbols=' + url += self._symbols + '&fields=regularMarketPrice,currency,regularMarketChange' + return requests.get(url).text.strip() + else: + logging.error('unable to retrieve stock exchange rate') + return None + + def update(self, widgets): + self._value = self.fetch() + +# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4