没有破折号但带有附加参数的可选python参数？

Question

我想在 Python 中做的是接受以下格式的参数：

script.py START | STOP | STATUS | MOVEABS <x> <y> | MOVEREL <x> <y>

也就是说，

1. 我不想处理连字符；
2. 我有多种可能性，其中一种是必需的；
3. 每个都是互斥的；
4. 某些命令（例如 moveabs 和 moverel）具有额外的必需参数，但这些参数不应与任何其他参数一起出现。

这可以在 python 中完成吗？我会使用 argparse 还是其他东西？谢谢。

Answer 1

add_parser 使用子解析器可以解决问题

import argparse
parser = argparse.ArgumentParser(prog='script.py')
sp = parser.add_subparsers(dest='cmd')
for cmd in ['START', 'STOP', 'STATUS']:
    sp.add_parser(cmd)
for cmd in ['MOVEABS', 'MOVEREL']:
    spp = sp.add_parser(cmd)
    spp.add_argument('x', type=float)
    spp.add_argument('y', type=float)
parser.print_help()
args = parser.parse_args()
print(args)

制作类似的东西：

2137:~/mypy$ python2.7 stack23304740.py MOVEREL -h
usage: script.py [-h] {START,STOP,STATUS,MOVEABS,MOVEREL} ...

positional arguments:
  {START,STOP,STATUS,MOVEABS,MOVEREL}

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

usage: script.py MOVEREL [-h] x y

positional arguments:
  x
  y

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

和

2146:~/mypy$ python2.7 stack23304740.py MOVEREL 1.0 2.0
...
Namespace(cmd='MOVEREL', x=1.0, y=2.0)

和

2147:~/mypy$ python2.7 stack23304740.py START
...
Namespace(cmd='START')

MOVEREL 参数可以命名为<x> 和<y>，但是您必须通过args['<y>'] 而不是args.y 访问它们。 metavar='<x>' 可用于更改显示，但不能更改命名空间名称。

您也可以使用spp.add_argument('point', nargs=2, type=float)。不幸的是，在nargs=2 案例http://bugs.python.org/issue14074 中，有一个错误使我们无法使用元变量。

Answer 2

使用docopt 你可以很容易地做到这一点。

先安装docopt：

$ pip install docopt

写script.py:

"""
Usage:
    script.py (start | stop | status | moveabs <x> <y> | moverel <x> <y>)
"""
from docopt import docopt

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

并运行它：

首先显示基本帮助：

$ python script.py
Usage:
    script.py (start | stop | status | moveabs <x> <y> | moverel <x> <y>)

然后尝试子命令：

开始

$ python script.py start
{'<x>': None,
 '<y>': None,
 'moveabs': False,
 'moverel': False,
 'start': True,
 'status': False,
 'stop': False}

停止

$ python script.py stop
{'<x>': None,
 '<y>': None,
 'moveabs': False,
 'moverel': False,
 'start': False,
 'status': False,
 'stop': True}

移动

$ python script.py moveabs 11 22
{'<x>': '11',
 '<y>': '22',
 'moveabs': True,
 'moverel': False,
 'start': False,
 'status': False,
 'stop': False}

移动

$ python script.py moverel 11 22
{'<x>': '11',
 '<y>': '22',
 'moveabs': False,
 'moverel': True,
 'start': False,
 'status': False,
 'stop': False}