Show individual rail legs as separate events

Closes: #64
This commit is contained in:
Edward Betts 2023-11-06 16:59:51 +00:00
parent 84ea916747
commit efbbf3e350

View file

@ -38,6 +38,25 @@ def get(
] ]
def get_trains(from_date: date, filepath: str) -> list[Event]:
"""Get train events."""
events: list[Event] = []
for item in yaml.safe_load(open(filepath)):
if item["depart"].date() < from_date:
continue
events += [
Event(
date=leg["depart"],
end_date=leg["arrive"],
name="transport",
title=f'train from {leg["from"]} to {leg["to"]}',
url=item.get("url"),
)
for leg in item["legs"]
]
return events
def flight_number(flight: Leg) -> str: def flight_number(flight: Leg) -> str:
"""Flight number.""" """Flight number."""
airline_code = flight["airline"] airline_code = flight["airline"]
@ -49,7 +68,7 @@ def flight_number(flight: Leg) -> str:
def all_events(from_date: date, data_dir: str) -> list[Event]: def all_events(from_date: date, data_dir: str) -> list[Event]:
"""Get all flights and rail journeys.""" """Get all flights and rail journeys."""
trains = get(from_date, "train", os.path.join(data_dir, "trains.yaml")) trains = get_trains(from_date, os.path.join(data_dir, "trains.yaml"))
flights = get( flights = get(
from_date, "flight", os.path.join(data_dir, "flights.yaml"), extra=flight_number from_date, "flight", os.path.join(data_dir, "flights.yaml"), extra=flight_number
) )