diff --git a/piker/brokers/kraken.py b/piker/brokers/kraken.py index dd048d45..605c5177 100644 --- a/piker/brokers/kraken.py +++ b/piker/brokers/kraken.py @@ -9,6 +9,7 @@ import json import time import trio_websocket +from trio_websocket._impl import ConnectionClosed, DisconnectionTimeout import arrow import asks import numpy as np @@ -247,7 +248,7 @@ async def stream_quotes( }) yield asdict(ohlc) ohlc_last = ohlc - except trio_websocket._impl.ConnectionClosed: + except (ConnectionClosed, DisconnectionTimeout): log.exception("Good job kraken...reconnecting")