nix-update/nix_update/utils.py

import os
import subprocess
import sys
from pathlib import Path
from typing import IO, Any, Callable, List, Optional, Union, Dict

HAS_TTY = sys.stdout.isatty()
ROOT = Path(os.path.dirname(os.path.realpath(__file__)))


def color_text(code: int, file: IO[Any] = sys.stdout) -> Callable[[str], None]:
    def wrapper(text: str) -> None:
        if HAS_TTY:
            print(f"\x1b[{code}m{text}\x1b[0m", file=file)
        else:
            print(text, file=file)

    return wrapper


warn = color_text(31, file=sys.stderr)
info = color_text(32)


def run(
    command: List[str],
    cwd: Optional[Union[Path, str]] = None,
    stdout: Union[None, int, IO[Any]] = subprocess.PIPE,
    stderr: Union[None, int, IO[Any]] = subprocess.PIPE,
    check: bool = True,
    extra_env: Dict[str, str] = {},
) -> "subprocess.CompletedProcess[str]":
    info("$ " + " ".join(command))
    env = os.environ.copy()
    env.update(extra_env)
    return subprocess.run(
        command, cwd=cwd, check=check, text=True, stdout=stdout, stderr=stderr, env=env
    )