mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-15 20:19:23 +00:00
211 lines
7.1 KiB
Python
Executable File
211 lines
7.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Fetch information about coins and tokens supported by Trezor and update it in coins_details.json."""
|
|
import time
|
|
import json
|
|
import requests
|
|
import pprint
|
|
|
|
SKIP_COINMARKETCAP = True
|
|
|
|
def coinmarketcap_info(shortcut):
|
|
if SKIP_COINMARKETCAP:
|
|
raise Exception("Skipping coinmarketcap call")
|
|
|
|
shortcut = shortcut.replace(' ', '-')
|
|
url = 'https://api.coinmarketcap.com/v1/ticker/%s/?convert=USD' % shortcut
|
|
try:
|
|
return requests.get(url).json()[0]
|
|
except KeyboardInterrupt:
|
|
raise
|
|
except:
|
|
print("Cannot fetch Coinmarketcap info for %s" % shortcut)
|
|
|
|
def update_marketcap(obj, shortcut):
|
|
try:
|
|
obj['marketcap_usd'] = int(float(coinmarketcap_info(shortcut)['market_cap_usd']))
|
|
except:
|
|
pass
|
|
|
|
def coinmarketcap_global():
|
|
if SKIP_COINMARKETCAP:
|
|
raise Exception("Skipping coinmarketcap call")
|
|
|
|
url = 'https://api.coinmarketcap.com/v1/global'
|
|
ret = requests.get(url)
|
|
data = ret.json()
|
|
return data
|
|
|
|
def set_default(obj, key, default_value):
|
|
obj[key] = obj.setdefault(key, default_value)
|
|
|
|
def update_info(details):
|
|
details['info']['updated_at'] = int(time.time())
|
|
details['info']['updated_at_readable'] = time.asctime()
|
|
details['info']['t1_coins'] = len([True for _, c in details['coins'].items() if c['t1_enabled'] == 'yes'])
|
|
details['info']['t2_coins'] = len([True for _, c in details['coins'].items() if c['t2_enabled'] == 'yes'])
|
|
|
|
try:
|
|
details['info']['total_marketcap_usd'] = int(coinmarketcap_global()['total_market_cap_usd'])
|
|
except:
|
|
pass
|
|
|
|
marketcap = 0
|
|
for k, c in details['coins'].items():
|
|
if c['t1_enabled'] == 'yes' or c['t2_enabled'] == 'yes':
|
|
marketcap += details['coins'][k].setdefault('marketcap_usd', 0)
|
|
details['info']['marketcap_usd'] = marketcap
|
|
|
|
def check_unsupported(details, prefix, supported):
|
|
for k in details['coins'].keys():
|
|
if not k.startswith(prefix):
|
|
continue
|
|
if k not in supported:
|
|
print("%s not supported by Trezor? (Possible manual entry)" % k)
|
|
|
|
def update_coins(details):
|
|
coins = json.load(open('coins.json', 'r'))
|
|
|
|
supported = []
|
|
for coin in coins:
|
|
if coin['firmware'] != 'stable':
|
|
continue
|
|
|
|
# print("Updating", coin['coin_label'], coin['coin_shortcut'])
|
|
key = "coin:%s" % coin['coin_shortcut']
|
|
supported.append(key)
|
|
out = details['coins'].setdefault(key, {})
|
|
out['type'] = 'coin'
|
|
set_default(out, 'shortcut', coin['coin_shortcut'])
|
|
set_default(out, 'name', coin['coin_label'])
|
|
set_default(out, 'links', {})
|
|
set_default(out, 't1_enabled', 'yes')
|
|
set_default(out, 't2_enabled', 'yes')
|
|
update_marketcap(out, coin.get('coinmarketcap_alias', coin['coin_label']))
|
|
|
|
check_unsupported(details, 'coin:', supported)
|
|
|
|
def update_erc20(details):
|
|
networks = [
|
|
('eth', 1),
|
|
# ('exp', 2),
|
|
# ('rop', 3),
|
|
('rin', 4),
|
|
('ubq', 8),
|
|
# ('rsk', 30),
|
|
('kov', 42),
|
|
('etc', 61),
|
|
]
|
|
|
|
# Fetch list of tokens already included in Trezor Core
|
|
r = requests.get('https://raw.githubusercontent.com/trezor/trezor-core/master/src/apps/ethereum/tokens.py')
|
|
d = {}
|
|
exec(r.text, d)
|
|
|
|
# TODO 'Qmede...' can be removed after ipfs_hash is being generated into tokens.py
|
|
ipfs_hash = d.get('ipfs_hash') or 'QmedefcF1fecLVpRymJJmyJFRpJuCTiNfPYBhzUdHPUq3T'
|
|
|
|
infos = {}
|
|
for n in networks:
|
|
# print("Updating info about erc20 tokens for", n[0])
|
|
url = 'https://gateway.ipfs.io/ipfs/%s/%s.json' % (ipfs_hash, n[0])
|
|
r = requests.get(url)
|
|
infos[n[0]] = r.json()
|
|
|
|
supported = []
|
|
for t in d['tokens']:
|
|
token = t[2]
|
|
# print('Updating', token)
|
|
|
|
try:
|
|
network = [ n[0] for n in networks if n[1] == t[0] ][0]
|
|
except:
|
|
raise Exception("Unknown network", t[0], "for erc20 token", token)
|
|
|
|
try:
|
|
info = [ i for i in infos[network] if i['symbol'] == token ][0]
|
|
except:
|
|
raise Exception("Unknown details for erc20 token", token)
|
|
|
|
key = "erc20:%s:%s" % (network, token)
|
|
supported.append(key)
|
|
out = details['coins'].setdefault(key, {})
|
|
out['type'] = 'erc20'
|
|
out['network'] = network
|
|
out['address'] = info['address']
|
|
|
|
set_default(out, 'shortcut', token)
|
|
set_default(out, 'name', info['name'])
|
|
set_default(out, 't1_enabled', 'yes')
|
|
set_default(out, 't2_enabled', 'yes')
|
|
set_default(out, 'links', {})
|
|
|
|
if info['website']:
|
|
out['links']['Homepage'] = info['website']
|
|
if info.get('social', {}).get('github', None):
|
|
out['links']['Github'] = info['social']['github']
|
|
|
|
update_marketcap(out, out.get('coinmarketcap_alias', token))
|
|
|
|
check_unsupported(details, 'erc20:', supported)
|
|
|
|
def update_ethereum(details):
|
|
# print('Updating Ethereum ETH')
|
|
out = details['coins'].setdefault('coin2:ETH', {})
|
|
out['type'] = 'coin'
|
|
set_default(out, 'shortcut', 'ETH')
|
|
set_default(out, 'name', 'Ethereum')
|
|
set_default(out, 't1_enabled', 'yes')
|
|
set_default(out, 't2_enabled', 'yes')
|
|
update_marketcap(out, 'ethereum')
|
|
|
|
def update_mosaics(details):
|
|
r = requests.get('https://raw.githubusercontent.com/trezor/trezor-mcu/master/firmware/nem_mosaics.json')
|
|
supported = []
|
|
for mosaic in r.json():
|
|
# print('Updating', mosaic['name'], mosaic['ticker'])
|
|
|
|
key = "mosaic:%s" % mosaic['ticker'].strip()
|
|
supported.append(key)
|
|
out = details['coins'].setdefault(key, {})
|
|
out['type'] = 'mosaic'
|
|
set_default(out, 'shortcut', mosaic['ticker'].strip())
|
|
set_default(out, 'name', mosaic['name'])
|
|
set_default(out, 't1_enabled', 'yes')
|
|
set_default(out, 't2_enabled', 'yes')
|
|
|
|
update_marketcap(out, out.get('coinmarketcap_alias', out['name']))
|
|
|
|
check_unsupported(details, 'mosaic:', supported)
|
|
|
|
def check_missing_details(details):
|
|
for k in details['coins'].keys():
|
|
coin = details['coins'][k]
|
|
|
|
if 'links' not in coin:
|
|
print("%s: Missing links" % k)
|
|
continue
|
|
if 'Homepage' not in coin['links']:
|
|
print("%s: Missing homepage" % k)
|
|
if coin['t1_enabled'] not in ('yes', 'no', 'planned', 'in progress'):
|
|
print("%s: Unknown t1_enabled" % k)
|
|
if coin['t2_enabled'] not in ('yes', 'no', 'planned', 'in progress'):
|
|
print("%s: Unknown t2_enabled" % k)
|
|
if 'TREZOR Wallet' in coin['links'] and coin['links']['TREZOR Wallet'] != 'https://wallet.trezor.io':
|
|
print("%s: Strange URL for TREZOR Wallet" % k)
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
details = json.load(open('coins_details.json', 'r'))
|
|
except FileNotFoundError:
|
|
details = {'coins': {}, 'info': {}}
|
|
|
|
update_coins(details)
|
|
update_erc20(details)
|
|
update_ethereum(details)
|
|
update_mosaics(details)
|
|
update_info(details)
|
|
check_missing_details(details)
|
|
|
|
print(json.dumps(details['info'], sort_keys=True, indent=4))
|
|
json.dump(details, open('coins_details.json', 'w'), sort_keys=True, indent=4)
|