首页 > 解决方案 > 使用 parse_known_args 一次解析多个子命令

问题描述

import argparse
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(help='config command help')

# create parser for config command 
parser_config = subparsers.add_parser('config', help='config help')
parser_config.add_argument('--configname', required=True, nargs=1)
parser_config.add_argument('--login', required=True, nargs=1)
parser_config.add_argument('--password', required=True, nargs=1)

parser_config1 = subparsers.add_parser('config1', help='config help')
parser_config1.add_argument('--configname', required=True, nargs=1)
parser_config1.add_argument('--login', required=True, nargs=1)
parser_config1.add_argument('--password', required=True, nargs=1)


rest = 'config --configname name1 --login abc1 --password def1 config1 --configname name2 --login abc2 --password def2'.split()

print(rest)
while rest:
        args,rest =  parser.parse_known_args(rest)
        argslist.append(args)
        print(args, rest)

给出:

$ python3 create_config.py 
['config', '--configname', 'name1', '--login', 'abc1', '--password', 'def1', 'config1', '--configname', 'name2', '--login', 'abc2', '--password', 'def2']
Namespace(configname=['name2'], login=['abc2'], password=['def2']) ['config1'] [Namespace(configname=['name2'], login=['abc2'], password=['def2'])]
usage: create_config.py config1 [-h] --configname CONFIGNAME --login LOGIN
                                --password PASSWORD
create_config.py config1: error: the following arguments are required: --configname, --login, --password

我希望看到

['config', '--configname', 'name1', '--login', 'abc1', '--password', 'def1', 'config1', '--configname', 'name2', '--login', 'abc2', '--password', 'def2']
Namespace(configname=['name1'], login=['abc1'], password=['def1']) ['config1', '--configname', 'name2', '--login', 'abc2', '--password', 'def2']
Namespace(configname=['name2'], login=['abc2'], password=['def2']) []

我怎样才能做到这一点?

标签: pythonpython-3.xparsingcommand-line-argumentsargparse

解决方案


config子解析器得到:

['--configname', 'name1', '--login', 'abc1', '--password', 'def1', 'config1', '--configname', 'name2', '--login', 'abc2', '--password', 'def2']

它解析

['--configname', 'name1', '--login', 'abc1', '--password', 'def1']

正如预期的那样。然后它会看到config1它无法处理的 (否positionals)。它把它放在extras列表中,并继续解析其余的

['--configname', 'name2', '--login', 'abc2', '--password', 'def2']

这会覆盖以前的optional's值,这就是您最终在args.

我认为如果两个子解析器采用不同的标志,那么这种方法会起作用。我认为在您之前的问题的链接中探讨了此类问题,但我必须研究它们才能确定。

无论如何,parse_known_args当它遇到一个它无法解析的字符串时,它不会只是退出。像常规一样,parse_args它尝试处理整个列表,只是它将剩菜作为列表返回,而不是引发错误。并且允许重复的选项,即使它们没有任何用处(除非定义为append操作)。


推荐阅读