GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2026-01-31 10:01:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Devaev Maxim
2020-12-21 01:59:49 +03:00
parent 2cb210c975
commit 42475809fc
6 changed files with 43 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
kvmd/plugins/msd/relay/__init__.py
View File

@@ -176,32 +176,26 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes
raise MsdCdromNotSupported()
@aiotools.atomic
async def connect(self) -> None:
async def set_connected(self, connected: bool) -> None:
async with self.__working():
async with self.__region:
if self.__connected:
raise MsdConnectedError()
self.__gpio.switch_to_server()
self.__connected = True
get_logger(0).info("MSD switched to Server")
@aiotools.atomic
async def disconnect(self) -> None:
async with self.__working():
async with self.__region:
if not self.__connected:
raise MsdDisconnectedError()
self.__gpio.switch_to_local()
try:
await self.__load_device_info()
except Exception:
if connected:
if self.__connected:
self.__gpio.switch_to_server()
raise
self.__connected = False
get_logger(0).info("MSD switched to KVM: %s", self.__device_info)
raise MsdConnectedError()
self.__gpio.switch_to_server()
get_logger(0).info("MSD switched to Server")
else:
if not self.__connected:
raise MsdDisconnectedError()
self.__gpio.switch_to_local()
try:
await self.__load_device_info()
except Exception:
if self.__connected:
self.__gpio.switch_to_server()
raise
get_logger(0).info("MSD switched to KVM: %s", self.__device_info)
self.__connected = connected
@contextlib.asynccontextmanager
async def write_image(self, name: str) -> AsyncGenerator[None, None]: