start work on in house shell completer classes #156

Author: bckohan

django_typer/management/commands/shellcompletion.py

@@ -46,12 +46,9 @@
 from django.utils.translation import gettext_lazy as _
 from shellingham import ShellDetectionFailure, detect_shell
 from typer import Argument, Option, echo
-from typer.completion import (  # type: ignore
-    Shells,  # pyright: ignore[reportPrivateImportUsage]
-    completion_init,  # pyright: ignore[reportPrivateImportUsage]
-)
 
 from django_typer.management import TyperCommand, command, get_command, initialize
+from django_typer.management.commands.shells import Shells, completion_init
 from django_typer.types import COMMON_PANEL
 from django_typer.utils import get_usage_script
 
@@ -65,6 +62,7 @@
 DJANGO_COMMAND = Path(__file__).name.split(".")[0]
 
 
+
 class Command(TyperCommand):
     """
     This command installs autocompletion for the current shell. This command uses the typer/click

django_typer/management/commands/shells/__init__.py

@@ -0,0 +1,23 @@
+from enum import Enum
+from click.shell_completion import add_completion_class
+
+
+class Shells(str, Enum):
+    bash = "bash"
+    zsh = "zsh"
+    fish = "fish"
+    powershell = "powershell"
+    pwsh = "pwsh"
+
+
+def completion_init():
+    from .bash import BashComplete
+    from .fish import FishComplete
+    from .powershell import PowerShellComplete
+    from .zsh import ZshComplete
+
+    add_completion_class(BashComplete, Shells.bash)
+    add_completion_class(ZshComplete, Shells.zsh)
+    add_completion_class(FishComplete, Shells.fish)
+    add_completion_class(PowerShellComplete, Shells.powershell)
+    add_completion_class(PowerShellComplete, Shells.pwsh)

django_typer/management/commands/shells/bash.py

@@ -0,0 +1,6 @@
+from click.shell_completion import BashComplete as ClickBashComplete
+
+
+class BashComplete(ClickBashComplete):
+    pass
+

django_typer/management/commands/shells/fish.py

@@ -0,0 +1,6 @@
+from click.shell_completion import FishComplete as ClickFishComplete
+
+
+class FishComplete(ClickFishComplete):
+    pass
+

django_typer/management/commands/shells/powershell.py

@@ -0,0 +1,6 @@
+from click.shell_completion import ShellComplete
+
+
+class PowerShellComplete(ShellComplete):
+    pass
+

django_typer/management/commands/shells/zsh.py

@@ -0,0 +1,5 @@
+from click.shell_completion import ZshComplete as ClickZshComplete
+
+
+class ZshComplete(ClickZshComplete):
+    pass

justfile

@@ -34,10 +34,10 @@ check-package:
     poetry run pip check
 
 clean-docs:
-    python -c "from shutil import rmtree; rmtree('./doc/build', ignore_errors=True)"
+    python -c "import shutil; shutil.rmtree('./doc/build', ignore_errors=True)"
 
 clean-env:
-    python -c "import shutil, sys; shutil.rmtree(sys.argv[1])" $(poetry env info --path)
+    python -c "import shutil, sys; shutil.rmtree(sys.argv[1], ignore_errors=True)" $(poetry env info --path)
 
 clean-git-ignored:
     git clean -fdX