edward/agenda
1
0
Fork 0
Code Issues 18 Pull requests Projects Releases Packages Wiki Activity

Include route distances on trips page

Browse source
This commit is contained in:
Edward Betts 2024-04-05 20:17:19 +02:00
parent b5c1e16901
commit fe4bde32ba
2 changed files with 20 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
agenda/trip.py
View file

@ -23,7 +23,9 @@ def load_travel(travel_type: str, data_dir: str) -> list[StrDict]:
return items
def load_trains(data_dir: str) -> list[StrDict]:
def load_trains(
data_dir: str, route_distances: travel.RouteDistances | None = None
) -> list[StrDict]:
"""Load trains."""
trains = load_travel("train", data_dir)
stations = travel.parse_yaml("stations", data_dir)
@ -43,6 +45,12 @@ def load_trains(data_dir: str) -> list[StrDict]:
leg["from_station"] = by_name[leg["from"]]
leg["to_station"] = by_name[leg["to"]]
if route_distances:
travel.add_leg_route_distance(leg, route_distances)
if all("distance" in leg for leg in train["legs"]):
train["distance"] = sum(leg["distance"] for leg in train["legs"])
return trains
@ -68,7 +76,10 @@ def load_flights(data_dir: str) -> list[StrDict]:
return flights
def build_trip_list(data_dir: str | None = None) -> list[Trip]:
def build_trip_list(
data_dir: str | None = None,
route_distances: travel.RouteDistances | None = None,
) -> list[Trip]:
"""Generate list of trips."""
trips: dict[date, Trip] = {}
@ -80,7 +91,8 @@ def build_trip_list(data_dir: str | None = None) -> list[Trip]:
yaml_trip_lookup = {item["trip"]: item for item in yaml_trip_list}
travel_items = sorted(
load_flights(data_dir) + load_trains(data_dir), key=depart_datetime
load_flights(data_dir) + load_trains(data_dir, route_distances=route_distances),
key=depart_datetime,
)
data = {