From dfdf4a291f3a686496ed9948a31e26ef43879f19 Mon Sep 17 00:00:00 2001 From: Ariel Costas Guerrero Date: Sun, 20 Apr 2025 21:37:10 +0200 Subject: Enhance stop data handling by adding support for multiple YAML override files and improving coordinate accuracy for various stops --- data/download-stops.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) (limited to 'data/download-stops.py') diff --git a/data/download-stops.py b/data/download-stops.py index 957e50d..54c57ed 100644 --- a/data/download-stops.py +++ b/data/download-stops.py @@ -26,11 +26,7 @@ def apply_overrides(stops, overrides): stop_id = stop.get("stopId") if stop_id in overrides: override = overrides[stop_id] - - # Apply name override - if "name" in override: - stop["name"] = override["name"] - + # Apply or add alternate names if "alternateNames" in override: for key, value in override["alternateNames"].items(): @@ -84,9 +80,16 @@ def main(): # Load and apply overrides script_dir = os.path.dirname(os.path.abspath(__file__)) - overrides_file = os.path.join(script_dir, "stop-overrides.yaml") - overrides = load_stop_overrides(overrides_file) - processed_stops = apply_overrides(processed_stops, overrides) + overrides_dir = os.path.join(script_dir, "overrides") + # For each YML/YAML file in the overrides directory, load and apply the overrides + for filename in os.listdir(overrides_dir): + if not filename.endswith(".yml") and not filename.endswith(".yaml"): + continue + + print(f"Loading overrides from {filename}") + overrides_file = os.path.join(overrides_dir, filename) + overrides = load_stop_overrides(overrides_file) + processed_stops = apply_overrides(processed_stops, overrides) # Filter out hidden stops visible_stops = [stop for stop in processed_stops if not stop.get("hide")] -- cgit v1.3