Whoopsie
forgot that TimeoutError is not the same as WebSocketException
This commit is contained in:
parent
9d672dbbba
commit
0e59717c91
|
@ -21,6 +21,6 @@ async def websocket_source(
|
||||||
raise Exception(event["error"])
|
raise Exception(event["error"])
|
||||||
if event["event"] == "update":
|
if event["event"] == "update":
|
||||||
yield Status.from_dict(loads(event["payload"]))
|
yield Status.from_dict(loads(event["payload"]))
|
||||||
except WebSocketException:
|
except (WebSocketException, TimeoutError):
|
||||||
if not reconnect:
|
if not reconnect:
|
||||||
raise
|
raise
|
||||||
|
|
Loading…
Reference in New Issue