我想写一个简单的工具任意数量的输入文件并执行one对他们每个人进行操作。语法非常简单:
mytool operation input1 input2 ... inputN
其中一些操作可能需要额外的参数
mytool operation op_argument input1 input2 ... inputN
除此之外,我希望用户能够指定是否应就地执行操作,并指定输出的目标目录。
mytool -t $TARGET --in-place operation op_argument input1 input2 input3
作为最后一个要求,我希望用户能够获得有关每个操作以及整个工具的使用的帮助。
这是我为上述工具设计一个参数解析器的尝试,以及一个最小的、完整的、可验证的示例:
#!/bin/env python
import argparse
from collections import namedtuple
Operations = namedtuple('Ops', 'name, argument, description')
IMPLEMENTED_OPERATIONS = {'echo': Operations('echo',
None,
'Echo inputs'),
'fancy': Operations('fancy',
'fancyarg',
'Do fancy stuff')}
if __name__ == "__main__":
# Parent parser with common stuff.
parent = argparse.ArgumentParser(add_help=False)
parent.add_argument('-t', '--target-directory', type=str, default='.',
help="An output directory to store output files.")
parent.add_argument('-i', '--in-place', action='store_true',
help="After succesful execution, delete the inputs.")
# The inputfiles should be the very last positional argument.
parent.add_argument('inputfiles', nargs='*', type=argparse.FileType('r'),
help="A list of input files to operate on.")
# Top level parser.
top_description = "This is mytool. It does stuff"
parser = argparse.ArgumentParser(prog="mytool",
description=top_description,
parents=[parent])
# Operation parsers.
subparsers = parser.add_subparsers(help='Sub-command help', dest='chosen_op')
op_parsers = {}
for op_name, op in IMPLEMENTED_OPERATIONS.items():
op_parsers[op_name] = subparsers.add_parser(op_name,
description=op.description,
parents=[parent])
if op.argument is not None:
op_parsers[op_name].add_argument(op.argument)
args = parser.parse_args()
op_args = {}
for key, subparser in op_parsers.items():
op_args[key] = subparser.parse_args()
print(args.chosen_op)
我遇到的问题是位置参数的顺序是错误的。不知何故,我实现这一点的方式使 Argparse 认为该操作(及其 op_argument)应该出现在输入文件之后,但显然情况并非如此。
如何将父位置参数(在我的例子中为输入文件)作为最后一个位置参数?