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."""
import os
import typing
from datetime import date, timedelta
import agenda
import agenda.conference
import agenda.data
import agenda.travel
import agenda.trip
import agenda.types
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)
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"])