Python argparse 自定义动作并传递额外参数

Question

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)

我想在触发 -e 选项时，把额外的参数传递给 customAction，比如说传递一个其他类的实例。我该怎么做呢？我尝试过的所有方法都出错了。

Answer 1

另一种解决方案是从基类 argparse.Action 继承，像这样：

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)

或者，你可以提供 *args 和 **kwargs，这样可以把任何额外的参数传递给父类的构造函数。

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)

Answer 2

这段代码是用来处理某些数据的。它可能涉及到一些循环、条件判断或者函数调用。具体来说，代码块中的内容会根据输入的数据进行不同的操作，最终输出结果。

如果你对代码的具体功能有疑问，可以尝试逐行阅读，看看每一行是做什么的，或者查找相关的编程基础知识，这样会帮助你更好地理解。

记得，编程就像做菜，先了解每个材料的作用，再一步步组合起来，就能做出美味的菜肴！

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