sh_cpl/unittests/unittests_shared/cli_commands.py
2023-02-20 15:55:20 +01:00

98 lines
2.7 KiB
Python

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")