cli.py

07-Automation/partner_pipeline/cli.py

1. """Command-line entry point: one subcommand per pipeline stage."""
3. from __future__ import annotations
5. import argparse
7. from .audit import cmd_outreach_audit, cmd_outreach_pull_list
8. from .baseline import cmd_seed_baseline
9. from .census import cmd_reconcile_deals, cmd_refresh_xml
10. from .classify import cmd_classify, cmd_verify
11. from .history import cmd_ingest
12. from .roster_workbook import cmd_refresh_workbook
13. from .route import cmd_route
14. from .stubs import cmd_write_stubs
17. def main():
18. p = argparse.ArgumentParser(description="Weekly Partner Client Pipeline: one subcommand per stage.")
19. sub = p.add_subparsers(dest="cmd", required=True)
21. s = sub.add_parser("seed-baseline", help="one-time baseline load from the roster workbook")
22. s.add_argument("--sheet", required=True, help="xlsx or csv")
23. s.add_argument("--tab", help="tab name inside an xlsx workbook (default: active tab)")
24. s.add_argument("--as-of", help="date the sheet's totals run through (YYYY-MM-DD): required to write")
25. s.add_argument("--map", action="append", help='override a column: --map "ltv=Total Spend"')
26. s.add_argument("--dry-run", action="store_true", help="show mapping + sample rows, write nothing")
27. s.add_argument("--force", action="store_true", help="replace existing baselines")
28. s.set_defaults(func=cmd_seed_baseline)
30. i = sub.add_parser("ingest", help="merge raw orders.fetch page dump(s) into the history")
31. i.add_argument("--orders", required=True, nargs="+", help="one or more raw JSON page files")
32. i.set_defaults(func=cmd_ingest)
34. c = sub.add_parser("classify", help="compute lifetime metrics from history, classify, export")
35. c.set_defaults(func=cmd_classify)
37. v = sub.add_parser("verify", help="no-shrink guardrail vs previous run (exit 2 on flags)")
38. v.set_defaults(func=cmd_verify)
40. w = sub.add_parser("write-stubs", help="Stage E: create/upgrade stubs (guarded)")
41. w.add_argument("--plan", required=True, help="the .plan.json produced by classify")
42. w.set_defaults(func=cmd_write_stubs)
44. x = sub.add_parser("refresh-xml", help="Stage F: stamp last_modified in XML map (guarded)")
45. x.add_argument("--plan", required=True, help="the .plan.json produced by classify")
46. x.set_defaults(func=cmd_refresh_xml)
48. rd = sub.add_parser("reconcile-deals",
49. help="absorb a Bitrix deals-pull JSON into census blocks + stubs (guarded)")
50. rd.add_argument("--deals", required=True, nargs="+",
51. help="deals JSON file(s): {\"deals\":[{deal_id,cat,stage,title,created}]}")
52. rd.set_defaults(func=cmd_reconcile_deals)
54. rt = sub.add_parser("route",
55. help="compute weekly outreach candidate lists (read-only; artifacts to 05-Outputs)")
56. rt.set_defaults(func=cmd_route)
58. wbk = sub.add_parser("refresh-workbook",
59. help="Stage F2: regenerate the Drive-swap workbook from the latest updated copy (guarded)")
60. wbk.add_argument("--plan", required=True, help="the .plan.json produced by classify")
61. wbk.add_argument("--base", help="base workbook to carry human columns from "
62. "(default: latest 05-Outputs/partner-runs/Complete-Partner-Clients-List_updated-*.xlsx)")
63. wbk.set_defaults(func=cmd_refresh_workbook)
65. opl = sub.add_parser("outreach-pull-list",
66. help="Stage F5a: list the deals whose stage history must be re-pulled this week")
67. opl.add_argument("--deals", required=True, nargs="+",
68. help="weekly enumeration JSON for cats 24/26/36/16")
69. opl.set_defaults(func=cmd_outreach_pull_list)
71. oa = sub.add_parser("outreach-audit",
72. help="Stage F5b: build the per-pipeline outreach audit workbook (read-only)")
73. oa.add_argument("--deals", required=True, nargs="+",
74. help="enumeration JSON where re-pulled deals also carry a \"history\" array")
75. oa.add_argument("--workbook", help="Drive-swap workbook to resolve Plinth identity from "
76. "(default: latest Complete-Partner-Clients-List_updated-*.xlsx)")
77. oa.set_defaults(func=cmd_outreach_audit)
79. args = p.parse_args()
80. args.func(args)