Allow setting default arguments for commands

Author: stefanv

example_pkg/.spin/cmds.py

@@ -7,7 +7,8 @@
 
 @click.command()
 @click.option("-f", "--flag")
-def example(flag):
+@click.option("-t", "--test", default="not set")
+def example(flag, test, default_kwd=None):
     """🧪 Example custom command.
 
     Accepts arbitrary flags, and shows how to access `pyproject.toml`
@@ -20,6 +21,11 @@ def example(flag):
     click.secho("Flag provided with --flag is: ", fg="yellow", nl=False)
     print(flag or None)
 
+    click.secho("Flag provided with --test is: ", fg="yellow", nl=False)
+    print(test or None)
+
+    click.secho(f"Default kwd is: {default_kwd}")
+
     click.secho("\nDefined commands:", fg="yellow")
     for section in commands:
         print(f"  {section}: ", end="")

example_pkg/pyproject.toml

@@ -48,3 +48,6 @@ package = 'example_pkg'
 "Install" = [
   "spin.cmds.pip.install"
 ]
+
+[tool.spin.kwargs]
+".spin/cmds.py:example" = {"test" = "default override", "default_kwd" = 3}

spin/__main__.py

@@ -94,6 +94,7 @@ def group(ctx):
         "spin.python": _cmds.meson.python,
         "spin.shell": _cmds.meson.shell,
     }
+    cmd_default_kwargs = toml_config.get("tool.spin.kwargs", {})
 
     for section, cmds in config_cmds.items():
         for cmd in cmds:
@@ -133,6 +134,18 @@ def group(ctx):
                     print(f"!! Could not load command `{func}` from file `{path}`.\n")
                     continue
 
+                default_kwargs = cmd_default_kwargs.get(cmd)
+                import functools
+
+                if default_kwargs:
+                    callback = cmd_func.callback
+                    cmd_func.callback = functools.partial(callback, **default_kwargs)
+
+                    # Also override option defaults
+                    for option in cmd_func.params:
+                        if option.name in default_kwargs:
+                            option.default = default_kwargs[option.name]
+
                 commands[cmd] = cmd_func
 
             group.add_command(commands[cmd], section=section)