aboutsummaryrefslogtreecommitdiff
path: root/data/santiago/download-stops.py
diff options
context:
space:
mode:
authorAriel Costas Guerrero <ariel@costas.dev>2025-11-07 11:04:43 +0100
committerAriel Costas Guerrero <ariel@costas.dev>2025-11-07 11:04:43 +0100
commit86ceab13fa9eab173522d67e5917c90840fb563f (patch)
tree14486b3c8e0ac437b7bb5e5b1ad68e484b89a667 /data/santiago/download-stops.py
parent1186795ace3075e309f984182951f47212786444 (diff)
Remove outdated stop overrides and examples for Vigo transit data
Diffstat (limited to 'data/santiago/download-stops.py')
-rw-r--r--data/santiago/download-stops.py196
1 files changed, 0 insertions, 196 deletions
diff --git a/data/santiago/download-stops.py b/data/santiago/download-stops.py
deleted file mode 100644
index 4900c41..0000000
--- a/data/santiago/download-stops.py
+++ /dev/null
@@ -1,196 +0,0 @@
-# /// script
-# requires-python = ">=3.12"
-# dependencies = [
-# "PyYAML>=6.0.2", # For YAML support
-# ]
-# ///
-import json
-import os
-import sys
-import urllib.request
-import yaml # Add YAML support for overrides
-
-OVERRIDES_DIR = "overrides"
-OUTPUT_FILE = "../../src/frontend/public/stops/santiago.json"
-
-def load_stop_overrides(file_path):
- """Load stop overrides from a YAML file"""
- if not os.path.exists(file_path):
- print(f"Warning: Overrides file {file_path} not found")
- return {}
-
- try:
- with open(file_path, 'r', encoding='utf-8') as f:
- overrides = yaml.safe_load(f)
- print(f"Loaded {len(overrides) if overrides else 0} stop overrides")
- return overrides or {}
- except Exception as e:
- print(f"Error loading overrides: {e}", file=sys.stderr)
- return {}
-
-def apply_overrides(stops, overrides):
- """Apply overrides to the stop data and add new stops"""
- # Track existing stop IDs
- existing_stop_ids = {stop.get("stopId") for stop in stops}
-
- # Apply overrides to existing stops
- for stop in stops:
- stop_id = stop.get("stopId")
- if stop_id in overrides:
- override = overrides[stop_id]
-
- # Override name if provided
- if "name" in override:
- stop["name"]["original"] = override["name"]
-
- # Apply or add alternate names
- if "alternateNames" in override:
- for key, value in override["alternateNames"].items():
- stop["name"][key] = value
-
- # Apply location override
- if "location" in override:
- if "latitude" in override["location"]:
- stop["latitude"] = override["location"]["latitude"]
- if "longitude" in override["location"]:
- stop["longitude"] = override["location"]["longitude"]
-
- # Add amenities
- if "amenities" in override:
- stop["amenities"] = override["amenities"]
-
- # Mark stop as hidden if needed
- if "hide" in override:
- stop["hide"] = override["hide"]
-
- # Mark stop as cancelled
- if "cancelled" in override:
- stop["cancelled"] = override["cancelled"]
-
- # Add alert title
- if "title" in override:
- stop["title"] = override["title"]
-
- # Add alert message
- if "message" in override:
- stop["message"] = override["message"]
-
- # Add alternate codes
- if "alternateCodes" in override:
- stop["alternateCodes"] = override["alternateCodes"]
-
- # Add new stops (those with "new: true" parameter)
- new_stops_added = 0
- for stop_id, override in overrides.items():
- # Check if this is a new stop
- if override.get("new") and stop_id not in existing_stop_ids:
- # Ensure stop_id is an integer for consistency
- stop_id_int = int(stop_id) if isinstance(stop_id, str) else stop_id
-
- # Create the new stop
- new_stop = {
- "stopId": stop_id_int,
- "name": {
- "original": override.get("name", f"Stop {stop_id_int}")
- },
- "latitude": override.get("location", {}).get("latitude"),
- "longitude": override.get("location", {}).get("longitude"),
- "lines": override.get("lines", [])
- }
-
- # Add optional fields (excluding the 'new' parameter)
- if "alternateNames" in override:
- for key, value in override["alternateNames"].items():
- new_stop["name"][key] = value
- if "amenities" in override:
- new_stop["amenities"] = override["amenities"]
- if "cancelled" in override:
- new_stop["cancelled"] = override["cancelled"]
- if "title" in override:
- new_stop["title"] = override["title"]
- if "message" in override:
- new_stop["message"] = override["message"]
- if "alternateCodes" in override:
- new_stop["alternateCodes"] = override["alternateCodes"]
-
- stops.append(new_stop)
- new_stops_added += 1
-
- if new_stops_added > 0:
- print(f"Added {new_stops_added} new stops from overrides")
-
- return stops
-
-def main():
- print("Fetching stop list data...")
-
- # Download stop list data
- url = "https://app.tussa.org/tussa/api/paradas"
- body = json.dumps({ "nombre": "" }).encode('utf-8')
- req = urllib.request.Request(url, data=body, headers={'Content-Type': 'application/json'}, method='POST')
-
- try:
- with urllib.request.urlopen(req) as response:
- content = response.read().decode('utf-8')
- data = json.loads(content)
-
- print(f"Downloaded {len(data)} stops")
-
- # Process the data
- processed_stops = []
- for stop in data:
- name = stop.get("nombre", "").strip()
-
- lines_sinoptic: list[str] = [line.get("sinoptico").strip() for line in stop.get("lineas", [])] if stop.get("lineas") else []
- lines_id: list[str] = []
-
- for line in lines_sinoptic:
- line_code = line.lstrip('L')
- lines_id.append(line_code)
-
- processed_stop = {
- "stopId": stop.get("id"),
- "name": {
- "original": name
- },
- "latitude": stop.get("coordenadas").get("latitud"),
- "longitude": stop.get("coordenadas").get("longitud"),
- "lines": lines_id,
- "hide": len(lines_id) == 0
- }
- processed_stops.append(processed_stop)
-
- # Load and apply overrides
- script_dir = os.path.dirname(os.path.abspath(__file__))
- overrides_dir = os.path.join(script_dir, OVERRIDES_DIR)
- # 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")]
- print(f"Removed {len(processed_stops) - len(visible_stops)} hidden stops")
-
- # Sort stops by ID ascending
- visible_stops.sort(key=lambda x: x["stopId"])
-
- output_file = os.path.join(script_dir, OUTPUT_FILE)
-
- with open(output_file, 'w', encoding='utf-8') as f:
- json.dump(visible_stops, f, ensure_ascii=False, indent=2)
-
- print(f"Saved processed stops data to {output_file}")
- return 0
-
- except Exception as e:
- print(f"Error processing stops data: {e}", file=sys.stderr)
- return 1
-
-if __name__ == "__main__":
- sys.exit(main())