Enable colored output for argparse help in Python 3.14 (#19021)

cdce8p · web-flow · commit e7405c90da87 · 2025-05-04T08:31:37.000-07:00
Support for colored output was just merged in cpython. Maybe a bit early to add it here already but it doesn't hurt either. | Current | With color | | --- | --- | | <img width="1058" alt="Without color" src="https://github.com/user-attachments/assets/48edae3e-9361-4af0-aff6-7931410bcd2e" /> | <img width="944" alt="With color" src="https://github.com/user-attachments/assets/38e64082-0321-4d36-a5fe-4d59c6eddc31" /> | The color output can be disable using various environment variables. https://docs.python.org/3.14/using/cmdline.html#using-on-controlling-color
diff --git a/mypy/dmypy/client.py b/mypy/dmypy/client.py
@@ -36,6 +36,9 @@ def __init__(self, prog: str, **kwargs: Any) -> None:
 parser = argparse.ArgumentParser(
     prog="dmypy", description="Client for mypy daemon mode", fromfile_prefix_chars="@"
 )
+if sys.version_info >= (3, 14):
+    parser.color = True  # Set as init arg in 3.14
+
 parser.set_defaults(action=None)
 parser.add_argument(
     "--status-file", default=DEFAULT_STATUS_FILE, help="status file to retrieve daemon details"
diff --git a/mypy/main.py b/mypy/main.py
@@ -491,6 +491,8 @@ def process_options(
         stdout=stdout,
         stderr=stderr,
     )
+    if sys.version_info >= (3, 14):
+        parser.color = True  # Set as init arg in 3.14
 
     strict_flag_names: list[str] = []
     strict_flag_assignments: list[tuple[str, bool]] = []
diff --git a/mypy/stubgen.py b/mypy/stubgen.py
@@ -1851,6 +1851,8 @@ def parse_options(args: list[str]) -> Options:
     parser = argparse.ArgumentParser(
         prog="stubgen", usage=HEADER, description=DESCRIPTION, fromfile_prefix_chars="@"
     )
+    if sys.version_info >= (3, 14):
+        parser.color = True  # Set as init arg in 3.14
 
     parser.add_argument(
         "--ignore-errors",
diff --git a/mypy/stubtest.py b/mypy/stubtest.py
@@ -2084,6 +2084,8 @@ def parse_options(args: list[str]) -> _Arguments:
     parser = argparse.ArgumentParser(
         description="Compares stubs to objects introspected from the runtime."
     )
+    if sys.version_info >= (3, 14):
+        parser.color = True  # Set as init arg in 3.14
     parser.add_argument("modules", nargs="*", help="Modules to test")
     parser.add_argument(
         "--concise",

Original file line number	Diff line number	Diff line change
`@@ -491,6 +491,8 @@ def process_options(`
`491`	`491`	`stdout=stdout,`
`492`	`492`	`stderr=stderr,`
`493`	`493`	`)`
	`494`	`+ if sys.version_info >= (3, 14):`
	`495`	`+ parser.color = True # Set as init arg in 3.14`
`494`	`496`
`495`	`497`	`strict_flag_names: list[str] = []`
`496`	`498`	`strict_flag_assignments: list[tuple[str, bool]] = []`
Original file line number	Diff line number	Diff line change
`@@ -1851,6 +1851,8 @@ def parse_options(args: list[str]) -> Options:`
`1851`	`1851`	`parser = argparse.ArgumentParser(`
`1852`	`1852`	`prog="stubgen", usage=HEADER, description=DESCRIPTION, fromfile_prefix_chars="@"`
`1853`	`1853`	`)`
	`1854`	`+ if sys.version_info >= (3, 14):`
	`1855`	`+ parser.color = True # Set as init arg in 3.14`
`1854`	`1856`
`1855`	`1857`	`parser.add_argument(`
`1856`	`1858`	`"--ignore-errors",`
Original file line number	Diff line number	Diff line change
`@@ -2084,6 +2084,8 @@ def parse_options(args: list[str]) -> _Arguments:`
`2084`	`2084`	`parser = argparse.ArgumentParser(`
`2085`	`2085`	`description="Compares stubs to objects introspected from the runtime."`
`2086`	`2086`	`)`
	`2087`	`+ if sys.version_info >= (3, 14):`
	`2088`	`+ parser.color = True # Set as init arg in 3.14`
`2087`	`2089`	`parser.add_argument("modules", na 473E rgs="*", help="Modules to test")`
`2088`	`2090`	`parser.add_argument(`
`2089`	`2091`	`"--concise",`