summaryrefslogtreecommitdiff
path: root/kvmd/plugins/msd
diff options
context:
space:
mode:
authorDevaev Maxim <[email protected]>2020-09-08 05:24:47 +0300
committerDevaev Maxim <[email protected]>2020-09-08 05:24:47 +0300
commit605b67ca761bdb5f5c3b255434f30624fa82a6c1 (patch)
treed7821a55ca5ec9e2da6b75bf2956dea6345de1a1 /kvmd/plugins/msd
parent4cc60e4d528c669af0c1f7478160ff57a4691af7 (diff)
refactoring
Diffstat (limited to 'kvmd/plugins/msd')
-rw-r--r--kvmd/plugins/msd/otg/__init__.py20
-rw-r--r--kvmd/plugins/msd/relay.py8
2 files changed, 14 insertions, 14 deletions
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: