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

Show flights and trains in different colours

Browse source
This commit is contained in:
Edward Betts 2024-01-12 19:52:00 +00:00
parent 4b62ec96dc
commit 4e719a07ab
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
web_view.py
View file

@ -303,20 +303,22 @@ def latlon_tuple(stop: StrDict) -> tuple[float, float]:
def get_trip_routes(
trip: Trip,
) -> list[tuple[tuple[float, float], tuple[float, float]]]:
) -> list[tuple[str, tuple[float, float], tuple[float, float]]]:
routes = []
for t in trip.travel:
if t["type"] == "flight":
if "from_airport" not in t or "to_airport" not in t:
continue
fly_from, fly_to = t["from_airport"], t["to_airport"]
routes.append((latlon_tuple(fly_from), latlon_tuple(fly_to)))
routes.append(("flight", latlon_tuple(fly_from), latlon_tuple(fly_to)))
else:
assert t["type"] == "train"
for leg in t["legs"]:
train_from, train_to = leg["from_station"], leg["to_station"]
routes.append((latlon_tuple(train_from), latlon_tuple(train_to)))
routes.append(
("train", latlon_tuple(train_from), latlon_tuple(train_to))
)
return routes