Lazy-load CLI commands so Usage: augur [OPTIONS] COMMAND [ARGS]...

  Augur is an application for open source community health analytics

Options:
  --help  Show this message and exit.

Commands:
  api         Commands for controlling the backend API server
  backend     Commands for controlling the backend API server & data
              collection workers
  cache       Commands for managing redis cache
  collection  Commands for controlling the backend API server & data
              collection workers
  config      Generate an augur.config.json
  csv_utils
  db          Database utilities
  github      Github utilities
  jumpstart
  tasks       Commands for controlling the backend API server & data
              collection workers
  user        Support for adding regular users or administrative users works without DB config
Signed-off-by: Hamza <mezohafez1@gmail.com>

Author: Hamza

augur/application/cli/_multicommand.py

@@ -1,40 +1,88 @@
-#SPDX-License-Identifier: MIT
+# SPDX-License-Identifier: MIT
 """
 Runs Augur with Gunicorn when called
 """
 
-import os
-import click
-import importlib
-import traceback
+# pylint: disable=missing-class-docstring,missing-function-docstring,unused-argument
 
+import importlib
+import re
 from pathlib import Path
-# import augur.application
+from typing import Optional
+
+import click
+
+CONTEXT_SETTINGS = {"auto_envvar_prefix": "AUGUR"}
+
+
+def _extract_short_help(py_file: Path) -> Optional[str]:
+    """
+    Best-effort parse short_help from the command module without importing it.
+    Looks for short_help='...' or short_help="..."
+    """
+    try:
+        text = py_file.read_text(encoding="utf-8", errors="ignore")
+    except (OSError, UnicodeError):
+        return None
+
+
+    m = re.search(r"short_help\s*=\s*(['\"])(.*?)\1", text)
+    if not m:
+        return None
+    return m.group(2).strip() or None
+
+
+class LazyAugurCommand(click.Command):
+    """
+    A lightweight proxy command that imports the real command only when invoked.
+    This prevents eager imports during `augur --help`.
+    """
+
+    def __init__(self, name: str, import_name: str, short_help: Optional[str]):
+        super().__init__(name=name, callback=self._proxy_callback, short_help=short_help)
+        self._import_name = import_name
+        self._real_cmd = None
+
+    def _load(self) -> click.Command:
+        if self._real_cmd is None:
+            module = importlib.import_module(self._import_name)
+            self._real_cmd = module.cli
+        return self._real_cmd
+
+    def _proxy_callback(self, *args, **kwargs):
+        # If click ends up calling callback directly, delegate
+        return self._load().callback(*args, **kwargs)
+
+    def invoke(self, ctx):
+        return self._load().invoke(ctx)
+
+    def get_params(self, ctx):
+        return self._load().get_params(ctx)
+
+    def get_help(self, ctx):
+        return self._load().get_help(ctx)
 
-CONTEXT_SETTINGS = dict(auto_envvar_prefix='AUGUR')
 
 class AugurMultiCommand(click.MultiCommand):
-    def __commands_folder(self):
-        return os.path.abspath(os.path.dirname(__file__))
+    def __commands_folder(self) -> Path:
+        return Path(__file__).resolve().parent
 
     def list_commands(self, ctx):
         rv = []
-        for filename in os.listdir(self.__commands_folder()):
-            if not filename.startswith('_') and filename.endswith('.py'):
-                rv.append(filename[:-3])
+        for p in self.__commands_folder().iterdir():
+            if p.is_file() and p.suffix == ".py" and not p.name.startswith("_"):
+                rv.append(p.stem)
         rv.sort()
         return rv
 
     def get_command(self, ctx, name):
-        cmdfile = "augur/application/cli" / Path(name + ".py")
-
-        # Check that the command exists before importing
+        cmdfile = self.__commands_folder() / f"{name}.py"
         if not cmdfile.is_file():
-            return
+            return None
+        short_help = _extract_short_help(cmdfile)
+        import_name = f"augur.application.cli.{name}"
+        return LazyAugurCommand(name=name, import_name=import_name, short_help=short_help)
 
-        # Prefer to raise exception instead of silcencing it
-        module = importlib.import_module('.' + name, 'augur.application.cli')
-        return module.cli
 
 @click.command(cls=AugurMultiCommand, context_settings=CONTEXT_SETTINGS)
 @click.pass_context