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

Add more flags on trip list

Browse source
This commit is contained in:
Edward Betts 2024-05-19 13:33:04 +02:00
parent 1948ab8ff5
commit 5758d3f1d0
1 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
agenda/types.py
View file

@ -241,7 +241,13 @@ class Trip:
) )
if item["type"] == "train": if item["type"] == "train":
for leg in item["legs"]: for leg in item["legs"]:
name = f"{leg['from']} → {leg['to']}" from_country = agenda.get_country(leg["from_station"]["country"])
to_country = agenda.get_country(leg["to_station"]["country"])
assert from_country and to_country
from_flag = from_country.flag
to_flag = to_country.flag
name = f"{leg['from']} {from_flag} → {leg['to']} {to_flag}"
elements.append( elements.append(
TripElement( TripElement(
when=leg["depart"], when=leg["depart"],
@ -251,7 +257,14 @@ class Trip:
) )
) )
if item["type"] == "ferry": if item["type"] == "ferry":
name = f"{item['from']} → {item['to']}" from_country = agenda.get_country(item["from_terminal"]["country"])
to_country = agenda.get_country(item["to_terminal"]["country"])
assert from_country and to_country
from_flag = from_country.flag
to_flag = to_country.flag
name = f"{item['from']} {from_flag} → {item['to']} {to_flag}"
elements.append( elements.append(
TripElement( TripElement(
when=item["depart"], when=item["depart"],