change: Fix autocomplete for espefuse

Author: peterdragun

espefuse/__init__.py

@@ -62,6 +62,7 @@
     "--port",
     "-p",
     envvar="ESPTOOL_PORT",
+    type=click.Path(),
     help="Serial port device.",
 )
 @click.option(

espefuse/cli_util.py

@@ -120,7 +120,11 @@ def parse_args(self, ctx: click.Context, args: list[str]):
                     # so we need to account for BLOCK2 being processed separately
                     param.nargs = args.index(arg) - arguments_count + 1
                     param_changed = param
-                    if param.nargs == 0 and param.required:
+                    if (
+                        param.nargs == 0
+                        and param.required
+                        and not ctx.resilient_parsing
+                    ):
                         raise click.UsageError(
                             f"Command `{self.name}` requires the `{param.name}` "
                             "argument."
@@ -211,7 +215,11 @@ def parse_args(self, ctx: click.Context, args: list[str]):
         args = self._replace_deprecated_args(args)
         cmd_groups, used_cmds = self._split_to_groups(args)
 
-        if len(used_cmds) == 0:
+        # Add commands for shell completion
+        if ctx.resilient_parsing:
+            commands = init_commands(port=None, chip=ctx.obj["chip"], skip_connect=True)
+            commands.add_cli_commands(self)
+        elif len(used_cmds) == 0:
             self.get_help(ctx)
             ctx.exit()
 

esptool/__init__.py

@@ -311,6 +311,7 @@ def check_flash_size(esp: ESPLoader, address: int, size: int) -> None:
 @click.option(
     "--port",
     "-p",
+    type=click.Path(),
     default=os.environ.get("ESPTOOL_PORT", None),
     help="Serial port device.",
 )

esptool/cli_util.py

@@ -265,9 +265,11 @@ def get_command(self, ctx: click.Context, cmd_name: str) -> click.Command | None
 
     def resolve_command(
         self, ctx: click.Context, args: list[str]
-    ) -> tuple[str, click.Command, list[str]]:
+    ) -> tuple[str | None, click.Command | None, list[str]]:
         # always return the full command name
         _, cmd, args = super().resolve_command(ctx, args)
+        if cmd is None:
+            return None, None, args
         return cmd.name, cmd, args