264 lines
8.5 KiB
Python
264 lines
8.5 KiB
Python
from datetime import datetime
|
|
import inspect
|
|
import requests
|
|
import sys
|
|
from threading import Thread
|
|
import time
|
|
import csv
|
|
from decimal import Decimal
|
|
|
|
from .bitcoin import COIN
|
|
from .i18n import _
|
|
from .util import PrintError, ThreadJob
|
|
|
|
|
|
# See https://en.wikipedia.org/wiki/ISO_4217
|
|
CCY_PRECISIONS = {}
|
|
'''
|
|
{'BHD': 3, 'BIF': 0, 'BYR': 0, 'CLF': 4, 'CLP': 0,
|
|
'CVE': 0, 'DJF': 0, 'GNF': 0, 'IQD': 3, 'ISK': 0,
|
|
'JOD': 3, 'JPY': 0, 'KMF': 0, 'KRW': 0, 'KWD': 3,
|
|
'LYD': 3, 'MGA': 1, 'MRO': 1, 'OMR': 3, 'PYG': 0,
|
|
'RWF': 0, 'TND': 3, 'UGX': 0, 'UYI': 0, 'VND': 0,
|
|
'VUV': 0, 'XAF': 0, 'XAU': 4, 'XOF': 0, 'XPF': 0}
|
|
'''
|
|
|
|
class ExchangeBase(PrintError):
|
|
|
|
def __init__(self, on_quotes, on_history):
|
|
self.history = {}
|
|
self.quotes = {}
|
|
self.on_quotes = on_quotes
|
|
self.on_history = on_history
|
|
|
|
def get_json(self, site, get_string):
|
|
# APIs must have https
|
|
url = ''.join(['https://', site, get_string])
|
|
response = requests.request('GET', url, headers={'User-Agent' : 'Electrum'})
|
|
return response.json()
|
|
|
|
def get_csv(self, site, get_string):
|
|
url = ''.join(['https://', site, get_string])
|
|
response = requests.request('GET', url, headers={'User-Agent' : 'Electrum'})
|
|
reader = csv.DictReader(response.content.decode().split('\n'))
|
|
return list(reader)
|
|
|
|
def name(self):
|
|
return self.__class__.__name__
|
|
|
|
def update_safe(self, ccy):
|
|
try:
|
|
self.print_error("getting fx quotes for", ccy)
|
|
self.quotes = self.get_rates(ccy)
|
|
self.print_error("received fx quotes")
|
|
except BaseException as e:
|
|
self.print_error("failed fx quotes:", e)
|
|
self.on_quotes()
|
|
|
|
def update(self, ccy):
|
|
t = Thread(target=self.update_safe, args=(ccy,))
|
|
t.setDaemon(True)
|
|
t.start()
|
|
|
|
def get_historical_rates_safe(self, ccy):
|
|
try:
|
|
self.print_error("requesting fx history for", ccy)
|
|
self.history[ccy] = self.historical_rates(ccy)
|
|
self.print_error("received fx history for", ccy)
|
|
self.on_history()
|
|
except BaseException as e:
|
|
self.print_error("failed fx history:", e)
|
|
|
|
def get_historical_rates(self, ccy):
|
|
result = self.history.get(ccy)
|
|
if not result and ccy in self.history_ccys():
|
|
t = Thread(target=self.get_historical_rates_safe, args=(ccy,))
|
|
t.setDaemon(True)
|
|
t.start()
|
|
return result
|
|
|
|
def history_ccys(self):
|
|
return []
|
|
|
|
def historical_rate(self, ccy, d_t):
|
|
return self.history.get(ccy, {}).get(d_t.strftime('%Y-%m-%d'))
|
|
|
|
def get_currencies(self):
|
|
rates = self.get_rates('')
|
|
return sorted([str(a) for (a, b) in rates.items() if b is not None and len(a) in [3,4]])
|
|
|
|
|
|
class CryptoCompare(ExchangeBase):
|
|
def get_rates(self, ccy):
|
|
json = self.get_json('min-api.cryptocompare.com',
|
|
"/data/pricehistorical?fsym=ZCL&tsyms=USD")
|
|
return {'USD': Decimal(json['ZCL']['USD'])}
|
|
|
|
|
|
|
|
def dictinvert(d):
|
|
inv = {}
|
|
for k, vlist in d.items():
|
|
for v in vlist:
|
|
keys = inv.setdefault(v, [])
|
|
keys.append(k)
|
|
return inv
|
|
|
|
def get_exchanges_and_currencies():
|
|
import os, json
|
|
path = os.path.join(os.path.dirname(__file__), 'currencies.json')
|
|
try:
|
|
with open(path, 'r') as f:
|
|
return json.loads(f.read())
|
|
except:
|
|
pass
|
|
d = {}
|
|
is_exchange = lambda obj: (inspect.isclass(obj)
|
|
and issubclass(obj, ExchangeBase)
|
|
and obj != ExchangeBase)
|
|
exchanges = dict(inspect.getmembers(sys.modules[__name__], is_exchange))
|
|
for name, klass in exchanges.items():
|
|
exchange = klass(None, None)
|
|
try:
|
|
d[name] = exchange.get_currencies()
|
|
except:
|
|
continue
|
|
with open(path, 'w') as f:
|
|
f.write(json.dumps(d, indent=4, sort_keys=True))
|
|
return d
|
|
|
|
|
|
CURRENCIES = get_exchanges_and_currencies()
|
|
|
|
|
|
def get_exchanges_by_ccy(history=True):
|
|
if not history:
|
|
return dictinvert(CURRENCIES)
|
|
d = {}
|
|
exchanges = CURRENCIES.keys()
|
|
for name in exchanges:
|
|
klass = globals()[name]
|
|
exchange = klass(None, None)
|
|
d[name] = exchange.history_ccys()
|
|
return dictinvert(d)
|
|
|
|
|
|
class FxThread(ThreadJob):
|
|
|
|
def __init__(self, config, network):
|
|
self.config = config
|
|
self.network = network
|
|
self.ccy = self.get_currency()
|
|
self.history_used_spot = False
|
|
self.ccy_combo = None
|
|
self.hist_checkbox = None
|
|
self.set_exchange(self.config_exchange())
|
|
|
|
def get_currencies(self, h):
|
|
d = get_exchanges_by_ccy(h)
|
|
return sorted(d.keys())
|
|
|
|
def get_exchanges_by_ccy(self, ccy, h):
|
|
d = get_exchanges_by_ccy(h)
|
|
return d.get(ccy, [])
|
|
|
|
def ccy_amount_str(self, amount, commas):
|
|
prec = CCY_PRECISIONS.get(self.ccy, 2)
|
|
fmt_str = "{:%s.%df}" % ("," if commas else "", max(0, prec))
|
|
return fmt_str.format(round(amount, prec))
|
|
|
|
def run(self):
|
|
# This runs from the plugins thread which catches exceptions
|
|
if self.is_enabled():
|
|
if self.timeout ==0 and self.show_history():
|
|
self.exchange.get_historical_rates(self.ccy)
|
|
if self.timeout <= time.time():
|
|
self.timeout = time.time() + 150
|
|
self.exchange.update(self.ccy)
|
|
|
|
def is_enabled(self):
|
|
return bool(self.config.get('use_exchange_rate'))
|
|
|
|
def set_enabled(self, b):
|
|
return self.config.set_key('use_exchange_rate', bool(b))
|
|
|
|
def get_history_config(self):
|
|
return bool(self.config.get('history_rates'))
|
|
|
|
def set_history_config(self, b):
|
|
self.config.set_key('history_rates', bool(b))
|
|
|
|
def get_fiat_address_config(self):
|
|
return bool(self.config.get('fiat_address'))
|
|
|
|
def set_fiat_address_config(self, b):
|
|
self.config.set_key('fiat_address', bool(b))
|
|
|
|
def get_currency(self):
|
|
# Use when dynamic fetching is needed
|
|
return self.config.get('currency', 'USD')
|
|
|
|
def config_exchange(self):
|
|
return self.config.get('use_exchange', 'CryptoCompare')
|
|
|
|
def show_history(self):
|
|
return self.is_enabled() and self.get_history_config() and self.ccy in self.exchange.history_ccys()
|
|
|
|
def set_currency(self, ccy):
|
|
self.ccy = ccy
|
|
self.config.set_key('currency', ccy, True)
|
|
self.timeout = 0 # Because self.ccy changes
|
|
self.on_quotes()
|
|
|
|
def set_exchange(self, name):
|
|
class_ = globals().get(name, CryptoCompare)
|
|
self.print_error("using exchange", name)
|
|
if self.config_exchange() != name:
|
|
self.config.set_key('use_exchange', name, True)
|
|
self.exchange = class_(self.on_quotes, self.on_history)
|
|
# A new exchange means new fx quotes, initially empty. Force
|
|
# a quote refresh
|
|
self.timeout = 0
|
|
|
|
def on_quotes(self):
|
|
self.network.trigger_callback('on_quotes')
|
|
|
|
def on_history(self):
|
|
self.network.trigger_callback('on_history')
|
|
|
|
def exchange_rate(self):
|
|
'''Returns None, or the exchange rate as a Decimal'''
|
|
rate = self.exchange.quotes.get(self.ccy)
|
|
if rate:
|
|
return Decimal(rate)
|
|
|
|
def format_amount_and_units(self, btc_balance):
|
|
rate = self.exchange_rate()
|
|
return '' if rate is None else "%s %s" % (self.value_str(btc_balance, rate), self.ccy)
|
|
|
|
def get_fiat_status_text(self, btc_balance, base_unit, decimal_point):
|
|
rate = self.exchange_rate()
|
|
return _(" (No exchange rate available)") if rate is None else " 1 %s=%s %s" % (base_unit,
|
|
self.value_str(COIN / (10**(8 - decimal_point)), rate), self.ccy)
|
|
|
|
def value_str(self, satoshis, rate):
|
|
if satoshis is None: # Can happen with incomplete history
|
|
return _("Unknown")
|
|
if rate:
|
|
value = Decimal(satoshis) / COIN * Decimal(rate)
|
|
return "%s" % (self.ccy_amount_str(value, True))
|
|
return _("No data")
|
|
|
|
def history_rate(self, d_t):
|
|
rate = self.exchange.historical_rate(self.ccy, d_t)
|
|
# Frequently there is no rate for today, until tomorrow :)
|
|
# Use spot quotes in that case
|
|
if rate is None and (datetime.today().date() - d_t.date()).days <= 2:
|
|
rate = self.exchange.quotes.get(self.ccy)
|
|
self.history_used_spot = True
|
|
return rate
|
|
|
|
def historical_value_str(self, satoshis, d_t):
|
|
rate = self.history_rate(d_t)
|
|
return self.value_str(satoshis, rate)
|