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

Rewrite TripElement.get_emoji() to use dict lookup

Browse source
This commit is contained in:
Edward Betts 2024-05-19 08:59:27 +02:00
parent 277e991869
commit 1948ab8ff5
1 changed files with 11 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
agenda/types.py
View file

@ -44,16 +44,17 @@ class TripElement:
detail: StrDict detail: StrDict
def get_emoji(self) -> str | None: def get_emoji(self) -> str | None:
"""Emjoji for trip element.""" """Emoji for trip element."""
if self.element_type in ("check-in", "check-out"): emoji_map = {
return emoji.emojize(":hotel:", language="alias") "check-in": ":hotel:",
if self.element_type == "train": "check-out": ":hotel:",
return emoji.emojize(":train:", language="alias") "train": ":train:",
if self.element_type == "flight": "flight": ":airplane:",
return emoji.emojize(":airplane:", language="alias") "ferry": ":ferry:",
if self.element_type == "ferry": }
return emoji.emojize(":ferry:", language="alias")
return None alias = emoji_map.get(self.element_type)
return emoji.emojize(alias, language="alias") if alias else None
def airport_label(airport: StrDict) -> str: def airport_label(airport: StrDict) -> str: