forgot that TimeoutError is not the same as WebSocketException
This commit is contained in:
Casey 2022-08-31 00:27:34 +03:00
parent 9d672dbbba
commit 0e59717c91
Signed by: hkc
GPG Key ID: F0F6CFE11CDB0960
1 changed files with 1 additions and 1 deletions

View File

@ -21,6 +21,6 @@ async def websocket_source(
raise Exception(event["error"])
if event["event"] == "update":
yield Status.from_dict(loads(event["payload"]))
except WebSocketException:
except (WebSocketException, TimeoutError):
if not reconnect:
raise