GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2026-01-29 00:51:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

handle closing ws event

Browse Source
This commit is contained in:
Maxim Devaev
2022-07-19 16:25:31 +03:00
parent 54cb5e1fed
commit b16359c53e
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
kvmd/clients/kvmd.py
View File

@@ -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: