Enhance stop data handling by adding support for multiple YAML override files and improving coordinate accuracy for various stops

1 files changed, 11 insertions, 8 deletions

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")]