proc-nodes / methods.py

import os
import sys
from enum import Enum


_colorize = bool(sys.stdout.isatty() or os.environ.get("CI"))


class ANSI(Enum):
    RESET = "\x1b[0m"
    BOLD = "\x1b[1m"
    REGULAR = "\x1b[22;23;24;29m"
    RED = "\x1b[31m"
    GREEN = "\x1b[32m"
    YELLOW = "\x1b[33m"

    def __str__(self) -> str:
        global _colorize
        return str(self.value) if _colorize else ""


def print_warning(*values: object) -> None:
    print(f"{ANSI.YELLOW}{ANSI.BOLD}WARNING:{ANSI.REGULAR}", *values, ANSI.RESET, file=sys.stderr)


def print_error(*values: object) -> None:
    print(f"{ANSI.RED}{ANSI.BOLD}ERROR:{ANSI.REGULAR}", *values, ANSI.RESET, file=sys.stderr)