fix #222

Author: bckohan

doc/source/changelog.rst

@@ -4,6 +4,11 @@
 Change Log
 ==========
 
+v3.2.2 (2025-07-17)
+===================
+
+* Fixed `Raise a CommandError instead of a KeyError if get_command does not find the command. <https://github.com/django-commons/django-typer/issues/222>`_
+
 v3.2.1 (2025-07-16)
 ===================
 

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "django-typer"
-version = "3.2.1"
+version = "3.2.2"
 requires-python = ">=3.9,<4.0"
 description = "Use Typer to define the CLI for your Django management commands."
 authors = [

src/django_typer/__init__.py

@@ -33,7 +33,7 @@
 and keep a tight version lock on Typer.
 """
 
-VERSION = (3, 2, 1)
+VERSION = (3, 2, 2)
 
 __title__ = "Django Typer"
 __version__ = ".".join(str(i) for i in VERSION)

src/django_typer/management/__init__.py

@@ -221,12 +221,15 @@ def get_command(
     :param no_color: whether to disable color
     :param force_color: whether to force color
     :param kwargs: t.Any other parameters to pass through to the command constructor
-    :raises ModuleNotFoundError: if the command is not found
+    :raises CommandError: if the command is not found
     :raises LookupError: if the subcommand is not found
     """
-    module = import_module(
-        f"{get_commands()[command_name]}.management.commands.{command_name}"
-    )
+    try:
+        module = import_module(
+            f"{get_commands()[command_name]}.management.commands.{command_name}"
+        )
+    except (KeyError, ModuleNotFoundError) as err:
+        raise CommandError(f"Unknown command: {command_name}") from err
     cmd: BaseCommand = module.Command(
         stdout=stdout,
         stderr=stderr,

src/django_typer/management/commands/shellcompletion.py

@@ -508,7 +508,7 @@ def get_completion() -> str:
                         cmd_idx += 1
                         try:
                             cmd = get_command(args[cmd_idx])
-                        except KeyError:
+                        except CommandError:
                             pass
                 except IndexError:
                     if command.endswith(" "):