diff --git a/channel.py b/channel.py index 2f33022..3bd07ae 100644 --- a/channel.py +++ b/channel.py @@ -21,6 +21,8 @@ logger = logging.getLogger("TwitchDrops") class Stream: + __slots__ = ("channel", "broadcast_id", "viewers", "drops_enabled", "game", "title") + def __init__( self, channel: Channel, @@ -180,7 +182,7 @@ class Channel: @property def game(self) -> Game | None: - if self._stream is not None and self._stream.game is not None: + if self._stream is not None: return self._stream.game return None diff --git a/inventory.py b/inventory.py index de8c10f..ebbf15e 100644 --- a/inventory.py +++ b/inventory.py @@ -25,6 +25,8 @@ def remove_dimensions(url: URLType) -> URLType: class Benefit: + __slots__ = ("id", "name", "image_url") + def __init__(self, data: JsonType): benefit_data: JsonType = data["benefit"] self.id: str = benefit_data["id"]