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

Author: cdce8p

python_typing_update/__main__.py

@@ -7,6 +7,7 @@
 import asyncio
 import logging
 import sys
+from typing import Any
 
 from .main import async_run
 
@@ -15,19 +16,32 @@
 
 class CustomHelpFormatter(argparse.HelpFormatter):
     def __init__(
-        self, prog: str, indent_increment: int = 2,
-        max_help_position: int = 24, width: int | None = None,
+        self,
+        prog: str,
+        indent_increment: int = 2,
+        max_help_position: int = 24,
+        width: int | None = None,
+        **kwargs: Any,
     ) -> None:
         max_help_position = 40
         super().__init__(
-            prog, indent_increment=indent_increment,
-            max_help_position=max_help_position, width=width)
+            prog,
+            indent_increment=indent_increment,
+            max_help_position=max_help_position,
+            width=width,
+            **kwargs,
+        )
 
 
 async def async_main(argv: list[str] | None = None) -> int:
+    parser_kwargs: dict[str, Any] = {}
+    if sys.version_info >= (3, 14):
+        parser_kwargs["color"] = True
+
     parser = argparse.ArgumentParser(
         description="Tool to update Python typing syntax.",
         formatter_class=CustomHelpFormatter,
+        **parser_kwargs,
     )
     formatter_options = parser.add_argument_group("select optional formatter")
     mode_options = parser.add_argument_group("select different mode")