Better airport labels

This commit is contained in:
Edward Betts 2024-05-18 14:22:08 +02:00
parent d5a92c9a8e
commit f1a472a944

View file

@ -43,6 +43,12 @@ class TripElement:
detail: StrDict
def airport_label(airport: StrDict) -> str:
"""Airport label: name and iata."""
name = airport.get("alt_name") or airport["city"]
return f"{name} ({airport['iata']})"
@dataclass
class Trip:
"""Trip."""
@ -204,12 +210,9 @@ class Trip:
for item in self.travel:
if item["type"] == "flight":
flight_from = item["from_airport"]
flight_to = item["to_airport"]
name = (
"✈️ "
+ f"{flight_from['name']} ({flight_from['iata']}) -> "
+ f"{flight_to['name']} ({flight_to['iata']})"
f"✈️ {airport_label(item['from_airport'])}"
+ f"{airport_label(item['to_airport'])}"
)
elements.append(