Whose wallet is this

export VIKE_API_KEY=...
python oc-whose-wallet.py                  # the default address is vitalik.eth's
python oc-whose-wallet.py 0x28C6c06298d514Db089934071355E5743bf21d60
python oc-whose-wallet.py --dry-run        # prints the requests it would make, calls nothing

"""Put a name to an Ethereum address: label, who said so, and its ENS name.

Run:  VIKE_API_KEY=... python oc-whose-wallet.py [0xADDRESS]    (--dry-run lists the requests and makes none)
"""
import os
import sys
from urllib.parse import urlencode

NAME = "oc-whose-wallet"
BASE = "https://data.vike.io"
DRY = "--dry-run" in sys.argv
ARGS = [a for a in sys.argv[1:] if not a.startswith("--")]
WALLET = ARGS[0] if ARGS else "0xd8dA6BF26964aF9D7eEd9e03E53415D37aA96045"


def fail(why):
    sys.exit(f"{NAME}: {why}")


def call(path, method="GET", **params):
    url = BASE + path + ("?" + urlencode(params) if params else "")
    if DRY:
        print(method, url)
        return None
    key = os.environ.get("VIKE_API_KEY") or fail("set VIKE_API_KEY")
    import requests

    r = requests.request(method, url, headers={"X-API-Key": key}, timeout=60)
    if r.status_code != 200:
        fail(f"{method} {path} answered HTTP {r.status_code}: {r.text[:200]}")
    return r.json()


def main():
    # A POST with no body; one address is a one-element list. An address that is not valid comes back as [].
    batch = call("/v1/onchain/labels/batch", method="POST", addresses=WALLET)
    # Who said it: the consensus plus every source's own claim. A thin list can mean a source is rebuilding.
    prov = call(f"/v1/onchain/labels/{WALLET}/provenance")
    # Reverse ENS. source says which path answered: local (our store) or rpc (a fresh lookup).
    ens = call(f"/v1/onchain/wallets/{WALLET}/ens")
    if DRY:
        return
    if not batch:
        fail(f"{WALLET} is not a valid address")
    row = batch[0]
    print(f"{WALLET}")
    if row["found"]:
        print(f"  label    {row['entity']} ({row['wallet_type']}), confidence {row['confidence']}")
        print(f"  30 days  ${row['volume_30d_usd']:,.0f} volume, {row['tx_30d']} txs, risk flag {row['risk_flag'] or 'none'}")
    else:
        print("  label    none on record")
    claims = "; ".join(s["label"] for s in prov.get("sources", []))
    print(f"  sources  {prov.get('source_count', 0)}: {claims or 'none name it'}")
    name = ens["ens"]
    if name:
        print(f"  ens      {name}")
    elif ens["source"] == "rpc":
        print("  ens      none found, but an rpc miss may only mean the resolvers were unreachable: retry before concluding")
    else:
        print("  ens      no name set")


if __name__ == "__main__":
    try:
        main()
    except Exception as e:  # a shape the recipe does not expect fails by name, not by traceback
        fail(f"{type(e).__name__}: {e}")