将可选列表传递给argparse

Question

有没有比传递分隔字符串并稍后解析它更优雅的方法来传递可选的整数列表到Argparse？我也有一个位置论点。

parser.add_argument('--ids', type=int, nargs='+')
parser.add_argument('cmd')

不工作，因为artparse试图获取cmd并抱怨它不是整数。

理想情况下，我想与其中一个执行

program.py --ids 6,32,12 refresh
program.py --ids 6 32 12 refresh

或者类似的东西，但也能

program.py refresh

Answer 1

如果您只想解析形式为--ids1,2,3（无空格）的参数，可以使用如下方法：

def convert(argument):
    return map(int, argument.split(','))  # 3.x: consider wrapping in list()

parser.add_argument('--ids', type=convert)

这不会处理由空白分隔的参数，尽管您可能可以通过更智能的转换（）函数来减轻这种情况。然而，您需要引用它们，否则shell会将它们作为单独的参数传递。

Answer 2

--是一种表示“位置参数从这里开始”的简便方法。

使用您的解析器，可以执行以下操作：

program.py refresh   # sets ids=None
program.py refresh --ids 1 2 3
program.py --ids 1 2 3 -- refresh

如果不喜欢None，可以为--ids参数指定一个默认值（例如[]）。

任何与program.pyreesh--id 1,2,3有关的问题都是由于shell如何分割命令行造成的。查看sys.argv列表。

出现program.py--id 1 2 3刷新的问题是因为在处理--id时，解析器尝试使用所有不明显标志的字符串（例如'-'）。它不使用int类型来测试使用哪些类型和离开哪些类型。

现在，如果id是定位的，那么它将处理123refresh：

parser.add_argument('ids',type=int,nargs='+')
parser.add_argument('cmd')
parser.parse_args('1 2 3 refresh'.split())

但这是因为解析器使用不同的策略将字符串分配给几个位置参数。它使用一个类似于A的匹配器。

Kevin的type方法可以通过一个简单的函数更好地实现：

def mytype(astring):
    ll = astring.split(',')
    return [int(l) for l in ll]
parser.add_argument('--ids', type=mytype)

它可以推广到处理带引号的字符串，如“1 2 3”type可以是任何接受字符串并返回所需值的函数，如果无法进行转换，则会引发错误。