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

时间:2013-12-27 22:35:53

标签: python python-2.7

说我有这样的事情:

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

但我并不确定如何制作它,因此它可以有多个输入并能够检索它们。

1 个答案:

答案 0 :(得分:5)

您可以允许选项接收任意数量的参数using the nargs parameter

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