gh-127353: Allow to force color output on Windows (#127354)

This commit is contained in:
Andrey Efremov
2024-12-13 03:17:39 +07:00
committed by GitHub
parent f823910bbd
commit 365451e283
3 changed files with 48 additions and 8 deletions

View File

@@ -32,14 +32,6 @@ def get_colors(colorize: bool = False) -> ANSIColors:
def can_colorize() -> bool:
if sys.platform == "win32":
try:
import nt
if not nt._supports_virtual_terminal():
return False
except (ImportError, AttributeError):
return False
if not sys.flags.ignore_environment:
if os.environ.get("PYTHON_COLORS") == "0":
return False
@@ -58,6 +50,15 @@ def can_colorize() -> bool:
if not hasattr(sys.stderr, "fileno"):
return False
if sys.platform == "win32":
try:
import nt
if not nt._supports_virtual_terminal():
return False
except (ImportError, AttributeError):
return False
try:
return os.isatty(sys.stderr.fileno())
except io.UnsupportedOperation: