feat: complete pre-ai local completion v3

2026-09-21 16:30:27 -03:00
parent 6731eeca08
commit 84b08c2150
41 changed files with 4141 additions and 9 deletions
@@ -0,0 +1,56 @@
+"""Review-only company questionnaire CLI; it never imports or writes Firebird."""
+import argparse
+import csv
+import json
+from pathlib import Path
+
+from central.app.ai.questionnaire import build_questionnaire, group_questions, render_csv, render_json, validate_answers
+
+
+def _load(path):
+    if Path(path).suffix.lower() == ".json":
+        return json.loads(Path(path).read_text(encoding="utf-8"))["questions"]
+    with Path(path).open(newline="", encoding="utf-8-sig") as stream:
+        return list(csv.DictReader(stream))
+
+
+def main(argv=None):
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--questions", default="docs/LOCAL_FISCAL_CONTEXT_QUESTIONS_V2.json")
+    parser.add_argument("--answers", help="JSON object keyed by QUESTION_ID; omit for guided mode")
+    parser.add_argument("--output", default="docs/OPERATOR_CONTEXT_ANSWERS_V3.json")
+    parser.add_argument("--format", choices=("json", "csv"), default="json")
+    parser.add_argument("--confirm", action="store_true", help="write the reviewed artifact")
+    args = parser.parse_args(argv)
+    rows = _load(args.questions)
+    databases = sorted({str(q.get("DATABASE") or "").strip() for q in rows if q.get("DATABASE")})
+    existing = {q.get("DATABASE") for q in rows if q.get("KEY") == "store_ipi_profile"}
+    rows += [{"QUESTION_ID": f"QIPI-{name}", "DATABASE": name, "SCOPE": "COMPANY", "KEY": "store_ipi_profile",
+              "VALID_OPTIONS": ["POPULATE_REFERENCE_RATE", "LEAVE_NOT_APPLICABLE"],
+              "QUESTION": "Guardar a aliquota TIPI como referencia no cadastro?", "PRODUCTS_AFFECTED": 0, "CELLS": 0}
+             for name in databases if name not in existing]
+    questions = build_questionnaire([q for q in rows if str(q.get("CATEGORY", "")).upper() == "C" or q.get("KEY") == "store_ipi_profile"])
+    groups = group_questions(questions)
+    if args.answers:
+        answers = json.loads(Path(args.answers).read_text(encoding="utf-8"))
+    else:
+        answers = {}
+        for group in groups:
+            print(f"\nEMPRESA: {group['database']}\nPERGUNTA: {group['question']}\nPOR QUE PRECISA: {group['consequence']}\nPRODUTOS AFETADOS: {group['affected']['products']}\nCÉLULAS AFETADAS: {group['affected']['cells']}\nOPÇÕES: {', '.join(group['options'])}")
+            value = input("RESPOSTA (ou CANCELAR): ").strip()
+            if value.upper() == "CANCELAR":
+                print("cancelado; nada foi persistido")
+                return 0
+            for question_id in group["question_ids"]:
+                answers[question_id] = value
+    result = validate_answers(questions, answers)
+    print(json.dumps({"questions": len(questions), "groups": len(groups), "entries": len(result["entries"]), "rollback_plan": result["rollback_plan"]}, ensure_ascii=False))
+    if not args.confirm:
+        print("review only; nothing was persisted")
+        return 0
+    Path(args.output).write_text(render_csv(result) if args.format == "csv" else render_json(result), encoding="utf-8")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())