Validate airport and station YAML

This commit is contained in:
Edward Betts 2024-05-18 20:37:10 +02:00
parent d4dda44768
commit 277e991869

View file

@ -2,12 +2,15 @@
"""Load YAML data to ensure validity.""" """Load YAML data to ensure validity."""
import os import os
import typing
from datetime import date, timedelta from datetime import date, timedelta
import agenda
import agenda.conference import agenda.conference
import agenda.data import agenda.data
import agenda.travel import agenda.travel
import agenda.trip import agenda.trip
import agenda.types
config = __import__("config.default", fromlist=[""]) config = __import__("config.default", fromlist=[""])
@ -35,3 +38,18 @@ next_year = today + timedelta(days=2 * 365)
events = agenda.events_yaml.read(data_dir, last_year, next_year) events = agenda.events_yaml.read(data_dir, last_year, next_year)
print(len(events), "events") print(len(events), "events")
airports = typing.cast(
dict[str, agenda.types.StrDict], agenda.travel.parse_yaml("airports", data_dir)
)
print(len(airports), "airports")
for airport in airports.values():
assert "country" in airport
assert agenda.get_country(airport["country"])
stations = agenda.travel.parse_yaml("stations", data_dir)
print(len(stations), "stations")
for station in stations:
assert "country" in station
assert agenda.get_country(station["country"])