import argparse
class customAction(argparse.Action):
def __call__(self, parser, args, values, option_string=None):
setattr(args, self.dest, values)
parser = argparse.ArgumentParser()
parser.add_argument('-e', '--example', action=customAction)
I want to pass additional arguments to customAction when the option -e is triggered, e.g. a instance of another class. How can I do this? Everything I have tried has errored out.
def make_action(additional_arg):
class customAction(argparse.Action):
def __call__(self, parser, args, values, option_string=None):
print(additional_arg)
setattr(args, self.dest, values)
return customAction
#...
parser.add_argument('-e', '--example', action=make_action('your arg'))
Another solution is to derive the based class argparse.Action like this:
class CustomAction(argparse.Action):
def __init__(self,option_strings,
additional_arg1,additional_arg2,
dest=None,
nargs=0,
default=None,
required=False,
type=None,
metavar=None,
help=None):
self._a1=additional_arg1
self._a2=additional_arg2
super(CustomAction, self).__init__(
option_strings=option_strings,
dest=dest,
nargs=nargs,
default=default,
required=required,
metavar=metavar,
type=type,
help=help)
def __call__(self, parser, namespace, values, option_string=None):
print(self._a1)
print(self._a2)
setattr(namespace, self.dest, values)
#........
parser.add_argument('-e', '--example', action=CustomAction, additional_arg1='your arg', additional_arg2=42)
Alternatively, supply *args and **kwargs to pass through any additional parameters to the parent constructor.
class CustomAction(argparse.Action):
def __init__(self, option_strings, additional_arg1, additional_arg2,
*args, **kwargs):
self._a1 = additional_arg1
self._a2 = additional_arg2
super(CustomAction, self).__init__(option_strings=option_strings,
*args, **kwargs)
def __call__(self, parser, namespace, values, option_string=None):
print(self._a1)
print(self._a2)
setattr(namespace, self.dest, values)
#........
parser.add_argument('-e', '--example', action=CustomAction, additional_arg1='your arg', additional_arg2=42)
*args and **kwargs instead of listing out all of argparse's default Action arguments.
