parent
993d3b6e3f
commit
9849d84a5e
@ -0,0 +1,91 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Fetch market capitalization data."""
|
||||
import json
|
||||
import os
|
||||
import time
|
||||
|
||||
import requests
|
||||
|
||||
COINMAKETCAP_CACHE = os.path.join(os.path.dirname(__file__), "coinmarketcap.json")
|
||||
COINMARKETCAP_API_BASE = "https://pro-api.coinmarketcap.com/v1/"
|
||||
|
||||
MARKET_CAPS = {}
|
||||
|
||||
|
||||
def call(endpoint, api_key, params=None):
|
||||
url = COINMARKETCAP_API_BASE + endpoint
|
||||
r = requests.get(url, params=params, headers={"X-CMC_PRO_API_KEY": api_key})
|
||||
r.raise_for_status()
|
||||
return r.json()
|
||||
|
||||
|
||||
def init(api_key, refresh=None):
|
||||
global MARKET_CAPS
|
||||
|
||||
force_refresh = refresh is True
|
||||
disable_refresh = refresh is False
|
||||
try:
|
||||
try:
|
||||
mtime = os.path.getmtime(COINMAKETCAP_CACHE)
|
||||
except FileNotFoundError:
|
||||
mtime = 0
|
||||
cache_is_fresh = mtime > time.time() - 3600
|
||||
if disable_refresh or (cache_is_fresh and not force_refresh):
|
||||
print("Using cached market cap data")
|
||||
with open(COINMAKETCAP_CACHE) as f:
|
||||
coinmarketcap_data = json.load(f)
|
||||
else:
|
||||
print("Fetching market cap data")
|
||||
coinmarketcap_data = call(
|
||||
"cryptocurrency/listings/latest",
|
||||
api_key,
|
||||
params={"limit": 5000, "convert": "USD"},
|
||||
)
|
||||
by_id = {str(coin["id"]): coin for coin in coinmarketcap_data["data"]}
|
||||
all_ids = list(by_id.keys())
|
||||
while all_ids:
|
||||
first_100 = all_ids[:100]
|
||||
all_ids = all_ids[100:]
|
||||
time.sleep(1)
|
||||
print("Fetching metadata, {} coins remaining...".format(len(all_ids)))
|
||||
metadata = call(
|
||||
"cryptocurrency/info", api_key, params={"id": ",".join(first_100)}
|
||||
)
|
||||
for coin_id, meta in metadata["data"].items():
|
||||
by_id[coin_id]["meta"] = meta
|
||||
|
||||
with open(COINMAKETCAP_CACHE, "w") as f:
|
||||
json.dump(coinmarketcap_data, f)
|
||||
except Exception as e:
|
||||
raise RuntimeError("market cap data unavailable") from e
|
||||
|
||||
coin_data = {}
|
||||
for coin in coinmarketcap_data["data"]:
|
||||
slug = coin["slug"]
|
||||
platform = coin["meta"]["platform"]
|
||||
market_cap = coin["quote"]["USD"]["market_cap"]
|
||||
if market_cap is not None:
|
||||
coin_data[slug] = int(market_cap)
|
||||
if platform is not None and platform["name"] == "Ethereum":
|
||||
address = platform["token_address"].lower()
|
||||
coin_data[address] = int(market_cap)
|
||||
|
||||
MARKET_CAPS = coin_data
|
||||
|
||||
return coin_data
|
||||
|
||||
|
||||
def marketcap(coin):
|
||||
cap = None
|
||||
if coin["type"] == "erc20":
|
||||
address = coin["address"].lower()
|
||||
return MARKET_CAPS.get(address)
|
||||
|
||||
if "coinmarketcap_alias" in coin:
|
||||
cap = MARKET_CAPS.get(coin["coinmarketcap_alias"])
|
||||
if cap is None:
|
||||
slug = coin["name"].replace(" ", "-").lower()
|
||||
cap = MARKET_CAPS.get(slug)
|
||||
if cap is None:
|
||||
cap = MARKET_CAPS.get(coin["shortcut"].lower())
|
||||
return cap
|
Loading…
Reference in new issue