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

aiohelpers

Browse Source
This commit is contained in:
Maxim Devaev 2022-04-10 08:08:05 +03:00
parent c92d17dea2
commit 7c44d733a9
2 changed files with 19 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
kvmd/plugins/msd/otg/helpers.py → kvmd/aiohelpers.py
View File

@ -20,35 +20,31 @@
# ========================================================================== #
import subprocess
from typing import List
from ....logging import get_logger
from .logging import get_logger
from .... import aioproc
from .. import MsdError
from . import tools
from . import aioproc
# =====
async def remount_storage(base_cmd: List[str], rw: bool) -> None:
logger = get_logger(0)
async def remount(name: str, base_cmd: List[str], rw: bool) -> bool:
logger = get_logger(1)
mode = ("rw" if rw else "ro")
cmd = [
part.format(mode=mode)
for part in base_cmd
]
logger.info("Remounting internal storage to %s ...", mode.upper())
logger.info("Remounting %s storage to %s: %s ...", name, mode.upper(), cmd)
try:
await _run_helper(cmd)
except Exception:
logger.error("Can't remount internal storage")
raise
# =====
async def _run_helper(cmd: List[str]) -> None:
logger = get_logger(0)
logger.info("Executing helper %s ...", cmd)
proc = await aioproc.log_process(cmd, logger)
if proc.returncode != 0:
raise MsdError(f"Error while helper execution: pid={proc.pid}; retcode={proc.returncode}")
assert proc.returncode is not None
raise subprocess.CalledProcessError(proc.returncode, cmd)
except Exception as err:
logger.error("Can't remount %s storage: %s", name, tools.efmt(err))
return False
return True

5
kvmd/plugins/msd/otg/__init__.py
View File

@ -46,6 +46,7 @@ from ....validators.os import valid_printable_filename
from ....validators.os import valid_command
from .... import aiotools
from .... import aiohelpers
from .. import MsdError
from .. import MsdIsBusyError
@ -59,7 +60,6 @@ from .. import BaseMsd
from .. import MsdImageWriter
from . import fs
from . import helpers
from .drive import Drive
@ -538,4 +538,5 @@ class Plugin(BaseMsd): # pylint: disable=too-many-instance-attributes
# =====
async def __remount_storage(self, rw: bool) -> None:
await helpers.remount_storage(self.__remount_cmd, rw)
if not (await aiohelpers.remount("MSD", self.__remount_cmd, rw)):
raise MsdError("Can't execute remount helper")