Alpha stage commit

scripts/finalize_routing_layer.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))
+
+from app.db import init_db, session_scope  # noqa: E402
+from app.pipeline.routing_layer import finalize_routing_layer  # noqa: E402
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Finalize an already imported routing graph.")
+    parser.add_argument("--dataset-id", type=int, default=None, help="Raw OSM PBF dataset id. Defaults to the active routing dataset.")
+    args = parser.parse_args()
+
+    init_db()
+    with session_scope() as session:
+        result = finalize_routing_layer(session, dataset_id=args.dataset_id, progress_callback=_progress)
+        print(result)
+
+
+def _progress(event_type: str, message: str, current: int | None, total: int | None, metadata: dict[str, object] | None) -> None:
+    if current is None and total is None:
+        progress = ""
+    elif total:
+        progress = f" [{current}/{total}]"
+    else:
+        progress = f" [{current}]"
+    print(f"{event_type}{progress}: {message} {metadata or {}}", flush=True)
+
+
+if __name__ == "__main__":
+    main()