"""
Helper module so we don't have to install types-termcolor in CI.

This is imported by `mypy_test.py` and `stubtest_third_party.py`.
"""

from typing import TYPE_CHECKING

if TYPE_CHECKING:

    def colored(__str: str, __style: str) -> str:
        ...

else:
    try:
        from termcolor import colored
    except ImportError:

        def colored(s: str, _: str) -> str:
            return s


def print_error(error: str, end: str = "\n") -> None:
    error_split = error.split("\n")
    for line in error_split[:-1]:
        print(colored(line, "red"))
    print(colored(error_split[-1], "red"), end=end)


def print_success_msg() -> None:
    print(colored("success", "green"))