Function force_color

Lib/test/support/__init__.py:3023–3040

(color: bool)

@contextlib.contextmanager
def force_color(color: bool):
    import _colorize
    from .os_helper import EnvironmentVarGuard

    if color:
        try:
            import _pyrepl  # noqa: F401
        except ModuleNotFoundError:
            # Can't force enable color without _pyrepl, so just skip.
            raise unittest.SkipTest("_pyrepl is missing")

    with (
        swap_attr(_colorize, "can_colorize", lambda *, file=None: color),
        EnvironmentVarGuard() as env,
    ):
        env.unset("FORCE_COLOR", "NO_COLOR", "PYTHON_COLORS")
        env.set("FORCE_COLOR" if color else "NO_COLOR", "1")
        yield


def force_colorized(func):