"""评析目标名单：主榜与观察档转点后缀式、去重、名次；写入可注入。"""
import review_targets as rt


def t(name, cond):
    print(("  ok  " if cond else "  FAIL") + name)
    assert cond, name


def main():
    snap = {"main": [{"code": "SH600519", "name": "贵州茅台"}, {"code": "SZ300750", "name": "宁德时代"}, {"code": "SH600519"}],
            "observe": [{"code": "SZ000001", "name": "平安银行"}, {"code": "bad"}]}
    rows = rt.build_rows("2026-09-08", snap["main"], snap["observe"])
    t("点后缀式且去重", [r["ts_code"] for r in rows] == ["600519.SH", "300750.SZ", "000001.SZ"])
    t("来源与名次", rows[0]["source"] == "main" and rows[0]["rank"] == 1 and rows[2]["source"] == "observe" and rows[2]["rank"] == 1)
    got = {}
    out = rt.persist("2026-09-08", snap, write=lambda ds, rs: got.setdefault(ds, rs))
    t("写入可注入且读数正确", out["rows"] == 3 and out["main"] == 2 and len(got["2026-09-08"]) == 3)
    t("认不出的代码丢弃", all(len(r["ts_code"]) == 9 for r in rows))
    big = {"main": [{"code": f"SH60{i:04d}"} for i in range(150)], "observe": [{"code": f"SZ30{i:04d}"} for i in range(40)]}
    out2 = rt.persist("2026-09-09", big, write=lambda ds, rs: got.setdefault(ds, rs))
    t("主榜取前一百、观察档取前二十", out2["rows"] == rt.TOP_MAIN + rt.TOP_OBS and out2["main"] == rt.TOP_MAIN)
    print("ALL OK — 评析目标名单")


if __name__ == "__main__":
    main()