gustavosantos/GTI_GERENCIADOR_FISCAL
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: complete pre-ai local completion v3

Browse Source
This commit is contained in:
OpenCode
2026-09-21 16:30:27 -03:00
parent 6731eeca08
commit 84b08c2150
41 changed files with 4141 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/rollback_field.py
+8 -5
View File
@@ -7,7 +7,7 @@ sys.path.insert(0, str(Path(__file__).resolve().parent))
from dotenv import load_dotenv
import local_completion as lc
ap = argparse.ArgumentParser(); ap.add_argument("field"); ap.add_argument("--db", nargs="*"); ap.add_argument("--plan", action="store_true", help="only the codes listed in rollback_plan.json"); a = ap.parse_args()
ap = argparse.ArgumentParser(); ap.add_argument("field"); ap.add_argument("--db", nargs="*"); ap.add_argument("--run-id"); ap.add_argument("--product"); ap.add_argument("--plan", action="store_true", help="only the codes listed in rollback_plan.json"); a = ap.parse_args()
plan = json.loads((Path(__file__).resolve().parents[1] / "docs" / "runs" / "local_completion" / "rollback_plan.json").read_text()) if a.plan else None
load_dotenv(lc.ROOT / ".env")
ex = lc.load_executor(); al = lc.allowlist()
@@ -16,7 +16,7 @@ for name in a.db or list(al):
path = al[name]; todo = []
for line in gzip.open(lc.STATE / f"{name}_write_audit.ndjson.gz", "rt", encoding="utf-8"):
r = json.loads(line)
if r.get("outcome") == "SUCCEEDED" and a.field in r["fields"] and (plan is None or r["product_code"] in plan.get(name, {}).get(a.field, [])):
if r.get("outcome") == "SUCCEEDED" and (a.run_id is None or r.get("run_id") == a.run_id) and (a.product is None or r.get("product_code") == a.product) and a.field in r["fields"] and (plan is None or r["product_code"] in plan.get(name, {}).get(a.field, [])):
todo.append((r["product_code"], r["fields"][a.field]["before"], r["fields"][a.field]["after"]))
conn = lc.connect(path, rw=True); ok = fail = 0; audit = []
for code, before, after in todo:
@@ -24,9 +24,12 @@ for name in a.db or list(al):
res = ex.execute_guarded_product_update(conn, database_path=path, product_code=code, changes={a.field: (after, before)}, now=datetime.now(timezone.utc).isoformat(), restore_before=True)
audit.append({**res, "outcome": "ROLLED_BACK"}); ok += 1
except Exception as e: # noqa: BLE001
audit.append({"product_code": code, "outcome": "FAILED", "error": str(e)}); fail += 1
message = str(e)
outcome = "CONFLICT" if "expected" in message.lower() or "changed" in message.lower() else "FAILED"
audit.append({"product_code": code, "outcome": outcome, "error": message}); fail += 1
conn.close()
with gzip.open(lc.STATE / f"{name}_rollback_{a.field}{"_plan" if plan else ""}.ndjson.gz", "wt", encoding="utf-8") as f:
suffix = "_plan" if plan else ""
with gzip.open(lc.STATE / f"{name}_rollback_{a.field}{suffix}.ndjson.gz", "wt", encoding="utf-8") as f:
for r in audit: f.write(json.dumps(r, default=str) + "\n")
# independent verification against the BEFORE snapshot
snap = pickle.load(gzip.open(lc.STATE / f"{name}_before.pkl.gz", "rb")); cols = snap["columns"]; i = cols.index(a.field)
@@ -36,4 +39,4 @@ for name in a.db or list(al):
mism = sum(1 for k, r in snap["rows"].items() if (plan is None or k in plan.get(name, {}).get(a.field, [])) and k in live and any(n(v) != n(r[i]) for v in live[k]))
out[name] = {"rollback_planned": len(todo), "rolled_back": ok, "failed": fail, "column_mismatch_vs_before_snapshot": mism, "rows_snapshot": len(snap["rows"])}
print(name, out[name], flush=True)
json.dump(out, open(lc.STATE / f"rollback_{a.field}{"_plan" if plan else ""}_summary.json", "w"), indent=1)
json.dump(out, open(lc.STATE / f"rollback_{a.field}{suffix}_summary.json", "w"), indent=1)