Add exception-catchers that point out exactly which game or channel c…

…auses a crash
DevilXD · Feb 17, 2024 · a536dde · a536dde
1 parent dbd3cd8
commit a536dde
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 13 deletions.
diff --git a/channel.py b/channel.py
@@ -244,9 +244,12 @@ async def get_spade_url(self) -> URLType:
         return URLType(match.group(1))
 
     async def get_stream(self) -> Stream | None:
-        response: JsonType = await self._twitch.gql_request(
-            GQL_OPERATIONS["GetStreamInfo"].with_variables({"channel": self._login})
-        )
+        try:
+            response: JsonType = await self._twitch.gql_request(
+                GQL_OPERATIONS["GetStreamInfo"].with_variables({"channel": self._login})
+            )
+        except MinerException as exc:
+            raise MinerException(f"Channel: {self._login}") from exc
         stream_data: JsonType | None = response["data"]["user"]
         if not stream_data:
             return None

diff --git a/twitch.py b/twitch.py
@@ -1710,16 +1710,19 @@ def get_active_drop(self, channel: Channel | None = None) -> TimedDrop | None:
         return None
 
     async def get_live_streams(self, game: Game, *, limit: int = 30) -> list[Channel]:
-        response = await self.gql_request(
-            GQL_OPERATIONS["GameDirectory"].with_variables({
-                "limit": limit,
-                "slug": game.slug,
-                "options": {
-                    "includeRestricted": ["SUB_ONLY_LIVE"],
-                    "systemFilters": ["DROPS_ENABLED"],
-                },
-            })
-        )
+        try:
+            response = await self.gql_request(
+                GQL_OPERATIONS["GameDirectory"].with_variables({
+                    "limit": limit,
+                    "slug": game.slug,
+                    "options": {
+                        "includeRestricted": ["SUB_ONLY_LIVE"],
+                        "systemFilters": ["DROPS_ENABLED"],
+                    },
+                })
+            )
+        except MinerException as exc:
+            raise MinerException(f"Game: {game.slug}") from exc
         if "game" in response["data"]:
             return [
                 Channel.from_directory(self, stream_channel_data["node"], drops_enabled=True)