Support SkipParse decorator

Author: link89

fire/core.py

@@ -723,6 +723,17 @@ def _MakeParseFn(fn, metadata):
 
   def _ParseFn(args):
     """Parses the list of `args` into (varargs, kwargs), 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.
@@ -755,6 +766,9 @@ def _ParseFn(args):
     remaining_args += remaining_kwargs
 
     consumed_args = args[:len(args) - len(remaining_args)]
+    # Ensure remaining_ars keeps the original order.
+    remaining_args = args[len(consumed_args):]
+
     return (varargs, kwargs), consumed_args, remaining_args, capacity
 
   return _ParseFn

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()