supports_ansi.py

"""
This code is from the discord.py repository.
Source: https://github.com/Rapptz/discord.py/blob/master/discord/utils.py#L1241-L1253
"""

import os
import sys
from typing import Any

__all__ = ("stream_supports_colour",)


def is_docker() -> bool:
    path = "/proc/self/cgroup"
    return os.path.exists("/.dockerenv") or (
        os.path.isfile(path) and any("docker" in line for line in open(path))
    )


def stream_supports_colour(stream: Any) -> bool:
    # Pycharm and Vscode support colour in their inbuilt editors
    if "PYCHARM_HOSTED" in os.environ or os.environ.get("TERM_PROGRAM") == "vscode":
        return True

    is_a_tty = hasattr(stream, "isatty") and stream.isatty()
    if sys.platform != "win32":
        # Docker does not consistently have a tty attached to it
        return is_a_tty or is_docker()

    # ANSICON checks for things like ConEmu
    # WT_SESSION checks if this is Windows Terminal
    return is_a_tty and ("ANSICON" in os.environ or "WT_SESSION" in os.environ)