kiwi-scp/src/kiwi/subcommands/_utils.py

import logging
import os
import subprocess

from ..core import Parser
from ..config import LoadedConfig


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

    return None


def get_exe_key(exe_name):
    return f'executables:{exe_name}'


class SubCommand:
    __name = None
    __parser = None

    def __init__(self, name, **kwargs):
        self.__name = name
        self.__parser = Parser().get_subparsers().add_parser(name, **kwargs)

    def __str__(self):
        return self.__name

    def get_parser(self):
        return self.__parser

    def run(self):
        pass


class DockerCommand:
    class __DockerCommand:
        __cmd = []

        def __init__(self, exe_name):
            config = LoadedConfig.get()
            self.__cmd = [config[get_exe_key(exe_name)]]

            if DockerCommand.__requires_root:
                self.__cmd = [config[get_exe_key("sudo")], *self.__cmd]

        def __build_cmd(self, args):
            cmd = [*self.__cmd, *args]
            logging.debug(f"DockerProgram: {cmd}")
            return cmd

        def run(self, args, **kwargs):
            return subprocess.run(
                self.__build_cmd(args),
                **kwargs
            )

        def run_less(self, args, **kwargs):
            process = subprocess.Popen(
                self.__build_cmd(args),
                stdout=subprocess.PIPE, stderr=subprocess.DEVNULL,
                **kwargs
            )

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

            process.communicate()
            return less_process

    __exe_name = None
    __instances = {}
    __requires_root = None

    def __init__(self, exe_name):
        if DockerCommand.__requires_root is None:
            try:
                config = LoadedConfig.get()
                subprocess.run(
                    [config[get_exe_key('docker')], 'ps'],
                    check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
                )
                DockerCommand.__requires_root = False
            except subprocess.CalledProcessError:
                DockerCommand.__requires_root = True

        self.__exe_name = exe_name

        if exe_name not in DockerCommand.__instances:
            DockerCommand.__instances[exe_name] = DockerCommand.__DockerCommand(exe_name)

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