From 7e28c6e3dd9ea27babdb10cb1246f3556b3b90aa Mon Sep 17 00:00:00 2001 From: Joscha Date: Fri, 27 Jul 2018 19:22:04 +0000 Subject: [PATCH] Deal with closed ws connection while sending --- yaboli/connection.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/yaboli/connection.py b/yaboli/connection.py index 0258ae3..97a6bae 100644 --- a/yaboli/connection.py +++ b/yaboli/connection.py @@ -48,7 +48,10 @@ class Connection: wait_for = self._wait_for_response(pid) logging.debug(f"Currently used websocket at self._ws: {self._ws}") - await self._ws.send(json.dumps(packet, separators=(',', ':'))) # minimum size + try: + await self._ws.send(json.dumps(packet, separators=(',', ':'))) # minimum size + except websockets.ConnectionClosed: + raise ConnectionClosed() if await_response: await wait_for