56 lines
1.5 KiB
Python
Executable file
56 lines
1.5 KiB
Python
Executable file
#!/usr/bin/python3
|
|
"""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=[""])
|
|
|
|
data_dir = config.PERSONAL_DATA
|
|
|
|
trip_list = agenda.trip.build_trip_list(data_dir)
|
|
print(len(trip_list), "trips")
|
|
|
|
coords, routes = agenda.trip.get_coordinates_and_routes(trip_list, data_dir)
|
|
print(len(coords), "coords")
|
|
print(len(routes), "routes")
|
|
|
|
flights = agenda.travel.parse_yaml("flights", data_dir)
|
|
print(len(flights), "flights")
|
|
|
|
trains = agenda.travel.parse_yaml("trains", data_dir)
|
|
print(len(trains), "trains")
|
|
|
|
conferences = agenda.conference.get_list(os.path.join(data_dir, "conferences.yaml"))
|
|
print(len(conferences), "conferences")
|
|
|
|
today = date.today()
|
|
last_year = today - timedelta(days=365)
|
|
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"])
|