#!/usr/bin/env python3
"""Wikidata identity check — EN author -> life dates + OFFICIAL RU name.

Usage: tools/wd.py "Erich Neumann" [top]

Resolves the Q-entity, prints birth/death (P569/P570), EN/DE/RU labels, RU aliases
(P725-equivalent via aliases block), and the P50 works list (often sparse).
Primary value: (1) disambiguation via life dates, (2) the canonical RU label — the
single most reliable RU spelling for RSL/libgen queries.
"""
import json, sys, urllib.parse, urllib.request

UA = "research-tools/1.0 (jung booklist research; contact: dmitry@kokorin.org)"

def api(params):
    url = "https://www.wikidata.org/w/api.php?" + urllib.parse.urlencode(params)
    req = urllib.request.Request(url, headers={"User-Agent": UA})
    with urllib.request.urlopen(req, timeout=40) as r:
        return json.load(r)

def main():
    if len(sys.argv) < 2:
        sys.exit(__doc__)
    name = " ".join(sys.argv[1:-1]) if len(sys.argv) > 2 and sys.argv[-1].isdigit() else " ".join(sys.argv[1:])
    top = int(sys.argv[-1]) if len(sys.argv) > 2 and sys.argv[-1].isdigit() else 3
    d = api({"action": "wbsearchentities", "search": name, "language": "en",
             "limit": top, "format": "json"})
    for r in d.get("search", []):
        qid = r["id"]
        label = r.get("label")
        desc = (r.get("description") or "")[:100]
        try:
            ed = api({"action": "wbgetentities", "ids": qid, "format": "json"})
            ent = ed["entities"][qid]
            c = ent.get("claims", {})
            bd = c.get("P569", [{}])[0].get("mainsnak", {}).get("datavalue", {}).get("value", {}).get("time", "?")
            dd = c.get("P570", [{}])[0].get("mainsnak", {}).get("datavalue", {}).get("value", {}).get("time", "?")
            labels = {k: v["value"] for k, v in ent.get("labels", {}).items() if k in ("en", "de", "ru")}
            ru_al = [a["value"] for a in ent.get("aliases", {}).get("ru", [])]
            works = [w["mainsnak"]["datavalue"]["value"].get("id") for w in c.get("P50", [])]
            print(f"{qid} | {label} | {desc}")
            print(f"    born/died: {bd} / {dd}")
            print(f"    labels: {labels}")
            if ru_al: print(f"    RU aliases: {ru_al}")
            if works: print(f"    P50 works: {works[:15]}")
        except Exception as e:
            print(f"{qid} | {label} | (entity fetch failed: {e})")
        print()

if __name__ == "__main__":
    main()