Handle (PIP_)NO_COLOR

Author: ichard26

src/pip/_internal/cli/parser.py

@@ -4,21 +4,23 @@
 
 import logging
 import optparse
+import os
 import shutil
 import sys
 import textwrap
 from collections.abc import Generator
 from contextlib import suppress
 from typing import Any, NoReturn
 
-from pip._vendor.rich.console import Console, RenderableType
+from pip._vendor.rich.console import RenderableType
 from pip._vendor.rich.markup import escape
 from pip._vendor.rich.style import StyleType
 from pip._vendor.rich.text import Text
 from pip._vendor.rich.theme import Theme
 
 from pip._internal.cli.status_codes import UNKNOWN_ERROR
 from pip._internal.configuration import Configuration, ConfigurationError
+from pip._internal.utils.logging import PipConsole
 from pip._internal.utils.misc import redact_auth_from_url, strtobool
 
 logger = logging.getLogger(__name__)
@@ -46,7 +48,14 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         kwargs["indent_increment"] = 1
         kwargs["width"] = shutil.get_terminal_size()[0] - 2
         super().__init__(*args, **kwargs)
-        self.console: Console = Console(theme=Theme(self.styles))
+        # This is unfortunate but necessary since arguments may have not been
+        # parsed yet at this point, so detect --no-color manually.
+        no_color = (
+            "--no-color" in sys.argv
+            or bool(strtobool(os.environ.get("PIP_NO_COLOR", "no") or "no"))
+            or "NO_COLOR" in os.environ
+        )
+        self.console = PipConsole(theme=Theme(self.styles), no_color=no_color)
         self.rich_option_strings: dict[optparse.Option, Text] = {}
 
     def stringify(self, text: RenderableType) -> str: