139 lines
4.4 KiB
Python
139 lines
4.4 KiB
Python
# -*- coding: UTF-8 -*-
|
|
# pylint: disable=C0111,R0903
|
|
|
|
"""Displays currency exchange rates. Currently, displays currency between GBP and USD/EUR only.
|
|
|
|
Requires the following python packages:
|
|
* requests
|
|
|
|
Parameters:
|
|
* currency.interval: Interval in minutes between updates, default is 1.
|
|
* currency.source: Source currency (ex. 'GBP', 'EUR'). Defaults to 'auto', which infers the local one from IP address.
|
|
* currency.destination: Comma-separated list of destination currencies (defaults to 'USD,EUR')
|
|
* currency.sourceformat: String format for source formatting; Defaults to '{}: {}' and has two variables,
|
|
the base symbol and the rate list
|
|
* currency.destinationdelimiter: Delimiter used for separating individual rates (defaults to '|')
|
|
|
|
Note: source and destination names right now must correspond to the names used by the API of https://markets.ft.com
|
|
"""
|
|
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
try:
|
|
import requests
|
|
except ImportError:
|
|
pass
|
|
try:
|
|
from babel.numbers import format_currency
|
|
except ImportError:
|
|
format_currency = None
|
|
import json
|
|
import os
|
|
|
|
SYMBOL = {
|
|
'GBP': u'£', 'EUR': u'€', 'USD': u'$', 'JPY': u'¥', 'KRW': u'₩'
|
|
}
|
|
DEFAULT_DEST = 'USD,EUR,auto'
|
|
DEFAULT_SRC = 'GBP'
|
|
|
|
API_URL = 'https://markets.ft.com/data/currencies/ajax/conversion?baseCurrency={}&comparison={}'
|
|
LOCATION_URL = 'https://ipvigilante.com/'
|
|
|
|
|
|
def get_local_country():
|
|
r = requests.get(LOCATION_URL)
|
|
location = r.json()
|
|
return location['data']['country_name']
|
|
|
|
|
|
def load_country_to_currency():
|
|
fname = os.path.join(
|
|
os.path.dirname(os.path.abspath(__file__)),
|
|
'data', 'country-by-currency-code.json')
|
|
with open(fname, 'r') as f:
|
|
data = json.load(f)
|
|
country2curr = {}
|
|
for dt in data:
|
|
country2curr[dt['country']] = dt['currency_code']
|
|
|
|
return country2curr
|
|
|
|
|
|
class Module(bumblebee.engine.Module):
|
|
def __init__(self, engine, config):
|
|
super(Module, self).__init__(engine, config,
|
|
bumblebee.output.Widget(full_text=self.price)
|
|
)
|
|
self._data = []
|
|
self.interval_factor(60)
|
|
self.interval(1)
|
|
self._nextcheck = 0
|
|
|
|
src = self.parameter('source', DEFAULT_SRC)
|
|
if src == 'auto':
|
|
self._base = self.find_local_currency()
|
|
else:
|
|
self._base = src
|
|
|
|
self._symbols = []
|
|
for d in self.parameter('destination', DEFAULT_DEST).split(','):
|
|
if d == 'auto':
|
|
new = self.find_local_currency()
|
|
else:
|
|
new = d
|
|
if new != self._base:
|
|
self._symbols.append(new)
|
|
|
|
def price(self, widget):
|
|
if len(self._data) == 0:
|
|
return '?'
|
|
|
|
rates = []
|
|
for sym, rate in self._data:
|
|
rate_float = float(rate.replace(',',''))
|
|
if format_currency:
|
|
rates.append(format_currency(rate_float, sym))
|
|
else:
|
|
rate = self.fmt_rate(rate)
|
|
rates.append(u'{}{}'.format(rate, SYMBOL[sym] if sym in SYMBOL else sym))
|
|
|
|
basefmt = u'{}'.format(self.parameter('sourceformat', '{}={}'))
|
|
ratefmt = u'{}'.format(self.parameter('destinationdelimiter', '='))
|
|
|
|
if format_currency:
|
|
base_val = format_currency(1, self._base)
|
|
else:
|
|
base_val = '1{}'.format(SYMBOL[self._base] if self._base in SYMBOL else self._base)
|
|
|
|
return basefmt.format(base_val, ratefmt.join(rates))
|
|
|
|
def update(self, widgets):
|
|
self._data = []
|
|
for symbol in self._symbols:
|
|
url = API_URL.format(self._base, symbol)
|
|
try:
|
|
response = requests.get(url).json()
|
|
self._data.append((symbol, response['data']['exchangeRate']))
|
|
except Exception:
|
|
pass
|
|
|
|
def find_local_currency(self):
|
|
"""Use geolocation lookup to find local currency"""
|
|
try:
|
|
country = get_local_country()
|
|
currency_map = load_country_to_currency()
|
|
return currency_map.get(country, DEFAULT_SRC)
|
|
except:
|
|
return DEFAULT_SRC
|
|
|
|
def fmt_rate(self, rate):
|
|
float_rate = float(rate.replace(',', ''))
|
|
if not 0.01 < float_rate < 100:
|
|
ret = rate
|
|
else:
|
|
ret = '%.3g' % float_rate
|
|
|
|
return ret
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|