GitHub/TwitchDropsMiner
1
0
Fork 0
mirror of https://github.com/rangermix/TwitchDropsMiner.git synced 2026-06-03 02:49:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add better error handling here

Browse Source
This commit is contained in:
DevilXD
2021-12-04 14:16:43 +01:00
parent fcd9e5b52d
commit 30792d08d5
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
websocket.py
View File

@@ -74,7 +74,10 @@ class Websocket:
if ping_future is not None and not ping_future.done():
ping_future.set_result(message)
elif msg_type == "RESPONSE":
self._recv_dict.pop(message["nonce"]).set_result(message)
try:
self._recv_dict.pop(message["nonce"]).set_result(message)
except KeyError:
logger.exception("Received response for a request we didn't send")
elif msg_type == "RECONNECT":
# We've received a reconnect request
logger.warning("Received a Websocket Reconnect Request")
@@ -84,7 +87,10 @@ class Websocket:
target_topic = message["data"]["topic"]
for topic in self._topics:
if target_topic == topic:
await topic.process(json.loads(message["data"]["message"]))
try:
await topic.process(json.loads(message["data"]["message"]))
except Exception:
logger.exception("Exception in websocket topic process")
break
else:
logger.error(f"Received unknown websocket payload: {message}")