Add target argument only if present in sig

2020-08-28 18:39:16 +02:00 · 2020-08-28 18:39:16 +02:00 · e97e900fc6
commit e97e900fc6
parent 7e9ec37103
1 changed files with 5 additions and 4 deletions
--- a/shlax/cli.py
+++ b/shlax/cli.py
@ -42,10 +42,11 @@ class TargetArgument(cli2.Argument):
 class Command(cli2.Command):
    def setargs(self):
        super().setargs()
-        self['target'] = TargetArgument(
-            self,
-            self.sig.parameters['target'],
-        )
+        if 'target' in self.sig.parameters:
+            self['target'] = TargetArgument(
+                self,
+                self.sig.parameters['target'],
+            )
        if 'actions' in self:
            del self['actions']