Alpha stage commit

2026-07-01 23:29:51 +02:00
parent b583bb1233
commit e23387738b
84 changed files with 40807 additions and 326 deletions
--- a/app/serializers.py
+++ b/app/serializers.py
@@ -0,0 +1,130 @@
+from __future__ import annotations
+
+import json
+from typing import Any, Iterable
+
+from app.models import GtfsRoute, GtfsStop, OsmFeature, RouteMatch, RoutePattern
+from app.osm_storage import osm_feature_public_id
+
+
+def feature_collection(features: Iterable[dict[str, Any]]) -> dict[str, Any]:
+    return {"type": "FeatureCollection", "features": list(features)}
+
+
+def gtfs_route_feature(route: GtfsRoute, extra: dict[str, Any] | None = None) -> dict[str, Any] | None:
+    if not route.geometry_geojson:
+        return None
+    props = {
+        "id": route.id,
+        "dataset_id": route.dataset_id,
+        "route_id": route.route_id,
+        "mode": route.mode,
+        "route_scope": route.route_scope,
+        "ref": route.short_name,
+        "name": route.long_name,
+        "operator": route.operator_name,
+        "source": "gtfs",
+    }
+    if extra:
+        props.update(extra)
+    return {"type": "Feature", "geometry": json.loads(route.geometry_geojson), "properties": props}
+
+
+def osm_feature_feature(feature: OsmFeature, extra: dict[str, Any] | None = None) -> dict[str, Any] | None:
+    if not feature.geometry_geojson:
+        return None
+    props = {
+        "id": osm_feature_public_id(feature),
+        "row_id": feature.id,
+        "dataset_id": feature.dataset_id,
+        "osm_type": feature.osm_type,
+        "osm_id": feature.osm_id,
+        "kind": feature.kind,
+        "mode": feature.mode,
+        "route_scope": feature.route_scope,
+        "ref": feature.ref,
+        "name": feature.name,
+        "operator": feature.operator,
+        "network": feature.network,
+        "source": "osm",
+    }
+    if extra:
+        props.update(extra)
+    return {"type": "Feature", "geometry": json.loads(feature.geometry_geojson), "properties": props}
+
+
+def route_pattern_feature(pattern: RoutePattern, extra: dict[str, Any] | None = None) -> dict[str, Any] | None:
+    if not pattern.geometry_geojson:
+        return None
+    props = {
+        "id": pattern.id,
+        "route_pattern_id": pattern.id,
+        "route_ref": pattern.route_ref,
+        "ref": pattern.route_ref,
+        "name": pattern.route_name,
+        "mode": pattern.mode,
+        "route_scope": pattern.route_scope,
+        "operator": pattern.operator_name,
+        "source": "route_layer",
+        "source_kind": pattern.source_kind,
+        "status": pattern.status,
+        "confidence": pattern.confidence,
+        "osm_feature_id": pattern.osm_feature_id,
+        "gtfs_route_id": pattern.gtfs_route_id,
+        "gtfs_shape_id": pattern.gtfs_shape_id,
+    }
+    if extra:
+        props.update(extra)
+    return {"type": "Feature", "geometry": json.loads(pattern.geometry_geojson), "properties": props}
+
+
+def gtfs_stop_feature(stop: GtfsStop) -> dict[str, Any] | None:
+    if stop.lon is None or stop.lat is None:
+        return None
+    return {
+        "type": "Feature",
+        "geometry": {"type": "Point", "coordinates": [stop.lon, stop.lat]},
+        "properties": {
+            "id": stop.id,
+            "dataset_id": stop.dataset_id,
+            "stop_id": stop.stop_id,
+            "name": stop.name,
+            "source": "gtfs",
+        },
+    }
+
+
+def match_row(match: RouteMatch) -> dict[str, Any]:
+    route = match.gtfs_route
+    feature = match.osm_feature
+    return {
+        "id": match.id,
+        "status": match.status,
+        "confidence": match.confidence,
+        "rule_source": match.rule_source,
+        "gtfs": {
+            "id": route.id,
+            "dataset_id": route.dataset_id,
+            "route_id": route.route_id,
+            "mode": route.mode,
+            "route_scope": route.route_scope,
+            "ref": route.short_name,
+            "name": route.long_name,
+            "operator": route.operator_name,
+        },
+        "osm": None
+        if feature is None
+        else {
+            "id": feature.id,
+            "dataset_id": feature.dataset_id,
+            "osm_type": feature.osm_type,
+            "osm_id": feature.osm_id,
+            "mode": feature.mode,
+            "route_scope": feature.route_scope,
+            "ref": feature.ref,
+            "name": feature.name,
+            "operator": feature.operator,
+            "network": feature.network,
+        },
+        "reasons": json.loads(match.reasons_json or "{}"),
+    }