Support SkipParse decorator

Author: link89

fire/core.py

@@ -722,7 +722,18 @@ def _MakeParseFn(fn, metadata):
   required_kwonly = set(fn_spec.kwonlyargs) - set(fn_spec.kwonlydefaults)
 
   def _ParseFn(args):
-    """Parses the list of `args` into (varargs, kwargs), remaining_args."""
+    """Parses the list of `args` into (varargs, kwargs), consumed_args, remaining_args."""
+
+    skip_parse = metadata.get(decorators.SKIP_PARSE, False)
+
+    if skip_parse:
+        kwargs = {}
+        remaining_kwargs = []
+        remaining_args = []
+        varargs = consumed_args = args[:]
+        capacity = False
+        return (varargs, kwargs), consumed_args, remaining_args, capacity
+
     kwargs, remaining_kwargs, remaining_args = _ParseKeywordArgs(args, fn_spec)
 
     # Note: _ParseArgs modifies kwargs.
@@ -754,8 +765,16 @@ def _ParseFn(args):
     varargs = parsed_args + varargs
     remaining_args += remaining_kwargs
 
-    consumed_args = args[:len(args) - len(remaining_args)]
-    return (varargs, kwargs), consumed_args, remaining_args, capacity
+    sorted_remaining_args = []
+    consumed_args = []
+    for arg in args:
+      if arg in remaining_args:
+        sorted_remaining_args.append(arg)
+        remaining_args.remove(arg)
+      else:
+        consumed_args.append(arg)
+
+    return (varargs, kwargs), consumed_args, sorted_remaining_args, capacity
 
   return _ParseFn
 
@@ -989,4 +1008,4 @@ def _ParseValue(value, index, arg, metadata):
     elif default is not None:
       parse_fn = default
 
-  return parse_fn(value)
+  return parse_fn(value)

fire/decorators.py

@@ -27,6 +27,14 @@
 FIRE_METADATA = 'FIRE_METADATA'
 FIRE_PARSE_FNS = 'FIRE_PARSE_FNS'
 ACCEPTS_POSITIONAL_ARGS = 'ACCEPTS_POSITIONAL_ARGS'
+SKIP_PARSE = 'SKIP_PARSE'
+
+
+def SkipParse(fn):
+  """Set a flag to tell Fire to pass original args to decorated fn.
+  """
+  _SetMetadata(fn, SKIP_PARSE, True)
+  return fn
 
 
 def SetParseFn(fn, *arguments):

fire/decorators_test.py

@@ -90,6 +90,13 @@ def example7(self, arg1, arg2=None, *varargs, **kwargs):  # pylint: disable=keyw
     return arg1, arg2, varargs, kwargs
 
 
+class SkipParseArgs(object):
+
+  @decorators.SkipParse
+  def example8(self, *args):
+    return args
+
+
 class FireDecoratorsTest(testutils.BaseTestCase):
 
   def testSetParseFnsNamedArgs(self):
@@ -169,6 +176,11 @@ def testSetParseFn(self):
                   command=['example7', '1', '--arg2=2', '3', '4', '--kwarg=5']),
         ('1', '2', ('3', '4'), {'kwarg': '5'}))
 
+  def testSkipParse(self):
+    command = ['example8', 'test', '1', '--arg2=2', '3', '4', '--kwarg=5', '--flag']
+    self.assertEqual(
+        core.Fire(SkipParseArgs, command=command), tuple(command[1:]))
+
 
 if __name__ == '__main__':
   testutils.main()