Python脚本接受可选的多个输入

Question

说我有这样的事情:

python example.py -i input.txt

脚本需要至少有一个输入,我希望它可以选择接受多个输入,例如:

python example.py -i input1.txt -i input2.txt -i input3.txt

我查看了Python的argparse模块,所以我有类似的东西:

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-i", help="takes in an input file", action="store_true")

但我不确定如何制作它所以它可以有多个输入并能够检索它们.

Answer 1

您可以允许选项使用nargs参数接收任意数量的参数:

>>> import argparse
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('-i', nargs='+')
>>> parser.parse_args('-i input1.txt input2.txt input3.txt'.split())
Namespace(i=['input1.txt', 'input2.txt', 'input3.txt'])