使用argparse获取选定的子命令

当我用python argparse使用子命令时,我可以得到选定的参数。

parser = argparse.ArgumentParser() parser.add_argument('-g', '--global') subparsers = parser.add_subparsers() foo_parser = subparsers.add_parser('foo') foo_parser.add_argument('-c', '--count') bar_parser = subparsers.add_parser('bar') args = parser.parse_args(['-g, 'xyz', 'foo', '--count', '42']) # args => Namespace(global='xyz', count='42') 

所以args不包含'foo' 。 简单地写sys.argv[1]是不行的,因为可能的全局参数。 我怎样才能得到子命令本身?

关于argparse子命令的Python文档的底部解释了如何执行此操作:

 >>> parser = argparse.ArgumentParser() >>> parser.add_argument('-g', '--global') >>> subparsers = parser.add_subparsers(dest="subparser_name") # this line changed >>> foo_parser = subparsers.add_parser('foo') >>> foo_parser.add_argument('-c', '--count') >>> bar_parser = subparsers.add_parser('bar') >>> args = parser.parse_args(['-g', 'xyz', 'foo', '--count', '42']) >>> args Namespace(count='42', global='xyz', subparser_name='foo') 

你也可以使用刚才在我发现的例子之上引用的set_defaults()方法。

这是一个简单的任务function布局的例子。

 import argparse def task_a(alpha): print('task a', alpha) def task_b(beta, gamma): print('task b', beta, gamma) if __name__ == '__main__': parser = argparse.ArgumentParser() subparsers = parser.add_subparsers(dest='subparser') parser_a = subparsers.add_parser('task_a') parser_a.add_argument( '-a', '--alpha', dest='alpha', help='Alpha description') parser_b = subparsers.add_parser('task_b') parser_b.add_argument( '-b', '--beta', dest='beta', help='Beta description') parser_b.add_argument( '-g', '--gamma', dest='gamma', default=42, help='Gamma description') kwargs = vars(parser.parse_args()) globals()[kwargs.pop('subparser')](**kwargs)