diff --git a/utils.py b/utils.py index 520809d..dd008e3 100644 --- a/utils.py +++ b/utils.py @@ -2,6 +2,10 @@ from shutil import which import atexit import subprocess +import logging + +from os import PathLike + def programs_available(programs) -> bool: if type(programs) is str: @@ -12,18 +16,18 @@ def programs_available(programs) -> bool: return True -def umount(dest): +def umount(dest: PathLike, lazy=False): return subprocess.run( [ 'umount', - '-lc', + '-c' + ('l' if lazy else ''), dest, ], capture_output=True, ) -def mount(src: str, dest: str, options=['bind'], fs_type=None, register_unmount=True) -> subprocess.CompletedProcess: +def mount(src: PathLike, dest: PathLike, options=['bind'], fs_type=None, register_unmount=True) -> subprocess.CompletedProcess: opts = [] for opt in options: opts += ['-o', opt] @@ -43,5 +47,26 @@ def mount(src: str, dest: str, options=['bind'], fs_type=None, register_unmount= return result +def check_findmnt(path: PathLike): + result = subprocess.run( + [ + 'findmnt', + '-n', + '-o', + 'source', + path, + ], + capture_output=True, + ) + return result.stdout.decode().strip() + + def git(cmd: list[str], dir='.', capture_output=False) -> subprocess.CompletedProcess: return subprocess.run(['git'] + cmd, cwd=dir, capture_output=capture_output) + + +def log_or_exception(raise_exception: bool, msg: str, exc_class=Exception, log_level=logging.WARNING): + if raise_exception: + raise exc_class(msg) + else: + logging.log(log_level, msg)