在使用argparse库进行命令行参数解析时,可以使用类型验证(type validation)来确保输入参数的类型正确。下面是一个示例代码,其中演示了如何使用argparse进行类型验证,并对nargs参数进行验证:
import argparse
def positive_int(value):
ivalue = int(value)
if ivalue <= 0:
raise argparse.ArgumentTypeError("%s is not a positive integer" % value)
return ivalue
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--numbers', nargs='+', type=positive_int)
args = parser.parse_args()
print("Numbers:", args.numbers)
if __name__ == "__main__":
main()
在上面的示例代码中,我们定义了一个类型验证函数positive_int
,它接受一个值作为输入,并将其转换为整数。在验证过程中,我们检查该值是否为正整数,如果不是,则抛出一个argparse.ArgumentTypeError
异常。
在argparse.ArgumentParser
中,我们使用nargs='+'
来指定参数--numbers
可以接受多个值,并将这些值作为一个列表存储在args.numbers
中。通过将type=positive_int
设置为参数的类型验证函数,我们确保了输入的每个值都是正整数。
当我们运行上述代码并提供一些正整数作为--numbers
参数时,它将打印出输入的数字列表。如果输入的任何一个值不是正整数,将会抛出一个错误。
以下是一些示例运行命令和输出的例子:
$ python script.py --numbers 1 2 3 4
Numbers: [1, 2, 3, 4]
$ python script.py --numbers 1 2 a 4
usage: script.py [-h] [--numbers NUMBERS [NUMBERS ...]]
script.py: error: argument --numbers: invalid positive_int value: 'a'