python - 如何使用 argparse 创建类型为 "list of pairs"的参数？

Question

我需要让我的 python 脚本的最终用户键入如下内容:

script.py -sizes <2,2> <3,3> <6,6>

-sizes 选项的每个元素都是一对两个正整数。如何使用 argparse 实现此目的？

Answer 1

定义自定义类型:

def pair(arg):
    # For simplity, assume arg is a pair of integers
    # separated by a comma. If you want to do more
    # validation, raise argparse.ArgumentError if you
    # encounter a problem.
    return [int(x) for x in arg.split(',')]

然后将其用作常规参数的类型:

p.add_argument('--sizes', type=pair, nargs='+')

然后

>>> p.parse_args('--sizes 1,3 4,6'.split())
Namespace(sizes=[[1, 3], [4, 6]])