Handle TimeoutError in FC2WebSocket._main_loop (#48)

Co-authored-by: user <user@localhost>
HoloArchivists · Sep 18, 2023 · ad4ee15 · ad4ee15
1 parent a6c973f
commit ad4ee15
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/fc2_live_dl/fc2.py b/fc2_live_dl/fc2.py
@@ -83,9 +83,15 @@ async def get_hls_information(self):
 
     async def _main_loop(self):
         while True:
-            msg = await asyncio.wait_for(
-                self._ws.receive_json(), self.heartbeat_interval
-            )
+            try:
+                msg = await asyncio.wait_for(
+                    self._ws.receive_json(), self.heartbeat_interval
+                )
+            except asyncio.TimeoutError:
+                self._logger.debug(f'Got no messages for {self.heartbeat_interval} seconds, sending heartbeat')
+                await self._try_heartbeat()
+                continue
+
             self._logger.trace("<", json.dumps(msg)[:100])
             if self._output_file is not None:
                 self._output_file.write("< ")