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

refactoring

Browse Source
This commit is contained in:
Maxim Devaev 2022-01-17 10:59:57 +03:00
parent a63b9e0b77
commit 902fc38778
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
kvmd/apps/kvmd/server.py
View File

@ -293,9 +293,9 @@ class KvmdServer(HttpServer): # pylint: disable=too-many-arguments,too-many-ins
# ===== SYSTEM STUFF # ===== SYSTEM STUFF
def run(self, **kwargs: Any) -> None: # type: ignore # pylint: disable=arguments-differ def run(self, **kwargs: Any) -> None: # type: ignore # pylint: disable=arguments-differ
for component in self.__components: for comp in self.__components:
if component.sysprep: if comp.sysprep:
component.sysprep() comp.sysprep()
aioproc.rename_process("main") aioproc.rename_process("main")
super().run(**kwargs) super().run(**kwargs)
@ -309,11 +309,11 @@ class KvmdServer(HttpServer): # pylint: disable=too-many-arguments,too-many-ins
app.on_cleanup.append(self.__on_cleanup) app.on_cleanup.append(self.__on_cleanup)
self.__run_system_task(self.__stream_controller) self.__run_system_task(self.__stream_controller)
for component in self.__components: for comp in self.__components:
if component.systask: if comp.systask:
self.__run_system_task(component.systask) self.__run_system_task(comp.systask)
if component.poll_state: if comp.poll_state:
self.__run_system_task(self.__poll_state, component.event_type, component.poll_state()) self.__run_system_task(self.__poll_state, comp.event_type, comp.poll_state())
self.__run_system_task(self.__stream_snapshoter) self.__run_system_task(self.__stream_snapshoter)
for api in self.__apis: for api in self.__apis:
@ -371,13 +371,13 @@ class KvmdServer(HttpServer): # pylint: disable=too-many-arguments,too-many-ins
async def __on_cleanup(self, _: aiohttp.web.Application) -> None: async def __on_cleanup(self, _: aiohttp.web.Application) -> None:
logger = get_logger(0) logger = get_logger(0)
for component in self.__components: for comp in self.__components:
if component.cleanup: if comp.cleanup:
logger.info("Cleaning up %s ...", component.name) logger.info("Cleaning up %s ...", comp.name)
try: try:
await component.cleanup() # type: ignore await comp.cleanup() # type: ignore
except Exception: except Exception:
logger.exception("Cleanup error on %s", component.name) logger.exception("Cleanup error on %s", comp.name)
logger.info("On-Cleanup complete") logger.info("On-Cleanup complete")
async def __send_event(self, ws: aiohttp.web.WebSocketResponse, event_type: str, event: Optional[Dict]) -> None: async def __send_event(self, ws: aiohttp.web.WebSocketResponse, event_type: str, event: Optional[Dict]) -> None: