diff --git a/agenda/calendar.py b/agenda/calendar.py index 56ef6c8..039c546 100644 --- a/agenda/calendar.py +++ b/agenda/calendar.py @@ -61,7 +61,7 @@ def build_events(events: list[Event]) -> list[dict[str, typing.Any]]: continue if e.has_time: - end = e.end_date or e.date + timedelta(hours=1) + end = e.end_date or e.date + timedelta(minutes=30) else: end = (e.end_as_date if e.end_date else e.as_date) + one_day item = { diff --git a/agenda/trip.py b/agenda/trip.py index 06fbf2a..db148e9 100644 --- a/agenda/trip.py +++ b/agenda/trip.py @@ -9,6 +9,12 @@ from agenda import travel from agenda.types import StrDict, Trip +class UnknownStation(Exception): + """Unknown station.""" + + pass + + def load_travel(travel_type: str, data_dir: str) -> list[StrDict]: """Read flight and train journeys.""" items = travel.parse_yaml(travel_type + "s", data_dir) @@ -24,8 +30,10 @@ def load_trains(data_dir: str) -> list[StrDict]: by_name = {station["name"]: station for station in stations} for train in trains: - assert train["from"] in by_name - assert train["to"] in by_name + if train["from"] not in by_name: + raise UnknownStation(train["from"]) + if train["to"] not in by_name: + raise UnknownStation(train["to"]) train["from_station"] = by_name[train["from"]] train["to_station"] = by_name[train["to"]] diff --git a/templates/base.html b/templates/base.html index 7b1cbde..08ccfb7 100644 --- a/templates/base.html +++ b/templates/base.html @@ -7,7 +7,11 @@