如何使用 argparse(python) 处理可选和强制位置参数

Question

我一直在尝试通过传递一些参数来执行我的 python 脚本。我想要的方式是这样的：

python script.py 文件路径

还要支持这个

python script.py 文件路径1 文件路径2

我的实现是：

parser = argparse.ArgumentParser()
mand = parser.add_argument_group("Mandatory arguments")
mand.add_argument("path1", help = "path1")
opt = parser.add_argument_group("Optional arguments")
opt.add_argument("path2", help = "path2")
args = parser.parse_args()

它似乎要求两个论点。有没有人对这样做的正确方法有什么建议？我的一个想法是做两个论证组；一个带有 path1，另一个带有两个路径，但它仍然请求两者。

谢谢大家！

附： Python 2.7

Answer 1

使用nargs="?"（与argparse）

import argparse
parser = argparse.ArgumentParser()
mand = parser.add_argument_group("Mandatory arguments")
mand.add_argument("path1", help = "path1")
opt = parser.add_argument_group("Optional arguments")
opt.add_argument("path2", help = "path2", nargs="?")
args = parser.parse_args()
print args

查看用法字符串：

$ python argscript.py 
usage: argscript.py [-h] path1 [path2]
argscript.py: error: too few arguments

或者-h:

$ python argscript.py -h
usage: argscript.py [-h] path1 [path2]

optional arguments:
  -h, --help  show this help message and exit

Mandatory arguments:
  path1       path1

Optional arguments:
  path2       path2

并仅使用一个位置参数进行尝试：

$ python argscript.py alfa
Namespace(path1='alfa', path2=None)

还有两个：

$ python argscript.py alfa beta
Namespace(path1='alfa', path2='beta')

Answer 2

抱歉，回答有点离题。我更喜欢使用docopt，以防你感兴趣。

使用docopt

先安装docopt：

$ pip install docopt

并使用脚本：

"""script.py
Usage:
    script.py <fname>...
"""

if __name__ == "__main__":
    from docopt import docopt
    args = docopt(__doc__)
    print args

你可以这样使用它：

询问使用脚本：

$ python script.py 
Usage:
    script.py <fname>...

用一个文件名调用：

$ python script.py alfa
{'<fname>': ['alfa']}

使用两个名称：

$ python script.py alfa beta
{'<fname>': ['alfa', 'beta']}

使用替代用法描述：

以下将允许一个或两个文件名，但不允许三个或更多：

"""script.py
Usage:
    script.py <input> [<output>]
"""

if __name__ == "__main__":
    from docopt import docopt
    args = docopt(__doc__)
    print args