diff --git a/inventory.py b/inventory.py index bf216ce..3bb5db9 100644 --- a/inventory.py +++ b/inventory.py @@ -462,7 +462,7 @@ class DropsCampaign: and ( ignore_channel_status or channel.game is not None and channel.game == self.game - or self.game.is_special_events() + or self.game.is_special() ) ) ) diff --git a/twitch.py b/twitch.py index bd2bf9b..e98406c 100644 --- a/twitch.py +++ b/twitch.py @@ -993,7 +993,7 @@ class Twitch: and channel.drops_enabled and channel.game in self.wanted_games # let the campaign ignore all channel-related checks - or campaign.game.is_special_events() + or campaign.game.is_special() ) ): return True diff --git a/utils.py b/utils.py index 7c8c8bf..9fbc169 100644 --- a/utils.py +++ b/utils.py @@ -419,7 +419,7 @@ class AwaitableValue(Generic[_T]): class Game: - SPECIAL_EVENTS_GAME_ID: int = 509663 + SPECIAL_GAME_IDS: set[int] = {509663, 509672} def __init__(self, data: JsonType): self.id: int = int(data["id"]) @@ -454,5 +454,5 @@ class Game: slug_text = re.sub(r'-{2,}', '-', slug_text.strip('-')) return slug_text - def is_special_events(self) -> bool: - return self.id == self.SPECIAL_EVENTS_GAME_ID + def is_special(self) -> bool: + return self.id in self.SPECIAL_GAME_IDS