From 563d9dc9274b790cf902580f4ed1825187159c69 Mon Sep 17 00:00:00 2001 From: Daylin Morgan Date: Tue, 22 Oct 2024 13:26:17 -0500 Subject: [PATCH] feat: add default option --- src/swydd/__init__.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/src/swydd/__init__.py b/src/swydd/__init__.py index 0b85cdb..3845d8d 100644 --- a/src/swydd/__init__.py +++ b/src/swydd/__init__.py @@ -135,9 +135,6 @@ class Context: for task in self._tasks.values(): if not task.targets: continue - print(task.name) - print(task.targets) - print(task.needs) for target in task.targets: if not task.needs: self._graph.add_nodes(task, target, None) @@ -676,7 +673,7 @@ def _task_repr(func: Callable) -> str: ) -def cli() -> None: +def cli(default: str | None = None) -> None: ctx._generate_graph() parser = ArgumentParser( @@ -708,11 +705,14 @@ def cli() -> None: for task in ctx._tasks.values(): _generate_task_subparser(shared, subparsers, task) - # TODO: add support for default arg? if len(sys.argv) == 1: - parser.print_help(sys.stderr) - sys.exit(1) - elif "--" in sys.argv: + if default: + sys.argv.append(default) + else: + parser.print_help(sys.stderr) + sys.exit(1) + + if "--" in sys.argv: i = sys.argv.index("--") args = vars(parser.parse_args(sys.argv[1:i])) ctx.rest = sys.argv[i + 1 :]