Python argpars中任意数量参数的自定义解析函数

Question

我有一个通过命令行获取命名参数的脚本。其中一个参数可以多次提供。例如，我想运行一个脚本：

./script.py --add-net=user1:10.0.0.0/24 --add-net=user2:10.0.1.0/24 --add-net=user3:10.0.2.0/24

现在我想要一个argparse操作，它将解析每个参数并将结果存储在dict中，如下所示：

^{pr2}$

此外，如果没有提供值，则应该提供一个默认值。像

./script.py

应该有dict像：

{'user': '192.168.0.0/24'}

我相信我必须为argparse构建一个自定义操作。我想到的是：

class ParseIPNets(argparse.Action):
    """docstring for ParseIPNets"""
    def __init__(self, option_strings, dest, nargs=None, **kwargs):
        super(ParseIPNets, self).__init__(option_strings, dest, **kwargs)

    def __call__(self, parser, namespace, values, option_string=None):
        for value in values:
            location, subnet = values.split(':')
            namespace.user_nets[location] = subnet

parser = argparse.ArgumentParser(description='foo')
parser.add_argument('--add-net',
                    nargs='*',
                    action=ParseIPNets,
                    dest='user_nets',
                    help='Nets subnets for users. Can be used multiple times',
                    default={"user1": "198.51.100.0/24"})

args = parser.parse_args()

当我需要使用默认值时，这很好：

test.py
Namespace(user_nets={'user1': '198.51.100.0/24'})

但是，当我添加参数时，它们被附加到默认值中。我的期望是将它们添加到一个空的dict中：

test.py --add-net=a:10.0.0.0/24 --add-net=b:10.1.0.0/24
Namespace(user_nets={'a': '10.0.0.0/24', 'b': '10.1.0.0/24', 'user1': '198.51.100.0/24'})

怎样才能得到我想要的东西？在

Answer 1

使用可变的默认参数（在您的例子中是dict）通常不是一个好主意，请参见here了解解释：

Create a new object each time the function is called, by using a default arg to signal that no argument was provided (None is often a good choice).