diff options
Diffstat (limited to 'kvmd/plugins/msd')
| -rw-r--r-- | kvmd/plugins/msd/otg/__init__.py | 3 | ||||
| -rw-r--r-- | kvmd/plugins/msd/relay.py | 5 |
2 files changed, 3 insertions, 5 deletions
diff --git a/kvmd/plugins/msd/otg/__init__.py b/kvmd/plugins/msd/otg/__init__.py index 41e1653f..f3695530 100644 --- a/kvmd/plugins/msd/otg/__init__.py +++ b/kvmd/plugins/msd/otg/__init__.py @@ -45,7 +45,6 @@ from ....validators.os import valid_abs_dir from ....validators.os import valid_command from .... import aiotools -from .... import aioregion from .. import MsdError from .. import MsdIsBusyError @@ -111,7 +110,7 @@ class _State: self.vd: Optional[_VirtualDriveState] = None self._lock = asyncio.Lock() - self._region = aioregion.AioExclusiveRegion(MsdIsBusyError) + self._region = aiotools.AioExclusiveRegion(MsdIsBusyError) @contextlib.asynccontextmanager async def busy(self, check_online: bool=True) -> AsyncGenerator[None, None]: diff --git a/kvmd/plugins/msd/relay.py b/kvmd/plugins/msd/relay.py index 4e621cc2..81456efa 100644 --- a/kvmd/plugins/msd/relay.py +++ b/kvmd/plugins/msd/relay.py @@ -40,7 +40,6 @@ import aiofiles.base from ...logging import get_logger from ... import aiotools -from ... import aioregion from ... import gpio from ...yamlconf import Option @@ -174,7 +173,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes self.__init_retries = init_retries self.__reset_delay = reset_delay - self.__region = aioregion.AioExclusiveRegion(MsdIsBusyError) + self.__region = aiotools.AioExclusiveRegion(MsdIsBusyError) self.__device_info: Optional[_DeviceInfo] = None self.__connected = False @@ -235,7 +234,7 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes @aiotools.atomic async def reset(self) -> None: - with aiotools.unregion_only_on_exception(self.__region): + with self.__region.exit_only_on_exception(): await self.__inner_reset() @aiotools.tasked |