sh_cpl/unittests/unittests_shared/cli_commands.py

import os
import subprocess
import sys

from unittests_cli.constants import CLI_PATH


class CLICommands:
    @staticmethod
    def _run(cmd: str, *args, output=False):
        env_vars = os.environ
        env_vars["CPL_IS_UNITTEST"] = "NO" if output else "YES"

        command = ["python", CLI_PATH, cmd]
        for arg in args:
            command.append(arg)

        if output:
            subprocess.run(command, env=env_vars)
        else:
            subprocess.run(
                command, env=env_vars, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, stdin=subprocess.DEVNULL
            )

    @staticmethod
    def _run_with_output(cmd: str, *args) -> str:
        env_vars = os.environ
        env_vars["CPL_IS_UNITTEST"] = "NO"

        command = ["python", CLI_PATH, cmd]
        for arg in args:
            command.append(arg)

        return subprocess.run(command, env=env_vars, check=True, capture_output=True, text=True).stdout

    @classmethod
    def add(cls, source: str, target: str, output=False):
        cls._run("add", source, target, output=output)

    @classmethod
    def build(cls, output=False):
        cls._run("build", output=output)

    @classmethod
    def generate(cls, schematic: str, name: str, output=False):
        cls._run("generate", schematic, name, output=output)

    @classmethod
    def install(cls, package: str = None, is_dev=False, output=False):
        if package is None:
            cls._run("install", output=output)
            return

        cls._run("install", package, "--dev" if is_dev else "", output=output)

    @classmethod
    def new(cls, project_type: str, name: str, *args, output=False):
        cls._run("new", project_type, name, *args, output=output)

    @classmethod
    def publish(cls, output=False):
        cls._run("publish", output=output)

    @classmethod
    def remove(cls, project: str, output=False):
        cls._run("remove", project, output=output)

    @classmethod
    def run(cls, project: str = None, is_dev=False, output=False):
        args = []
        if is_dev:
            args.append("--dev")

        if project is None:
            cls._run("run", *args, output=output)
            return
        cls._run("run", project, *args, output=output)

    @classmethod
    def start(cls, is_dev=False, output=False):
        args = []
        if is_dev:
            args.append("--dev")

        cls._run("start", *args, output=output)

    @classmethod
    def uninstall(cls, package: str, is_dev=False, output=False):
        cls._run("uninstall", package, "--dev" if is_dev else "", output=output)

    @classmethod
    def update(cls, output=False):
        cls._run("update", output=output)

    @classmethod
    def version(cls) -> str:
        return cls._run_with_output("version")