fix: replace raise with logger.exception and break on websocket close/error/unknown in rpc receive loop

Browse Source
This commit is contained in:
retoor 2025-02-17 19:11:45 +00:00
parent 4d09fad32f
commit de5b20bed9
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/snekbot/rpc.py
View File

@ -110,9 +110,11 @@ class RPC:
logger.exception("Error while receiving:", ex)
break
if msg.type == aiohttp.WSMsgType.CLOSED:
raise Exception("WebSocket closed.")
logger.exception("WebSocket closed.")
break
elif msg.type == aiohttp.WSMsgType.ERROR:
raise Exception("WebSocket error:", msg)
logger.exception("WebSocket error.")
break
elif msg.type == aiohttp.WSMsgType.TEXT:
if self.current_call_id and not msg.json().get('callId') != self.current_call_id:
@ -126,5 +128,6 @@ class RPC:
logger.exception(ex)
break
else:
raise Exception("Unexpected message type.")
logger.exception("Unexpected message type.")
break
return None