GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-14 02:00:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

better exc handling

Browse Source
This commit is contained in:
Devaev Maxim 2020-03-20 16:00:15 +03:00
parent f85ee6b4bf
commit 1fdc26f136
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
kvmd/apps/vnc/server.py
View File

@ -36,6 +36,7 @@ from ...logging import get_logger
from ... import aiotools from ... import aiotools
from .rfb import RfbError
from .rfb import RfbClient from .rfb import RfbClient
from .kvmd import KvmdClient from .kvmd import KvmdClient
@ -125,8 +126,10 @@ class _Client(RfbClient): # pylint: disable=too-many-instance-attributes
msg = receive_task.result() msg = receive_task.result()
if msg.type == aiohttp.WSMsgType.TEXT: if msg.type == aiohttp.WSMsgType.TEXT:
await self.__process_ws_event(json.loads(msg.data)) await self.__process_ws_event(json.loads(msg.data))
elif msg.type == aiohttp.WSMsgType.CLOSE:
raise RfbError("KVMD closed the wesocket (it may have been stopped)")
else: else:
raise RuntimeError(f"Unknown WS message type: {msg!r}") raise RuntimeError(f"Unhandled WS message type: {msg!r}")
receive_task = None receive_task = None
if writer_task in done: if writer_task in done: