handle closing ws event

1 files changed, 6 insertions, 0 deletions

diff --git a/kvmd/clients/kvmd.py b/kvmd/clients/kvmd.py
index 84576fb2..b4585bcc 100644
--- a/kvmd/clients/kvmd.py
+++ b/kvmd/clients/kvmd.py
@@ -153,6 +153,8 @@ class KvmdClientWs:
                     msg = receive_task.result()
                     if msg.type == aiohttp.WSMsgType.TEXT:
                         yield json.loads(msg.data)
+                    elif msg.type == aiohttp.WSMsgType.CLOSE:
+                        await self.__ws.close()
                     elif msg.type == aiohttp.WSMsgType.CLOSED:
                         break
                     else:
@@ -167,6 +169,10 @@ class KvmdClientWs:
                 receive_task.cancel()
             if writer_task:
                 writer_task.cancel()
+            try:
+                await self.__ws.close()
+            except Exception:
+                pass
             self.__communicated = False
 
     async def send_key_event(self, key: str, state: bool) -> None: