diff --git a/mastoposter/sources.py b/mastoposter/sources.py index 9af446a..e2bdbb9 100644 --- a/mastoposter/sources.py +++ b/mastoposter/sources.py @@ -1,3 +1,4 @@ +from asyncio import exceptions from json import loads from typing import AsyncGenerator from urllib.parse import urlencode @@ -21,6 +22,6 @@ async def websocket_source( raise Exception(event["error"]) if event["event"] == "update": yield Status.from_dict(loads(event["payload"])) - except (WebSocketException, TimeoutError): + except (WebSocketException, TimeoutError, exceptions.TimeoutError): if not reconnect: raise