From 0da1a9e497761d8e0a7a33ae9b84a1c566a16ca0 Mon Sep 17 00:00:00 2001 From: Benedict Etzel Date: Fri, 6 Sep 2024 09:36:46 +0200 Subject: [PATCH] feat: support multiple tourists per class --- hearthstone/enums.py | 4 ++-- hearthstone/utils/__init__.py | 22 +++++++++++----------- 2 files changed, 13 insertions(+), 13 deletions(-) diff --git a/hearthstone/enums.py b/hearthstone/enums.py index d0f85fd..f3cc214 100644 --- a/hearthstone/enums.py +++ b/hearthstone/enums.py @@ -1185,9 +1185,9 @@ def is_playable(self): return self != CardClass.WHIZBANG and self.default_hero @property - def visiting_tourist(self): + def visiting_tourists(self): from .utils import VISITING_TOURISTS - return VISITING_TOURISTS.get(self, None) + return VISITING_TOURISTS.get(self, []) @property def name_global(self): diff --git a/hearthstone/utils/__init__.py b/hearthstone/utils/__init__.py index ca88ed7..40e6497 100644 --- a/hearthstone/utils/__init__.py +++ b/hearthstone/utils/__init__.py @@ -114,17 +114,17 @@ } VISITING_TOURISTS = { - CardClass.DEATHKNIGHT: "VAC_503", - CardClass.DRUID: "VAC_340", - CardClass.HUNTER: "VAC_957", - CardClass.MAGE: "VAC_519", - CardClass.PALADIN: "VAC_424", - CardClass.PRIEST: "VAC_501", - CardClass.ROGUE: "VAC_507", - CardClass.SHAMAN: "VAC_437", - CardClass.WARLOCK: "VAC_336", - CardClass.WARRIOR: "VAC_413", - CardClass.DEMONHUNTER: "VAC_450", + CardClass.DEATHKNIGHT: ["VAC_503"], + CardClass.DRUID: ["VAC_340"], + CardClass.HUNTER: ["VAC_957"], + CardClass.MAGE: ["VAC_519"], + CardClass.PALADIN: ["VAC_424"], + CardClass.PRIEST: ["VAC_501"], + CardClass.ROGUE: ["VAC_507"], + CardClass.SHAMAN: ["VAC_437"], + CardClass.WARLOCK: ["VAC_336"], + CardClass.WARRIOR: ["VAC_413"], + CardClass.DEMONHUNTER: ["VAC_450"], }