自定义操作 argparse Python

Question

我想对 2 个不同的参数使用相同的自定义操作：

class TestAction(Action):

    def __call__(self, parser, namespace, values, option_string=None):
        car, color, state, code = values
        namespace.car = car
        namespace.color = color
        namespace.state = state
        namespace.code = code

当我定义参数时，我想按如下方式定义它们：

    parser.add_argument('--vehicle', nargs="2", action=TestAction, metavar=('car', 'color'), required=True)
    parser.add_argument('--country', nargs="2", type=tuple(str, int), action=TestAction,metavar=('state', 'code'), required=True)

如果我按照编写的方式使用自定义操作，则会收到以下错误：

ValueError: not enough values to unpack (expected 4, got 2)

I一直试图定义一些虚拟的“_”并在解包时用“_”填充空值，但没有成功。 第二个问题是我想强制代码为整数，但语法 tuple(str, int) 不正确。 有什么想法可以解决这两个问题吗？ 谢谢。

Answer 1

我认为更简单的方法是使用这样的东西（没有 argparse.Action）：

import argparse


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('car', type=str)
    parser.add_argument('color', type=str)
    parser.add_argument('state', type=str)
    parser.add_argument('code', type=int)
    args = parser.parse_args()
    print(args)

输出：

py .\test.py acar acolor acountry 1
>>> Namespace(car='acar', code=1, color='acolor', state='acountry')