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:
Devaev Maxim 2020-12-02 15:35:11 +03:00
parent 5bc868662a
commit 7debaa6776
5 changed files with 14 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
kvmd/aioproc.py
View File

@ -30,6 +30,8 @@ from typing import List
import setproctitle
from .logging import get_logger
# =====
async def run_process(cmd: List[str], err_to_null: bool=False) -> asyncio.subprocess.Process: # pylint: disable=no-member
@ -71,3 +73,11 @@ async def log_stdout_infinite(proc: asyncio.subprocess.Process, logger: logging.
def rename_process(suffix: str, prefix: str="kvmd") -> None:
setproctitle.setproctitle(f"{prefix}/{suffix}: {setproctitle.getproctitle()}")
def settle(name: str, suffix: str, prefix: str="kvmd") -> logging.Logger:
logger = get_logger(1)
logger.info("Started %s pid=%d", name, os.getpid())
os.setpgrp()
rename_process(suffix, prefix)
return logger

8
kvmd/plugins/hid/_mcu/__init__.py
View File

@ -20,7 +20,6 @@
# ========================================================================== #
import os
import multiprocessing
import contextlib
import queue
@ -270,12 +269,7 @@ class BaseMcuHid(BaseHid, multiprocessing.Process): # pylint: disable=too-many-
self.__events_queue.put_nowait(event)
def run(self) -> None: # pylint: disable=too-many-branches
logger = get_logger(0)
logger.info("Started HID pid=%d", os.getpid())
os.setpgrp()
aioproc.rename_process("hid")
logger = aioproc.settle("HID", "hid")
while not self.__stop_event.is_set():
try:
with self.__gpio:

8
kvmd/plugins/hid/bt/__init__.py
View File

@ -20,7 +20,6 @@
# ========================================================================== #
import os
import multiprocessing
import time
@ -194,12 +193,7 @@ class Plugin(BaseHid): # pylint: disable=too-many-instance-attributes
# =====
def __server_worker(self) -> None: # pylint: disable=too-many-branches
logger = get_logger(0)
logger.info("Started HID pid=%d", os.getpid())
os.setpgrp()
aioproc.rename_process("hid")
logger = aioproc.settle("HID", "hid")
while not self.__stop_event.is_set():
try:
self.__server.run()

7
kvmd/plugins/hid/otg/device.py
View File

@ -78,12 +78,7 @@ class BaseDeviceProcess(multiprocessing.Process): # pylint: disable=too-many-in
self.__stop_event = multiprocessing.Event()
def run(self) -> None:
logger = get_logger(0)
logger.info("Started HID-%s pid=%d", self.__name, os.getpid())
os.setpgrp()
aioproc.rename_process(f"hid-{self.__name}")
logger = aioproc.settle(f"HID-{self.__name}", f"hid-{self.__name}")
while not self.__stop_event.is_set():
try:
while not self.__stop_event.is_set():

8
kvmd/plugins/ugpio/ezcoo.py
View File

@ -20,7 +20,6 @@
# ========================================================================== #
import os
import re
import multiprocessing
import errno
@ -130,12 +129,7 @@ class Plugin(BaseUserGpioDriver): # pylint: disable=too-many-instance-attribute
)
def __serial_worker(self) -> None:
logger = get_logger(0)
logger.info("Started %s pid=%d", self, os.getpid())
os.setpgrp()
aioproc.rename_process(f"gpio-ezcoo-{self._instance_name}")
logger = aioproc.settle(str(self), f"gpio-ezcoo-{self._instance_name}")
while not self.__stop_event.is_set():
try:
with self.__get_serial() as tty: