如何用argparse创建类型为“pair of list”的参数？

Question

我需要让我的python脚本类型有点像最终用户：

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

其中-sizes选项中的每个元素是一对两个正整数。我怎样才能达到这个​​？

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

Answer 2

Argparse不会尝试覆盖所有可能的输入数据格式。你总是可以得到sizes为字符串，并用几行代码解析它们：

import argparse 

parser = argparse.ArgumentParser() 
parser.add_argument('--sizes', nargs='+') 

args = parser.parse_args() 

try: 
    sizes = [tuple(map(int, s.split(',', maxsplit=1))) for s in args.sizes] 
except Exception: 
    print('sizes cannot be parsed') 

print(sizes) 

“特殊情况是不够的特殊打破规则。”