kiwi-scp/src/kiwi/subcommands/utils/executable.py

# system
import logging
import os
import subprocess


def _update_kwargs(config, **kwargs):
    if config is not None:
        # ensure there is an environment
        if 'env' not in kwargs:
            kwargs['env'] = {}

        # add common environment from config
        if config['runtime:env'] is not None:
            kwargs['env'].update(config['runtime:env'])

        logging.debug(f"kwargs updated: {kwargs}")

    return kwargs


def _is_executable(filename):
    if filename is None:
        return False

    return os.path.isfile(filename) and os.access(filename, os.X_OK)


def _find_exe_file(exe_name):
    for path in os.environ['PATH'].split(os.pathsep):
        exe_file = os.path.join(path, exe_name)
        if _is_executable(exe_file):
            return exe_file

    raise FileNotFoundError(f"Executable '{exe_name}' not found in $PATH!")


class Executable:
    class __Executable:
        __exe_path = None

        def __init__(self, exe_name):
            self.__exe_path = _find_exe_file(exe_name)

        def __build_cmd(self, args, requires_root=False, **kwargs):
            cmd = [self.__exe_path, *args]

            if requires_root:
                self.__exe_path = [_find_exe_file("sudo"), self.__exe_path]

            logging.debug(f"Executable cmd{cmd}, kwargs{kwargs}")
            return cmd

        def run(self, process_args, config=None, requires_root=False, **kwargs):
            kwargs = _update_kwargs(config, **kwargs)

            return subprocess.run(
                self.__build_cmd(process_args, requires_root, **kwargs),
                **kwargs
            )

        def Popen(self, process_args, config=None, requires_root=False, **kwargs):
            kwargs = _update_kwargs(config, **kwargs)

            return subprocess.Popen(
                self.__build_cmd(process_args, requires_root, **kwargs),
                **kwargs
            )

        def run_less(self, process_args, config=None, requires_root=False, **kwargs):
            kwargs['stdout'] = subprocess.PIPE
            kwargs['stderr'] = subprocess.DEVNULL

            process = self.Popen(
                process_args, config, requires_root,
                **kwargs
            )

            less_process = Executable('less').run(
                ['-R', '+G'],
                stdin=process.stdout
            )

            process.communicate()
            return less_process

    __exe_name = None
    __instances = {}

    def __init__(self, exe_name):
        self.__exe_name = exe_name

        if exe_name not in Executable.__instances:
            Executable.__instances[exe_name] = Executable.__Executable(exe_name)

    def __getattr__(self, item):
        return getattr(self.__instances[self.__exe_name], item)