Move code around a bit

This commit is contained in:
Edward Betts 2024-04-16 12:08:14 +01:00
parent b1507702cf
commit ab60721e15
2 changed files with 47 additions and 25 deletions

View file

@ -1,6 +1,7 @@
"""Trips."""
import os
import typing
from datetime import date, datetime, time
from zoneinfo import ZoneInfo
@ -121,6 +122,29 @@ def build_trip_list(
return [trip for _, trip in sorted(trips.items())]
def add_coordinates_for_unbooked_flights(
routes: list[StrDict], coordinates: list[StrDict]
) -> None:
"""Add coordinates for flights that haven't been booked yet."""
if not (
any(route["type"] == "unbooked_flight" for route in routes)
and not any(pin["type"] == "airport" for pin in coordinates)
):
return
data_dir = flask.current_app.config["PERSONAL_DATA"]
airports = typing.cast(dict[str, StrDict], travel.parse_yaml("airports", data_dir))
lhr = airports["LHR"]
coordinates.append(
{
"name": lhr["name"],
"type": "airport",
"latitude": lhr["latitude"],
"longitude": lhr["longitude"],
}
)
def collect_trip_coordinates(trip: Trip) -> list[StrDict]:
"""Extract and deduplicate airport and station coordinates from trip."""
stations = {}