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

simplified binary expose_ws()

Browse Source
This commit is contained in:
Maxim Devaev
2023-06-07 01:21:23 +03:00
parent 7c507810f0
commit 9c694da00c
2 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
kvmd/htserver.py
View File

@@ -132,15 +132,10 @@ _WS_BINARY = "_ws_binary"
_WS_EVENT_TYPE = "_ws_event_type"
def exposed_ws(event_type: (str | int), binary: bool=False) -> Callable:
if binary:
assert isinstance(event_type, int)
else:
assert isinstance(event_type, str)
def exposed_ws(event_type: (str | int)) -> Callable:
def set_attrs(handler: Callable) -> Callable:
setattr(handler, _WS_EXPOSED, True)
setattr(handler, _WS_BINARY, binary)
setattr(handler, _WS_BINARY, isinstance(event_type, int))
setattr(handler, _WS_EVENT_TYPE, str(event_type))
return handler
return set_attrs
@@ -341,8 +336,11 @@ class HttpServer:
def __add_exposed_ws(self, exposed: WsExposed) -> None:
if exposed.binary:
self.__ws_bin_handlers[int(exposed.event_type)] = exposed.handler
event_type = int(exposed.event_type)
assert event_type not in self.__ws_bin_handlers
self.__ws_bin_handlers[event_type] = exposed.handler
else:
assert exposed.event_type not in self.__ws_handlers
self.__ws_handlers[exposed.event_type] = exposed.handler
# =====