diff --git a/src/app.py b/src/app.py index 39662cf..57c9d39 100644 --- a/src/app.py +++ b/src/app.py @@ -62,7 +62,7 @@ class App: packs_remaining, packs_last_regen_at, cards = user.pull() content = "" for card in cards: - content += "- [%s] [%s](%s)\n" % (card.rarity, card.name, card.url) + content += "- %s [%s](%s)\n" % (card.rarity.icon, card.name, card.url) DiscordClient.send_message(content) diff --git a/src/services/wiki_masters.py b/src/services/wiki_masters.py index 75b9be5..29a3ba4 100644 --- a/src/services/wiki_masters.py +++ b/src/services/wiki_masters.py @@ -23,6 +23,15 @@ class Rarity(StrEnum): UltraRare = "UR" Legendary = "L" + def icon(self) -> str: + return { + Self.Common: "⚪", + Self.LessCommon: "🟢", + Self.Rare: "🔵", + Self.SuperRare: "🟣", + Self.UltraRare: "🟠", + Self.Legendary: "🌟", + }[self] @dataclass class CardDataSubset: