Research-Stack/5-Applications/tools-scripts/utils/verify_pre_post_pairs.py

91 lines
3.1 KiB
Python

#!/usr/bin/env python3
# ==============================================================================
# COPYRIGHT NO ONE EVERYWHERE LLC (WYOMING HOLDING COMPANY)
# PROJECT: SOVEREIGN STACK
# This artifact is entirely proprietary and cryptographically proven.
# Open-Source usage requires explicit permission from Brandon Scott Schneider.
# ==============================================================================
import argparse
import json
from pathlib import Path
from typing import Any, Dict, List
from jsonschema import validate
PROJECT_ROOT = Path(__file__).resolve().parent.parent
PRE_SCHEMA = PROJECT_ROOT / "schemas" / "pre_record.schema.json"
POST_SCHEMA = PROJECT_ROOT / "schemas" / "post_record.schema.json"
def load_jsonl(path: Path) -> List[Dict[str, Any]]:
rows: List[Dict[str, Any]] = []
with path.open("r", encoding="utf-8") as handle:
for line in handle:
s = line.strip()
if s:
rows.append(json.loads(s))
return rows
def load_schema(path: Path) -> Dict[str, Any]:
return json.loads(path.read_text(encoding="utf-8"))
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Verify PRE/POST record schema validity and one-to-one pairing.")
parser.add_argument("--pre", required=True, help="Path to pre_records.jsonl")
parser.add_argument("--post", required=True, help="Path to post_records.jsonl")
parser.add_argument("--out", help="Optional output JSON report path")
return parser.parse_args()
def main() -> int:
args = parse_args()
pre_rows = load_jsonl(Path(args.pre))
post_rows = load_jsonl(Path(args.post))
pre_schema = load_schema(PRE_SCHEMA)
post_schema = load_schema(POST_SCHEMA)
for row in pre_rows:
validate(instance=row, schema=pre_schema)
for row in post_rows:
validate(instance=row, schema=post_schema)
pre_ids = [str(r["pre_record_id"]) for r in pre_rows]
post_pre_ids = [str(r["pre_record_id"]) for r in post_rows]
pre_set = set(pre_ids)
post_ref_set = set(post_pre_ids)
missing_post_for_pre = sorted(pre_set - post_ref_set)
orphan_post_refs = sorted(post_ref_set - pre_set)
duplicate_pre = sorted([x for x in pre_set if pre_ids.count(x) > 1])
duplicate_post_ref = sorted([x for x in post_ref_set if post_pre_ids.count(x) > 1])
ok = not (missing_post_for_pre or orphan_post_refs or duplicate_pre or duplicate_post_ref)
report: Dict[str, Any] = {
"ok": ok,
"pre_count": len(pre_rows),
"post_count": len(post_rows),
"missing_post_for_pre": missing_post_for_pre,
"orphan_post_refs": orphan_post_refs,
"duplicate_pre_record_ids": duplicate_pre,
"duplicate_post_pre_refs": duplicate_post_ref,
"pairing_ratio": (len(post_rows) / len(pre_rows)) if pre_rows else 0,
}
if args.out:
out_path = Path(args.out)
out_path.parent.mkdir(parents=True, exist_ok=True)
out_path.write_text(json.dumps(report, indent=2) + "\n", encoding="utf-8")
print(json.dumps(report, indent=2))
return 0 if ok else 2
if __name__ == "__main__":
raise SystemExit(main())