Update intranet tools and voucher comparison

2026-06-04 09:00:51 +09:00
parent 2ab74bac88
commit b93289bfa8
55 changed files with 32809 additions and 1063 deletions
@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+import argparse
+import fcntl
+from pathlib import Path
+
+import sys
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from main import engine
+from wehago_compare import (
+    _build_db_state_signature,
+    _get_fast_year_export_row_cache_signature,
+    _load_snapshot_status_map,
+    _project_year_export_row_cache_from_latest_resolved,
+    _rebuild_compare_query_projection,
+    _refresh_year_resolved_sections,
+    _status_row_counts_from_sections,
+    _upsert_snapshot_status,
+)
+
+
+def parse_range(value: str) -> tuple[int, int]:
+    raw = str(value or "").strip()
+    if "-" not in raw:
+        year = int(raw)
+        return year, year
+    left, right = raw.split("-", 1)
+    start_year = int(left)
+    end_year = int(right)
+    if start_year > end_year:
+        start_year, end_year = end_year, start_year
+    return start_year, end_year
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Rebuild WEHAGO compare snapshots and query projections.")
+    parser.add_argument("--years", nargs="*", type=int, default=[])
+    parser.add_argument("--fast-years", nargs="*", type=int, default=[])
+    parser.add_argument("--ranges", nargs="*", default=[])
+    args = parser.parse_args()
+
+    lock_path = Path("/tmp/wehago_compare_compute.lock")
+    lock_path.parent.mkdir(parents=True, exist_ok=True)
+    with lock_path.open("w") as lock_file:
+        fcntl.flock(lock_file.fileno(), fcntl.LOCK_EX)
+        with engine.connect().execution_options(isolation_level="AUTOCOMMIT") as conn:
+            for year in sorted({int(year) for year in args.fast_years if int(year or 0) > 0}):
+                signature = _build_db_state_signature(conn, year, year)
+                print({"step": "fast_year_projection_start", "year": year, "signature": signature}, flush=True)
+                projected = _project_year_export_row_cache_from_latest_resolved(conn, year, signature)
+                if not projected:
+                    print({"step": "fast_year_projection_miss", "year": year}, flush=True)
+                    _refresh_year_resolved_sections(conn, year)
+                else:
+                    _upsert_snapshot_status(
+                        conn,
+                        year,
+                        signature=signature,
+                        state="ready",
+                        row_counts={},
+                        built_now=True,
+                    )
+                selected_signature = _get_fast_year_export_row_cache_signature(conn, year)
+                print(
+                    {
+                        "step": "fast_year_projection_done",
+                        "year": year,
+                        "selected_signature": selected_signature,
+                    },
+                    flush=True,
+                )
+            for year in sorted({int(year) for year in args.years if int(year or 0) > 0}):
+                signature = _build_db_state_signature(conn, year, year)
+                print({"step": "year_snapshot_start", "year": year, "signature": signature}, flush=True)
+                _refresh_year_resolved_sections(conn, year)
+                print({"step": "year_snapshot_done", "year": year}, flush=True)
+            for start_year, end_year in [parse_range(item) for item in args.ranges]:
+                print({"step": "projection_start", "start_year": start_year, "end_year": end_year}, flush=True)
+                counts, snapshot_state = _rebuild_compare_query_projection(engine, conn, start_year, end_year)
+                print(
+                    {
+                        "step": "projection_done",
+                        "start_year": start_year,
+                        "end_year": end_year,
+                        "counts": counts,
+                        "snapshot_state": snapshot_state,
+                    },
+                    flush=True,
+                )
+
+
+if __name__ == "__main__":
+    main()