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

Show trip country list in order visited

Browse source
This commit is contained in:
Edward Betts 2024-01-06 09:21:54 +00:00
parent fd6d3b674b
commit 50127417f0
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
agenda/types.py
View file

@ -55,17 +55,21 @@ class Trip:
return max_date if max_date != date.min else None return max_date if max_date != date.min else None
@property @property
def countries(self) -> set[Country]: def countries(self) -> list[Country]:
"""Trip countries.""" """Countries visited as part of trip, in order."""
found: set[Country] = set() seen: set[str] = set()
items: list[Country] = []
for item in self.conferences + self.accommodation: for item in self.conferences + self.accommodation:
if "country" not in item: if "country" not in item:
continue continue
if item["country"] in seen:
continue
seen.add(item["country"])
country = agenda.get_country(item["country"]) country = agenda.get_country(item["country"])
assert country assert country
found.add(country) items.append(country)
return found return items
@property @property
def countries_str(self) -> str: def countries_str(self) -> str: