rework argparsee help

Author: bj00rn

src/configuration/argparse_extensions.py

@@ -2,13 +2,46 @@
 
 import argparse
 from argparse import ArgumentParser, Namespace
+from gettext import gettext as _
 import os
 from typing import TYPE_CHECKING, Any, override
 
 if TYPE_CHECKING:
     from collections.abc import Callable, Sequence
 
 
+class ArgumentHelpFormatter(argparse.RawTextHelpFormatter):
+    """Custom argument formatter.
+
+    Appends environment variable and default value to help.
+    """
+
+    def _get_help_string(self, action: argparse.Action) -> str | None:
+        _help = action.help
+        if _help is None:
+            _help = ""
+
+        if isinstance(action, EnvDefault):
+            t = action.type
+            if t is not None:
+                if (
+                    hasattr(t, "__annotations__")
+                    and t.__annotations__.get("return", None) is not None
+                ):
+                    _help += f"\n(type: {t.__annotations__.get('return', None)})"
+                elif hasattr(t, "__name__"):
+                    _help += f"\n(type: {t.__name__})"
+
+            if "%(default)" not in _help:
+                if action.default is not argparse.SUPPRESS:
+                    defaulting_nargs = [argparse.OPTIONAL, argparse.ZERO_OR_MORE]
+                    if action.option_strings or action.nargs in defaulting_nargs:
+                        _help += _("\n(default: %(default)s)")
+            _help += f"\n(environment variable: {action.envvar})"
+
+        return "\n".join([m.lstrip() for m in _help.split("\n")])
+
+
 class EnvDefault(argparse.Action):
     def __init__(
         self,
@@ -17,6 +50,7 @@ def __init__(
         default: str | None = None,
         **kwargs: dict[str, Any],
     ) -> None:
+        self.envvar = envvar
         if os.environ.get(envvar):
             default = os.environ[envvar]
         if required and default: