diff options
author | Devaev Maxim <[email protected]> | 2020-09-08 05:24:47 +0300 |
---|---|---|
committer | Devaev Maxim <[email protected]> | 2020-09-08 05:24:47 +0300 |
commit | 605b67ca761bdb5f5c3b255434f30624fa82a6c1 (patch) | |
tree | d7821a55ca5ec9e2da6b75bf2956dea6345de1a1 /kvmd | |
parent | 4cc60e4d528c669af0c1f7478160ff57a4691af7 (diff) |
refactoring
Diffstat (limited to 'kvmd')
-rw-r--r-- | kvmd/apps/kvmd/streamer.py | 8 | ||||
-rw-r--r-- | kvmd/apps/kvmd/ugpio.py | 8 | ||||
-rw-r--r-- | kvmd/plugins/atx/gpio.py | 8 | ||||
-rw-r--r-- | kvmd/plugins/hid/otg/__init__.py | 8 | ||||
-rw-r--r-- | kvmd/plugins/hid/otg/device.py | 4 | ||||
-rw-r--r-- | kvmd/plugins/hid/serial.py | 6 | ||||
-rw-r--r-- | kvmd/plugins/msd/otg/__init__.py | 20 | ||||
-rw-r--r-- | kvmd/plugins/msd/relay.py | 8 |
8 files changed, 35 insertions, 35 deletions
diff --git a/kvmd/apps/kvmd/streamer.py b/kvmd/apps/kvmd/streamer.py index 6b2d597a..f9fffc16 100644 --- a/kvmd/apps/kvmd/streamer.py +++ b/kvmd/apps/kvmd/streamer.py @@ -172,7 +172,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes self.__snapshot: Optional[StreamerSnapshot] = None - self.__state_notifier = aiotools.AioNotifier() + self.__notifier = aiotools.AioNotifier() # ===== @@ -277,7 +277,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes async def poll_state(self) -> AsyncGenerator[Dict, None]: def signal_handler(*_: Any) -> None: get_logger(0).info("Got SIGUSR2, checking the stream state ...") - asyncio.ensure_future(self.__state_notifier.notify()) + asyncio.ensure_future(self.__notifier.notify()) get_logger(0).info("Installing SIGUSR2 streamer handler ...") asyncio.get_event_loop().add_signal_handler(signal.SIGUSR2, signal_handler) @@ -291,7 +291,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes prev_state = state if waiter_task is None: - waiter_task = asyncio.create_task(self.__state_notifier.wait()) + waiter_task = asyncio.create_task(self.__notifier.wait()) if waiter_task in (await aiotools.wait_first(asyncio.sleep(self.__state_poll), waiter_task))[0]: waiter_task = None @@ -328,7 +328,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes ) if save: self.__snapshot = snapshot - await self.__state_notifier.notify() + await self.__notifier.notify() return snapshot logger.error("Stream is offline, no signal or so") except (aiohttp.ClientConnectionError, aiohttp.ServerConnectionError) as err: diff --git a/kvmd/apps/kvmd/ugpio.py b/kvmd/apps/kvmd/ugpio.py index 8e7ae90a..ab034365 100644 --- a/kvmd/apps/kvmd/ugpio.py +++ b/kvmd/apps/kvmd/ugpio.py @@ -215,12 +215,12 @@ class UserGpio: def __init__(self, config: Section) -> None: self.__view = config.view - self.__state_notifier = aiotools.AioNotifier() + self.__notifier = aiotools.AioNotifier() self.__drivers = { driver: get_ugpio_driver_class(drv_config.type)( instance_name=driver, - notifier=self.__state_notifier, + notifier=self.__notifier, **drv_config._unpack(ignore=["instance_name", "notifier", "type"]), ) for (driver, drv_config) in config.drivers.items() @@ -236,7 +236,7 @@ class UserGpio: if ch_config.mode == "input": self.__inputs[channel] = _GpioInput(channel, ch_config, driver) else: # output: - self.__outputs[channel] = _GpioOutput(channel, ch_config, driver, self.__state_notifier) + self.__outputs[channel] = _GpioOutput(channel, ch_config, driver, self.__notifier) async def get_model(self) -> Dict: return { @@ -260,7 +260,7 @@ class UserGpio: if state != prev_state: yield state prev_state = state - await self.__state_notifier.wait() + await self.__notifier.wait() def sysprep(self) -> None: get_logger().info("Preparing User-GPIO drivers ...") diff --git a/kvmd/plugins/atx/gpio.py b/kvmd/plugins/atx/gpio.py index 91ab7b5f..16f3e23f 100644 --- a/kvmd/plugins/atx/gpio.py +++ b/kvmd/plugins/atx/gpio.py @@ -71,13 +71,13 @@ class Plugin(BaseAtx): # pylint: disable=too-many-instance-attributes self.__click_delay = click_delay self.__long_click_delay = long_click_delay - self.__state_notifier = aiotools.AioNotifier() - self.__region = aiotools.AioExclusiveRegion(AtxIsBusyError, self.__state_notifier) + self.__notifier = aiotools.AioNotifier() + self.__region = aiotools.AioExclusiveRegion(AtxIsBusyError, self.__notifier) self.__reader = gpio.BatchReader( pins=set([self.__power_led_pin, self.__hdd_led_pin]), interval=state_poll, - notifier=self.__state_notifier, + notifier=self.__notifier, ) @classmethod @@ -113,7 +113,7 @@ class Plugin(BaseAtx): # pylint: disable=too-many-instance-attributes if state != prev_state: yield state prev_state = state - await self.__state_notifier.wait() + await self.__notifier.wait() async def systask(self) -> None: await self.__reader.poll() diff --git a/kvmd/plugins/hid/otg/__init__.py b/kvmd/plugins/hid/otg/__init__.py index 575190e5..b9c2c5b8 100644 --- a/kvmd/plugins/hid/otg/__init__.py +++ b/kvmd/plugins/hid/otg/__init__.py @@ -51,10 +51,10 @@ class Plugin(BaseHid): noop: bool, ) -> None: - self.__state_notifier = aiomulti.AioProcessNotifier() + self.__notifier = aiomulti.AioProcessNotifier() - self.__keyboard_proc = KeyboardProcess(noop=noop, state_notifier=self.__state_notifier, **keyboard) - self.__mouse_proc = MouseProcess(noop=noop, state_notifier=self.__state_notifier, **mouse) + self.__keyboard_proc = KeyboardProcess(noop=noop, notifier=self.__notifier, **keyboard) + self.__mouse_proc = MouseProcess(noop=noop, notifier=self.__notifier, **mouse) @classmethod def get_plugin_options(cls) -> Dict: @@ -101,7 +101,7 @@ class Plugin(BaseHid): if state != prev_state: yield state prev_state = state - await self.__state_notifier.wait() + await self.__notifier.wait() async def reset(self) -> None: self.__keyboard_proc.send_reset_event() diff --git a/kvmd/plugins/hid/otg/device.py b/kvmd/plugins/hid/otg/device.py index 45681aba..b777dd56 100644 --- a/kvmd/plugins/hid/otg/device.py +++ b/kvmd/plugins/hid/otg/device.py @@ -47,7 +47,7 @@ class BaseDeviceProcess(multiprocessing.Process): # pylint: disable=too-many-in name: str, read_size: int, initial_state: Dict, - state_notifier: aiomulti.AioProcessNotifier, + notifier: aiomulti.AioProcessNotifier, device_path: str, select_timeout: float, @@ -69,7 +69,7 @@ class BaseDeviceProcess(multiprocessing.Process): # pylint: disable=too-many-in self.__fd = -1 self.__events_queue: multiprocessing.queues.Queue = multiprocessing.Queue() - self.__state_flags = aiomulti.AioSharedFlags({"online": True, **initial_state}, state_notifier) + self.__state_flags = aiomulti.AioSharedFlags({"online": True, **initial_state}, notifier) self.__stop_event = multiprocessing.Event() def run(self) -> None: diff --git a/kvmd/plugins/hid/serial.py b/kvmd/plugins/hid/serial.py index 5abab1f5..bb97aec4 100644 --- a/kvmd/plugins/hid/serial.py +++ b/kvmd/plugins/hid/serial.py @@ -191,13 +191,13 @@ class Plugin(BaseHid, multiprocessing.Process): # pylint: disable=too-many-inst self.__events_queue: multiprocessing.queues.Queue = multiprocessing.Queue() - self.__state_notifier = aiomulti.AioProcessNotifier() + self.__notifier = aiomulti.AioProcessNotifier() self.__state_flags = aiomulti.AioSharedFlags({ "online": True, "caps": False, "scroll": False, "num": False, - }, self.__state_notifier) + }, self.__notifier) self.__stop_event = multiprocessing.Event() @@ -243,7 +243,7 @@ class Plugin(BaseHid, multiprocessing.Process): # pylint: disable=too-many-inst if state != prev_state: yield state prev_state = state - await self.__state_notifier.wait() + await self.__notifier.wait() @aiotools.atomic async def reset(self) -> None: diff --git a/kvmd/plugins/msd/otg/__init__.py b/kvmd/plugins/msd/otg/__init__.py index 8d1cc1b2..bb5ebd66 100644 --- a/kvmd/plugins/msd/otg/__init__.py +++ b/kvmd/plugins/msd/otg/__init__.py @@ -155,8 +155,8 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes self.__new_file_written = 0 self.__new_file_tick = 0.0 - self.__state_notifier = aiotools.AioNotifier() - self.__state = _State(self.__state_notifier) + self.__notifier = aiotools.AioNotifier() + self.__state = _State(self.__notifier) logger = get_logger(0) logger.info("Using OTG gadget %r as MSD", gadget) @@ -212,7 +212,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes if state != prev_state: yield state prev_state = state - await self.__state_notifier.wait() + await self.__notifier.wait() async def systask(self) -> None: await self.__watch_inotify() @@ -294,7 +294,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes async with self.__state._region: # pylint: disable=protected-access try: async with self.__state._lock: # pylint: disable=protected-access - await self.__state_notifier.notify() + await self.__notifier.notify() assert self.__state.storage assert self.__state.vd @@ -310,7 +310,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes self.__new_file_written = 0 self.__new_file = await aiofiles.open(path, mode="w+b", buffering=0) - await self.__state_notifier.notify() + await self.__notifier.notify() yield self.__set_image_complete(name, True) @@ -324,7 +324,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes # Между закрытием файла и эвентом айнотифи состояние может быть не обновлено, # так что форсим обновление вручную, чтобы получить актуальное состояние. await self.__reload_state() - await self.__state_notifier.notify() + await self.__notifier.notify() async def write_image_chunk(self, chunk: bytes) -> int: assert self.__new_file @@ -334,7 +334,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes if self.__new_file_tick + 1 < now: # Это нужно для ручного оповещения о свободном пространстве на диске, см. get_state() self.__new_file_tick = now - await self.__state_notifier.notify() + await self.__notifier.notify() return self.__new_file_written @aiotools.atomic @@ -382,7 +382,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes while True: # Активно ждем, пока не будут на месте все каталоги. await self.__reload_state() - await self.__state_notifier.notify() + await self.__notifier.notify() if self.__state.vd: break await asyncio.sleep(5) @@ -394,7 +394,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes # После установки вотчеров еще раз проверяем стейт, чтобы ничего не потерять await self.__reload_state() - await self.__state_notifier.notify() + await self.__notifier.notify() while self.__state.vd: # Если живы после предыдущей проверки need_restart = False @@ -410,7 +410,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes break if need_reload_state: await self.__reload_state() - await self.__state_notifier.notify() + await self.__notifier.notify() except Exception: logger.exception("Unexpected MSD watcher error") diff --git a/kvmd/plugins/msd/relay.py b/kvmd/plugins/msd/relay.py index 96cb3c63..409d3d71 100644 --- a/kvmd/plugins/msd/relay.py +++ b/kvmd/plugins/msd/relay.py @@ -179,8 +179,8 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes self.__device_file: Optional[aiofiles.base.AiofilesContextManager] = None self.__written = 0 - self.__state_notifier = aiotools.AioNotifier() - self.__region = aiotools.AioExclusiveRegion(MsdIsBusyError, self.__state_notifier) + self.__notifier = aiotools.AioNotifier() + self.__region = aiotools.AioExclusiveRegion(MsdIsBusyError, self.__notifier) logger = get_logger(0) logger.info("Using %r as MSD", self.__device_path) @@ -234,7 +234,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes if state != prev_state: yield state prev_state = state - await self.__state_notifier.wait() + await self.__notifier.wait() @aiotools.atomic async def reset(self) -> None: @@ -317,7 +317,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes self.__written = 0 await self.__write_image_info(name, complete=False) - await self.__state_notifier.notify() + await self.__notifier.notify() yield await self.__write_image_info(name, complete=True) finally: |