在argparse中使用变量arg名称

Question

我有一个上游系统用不同的arg名称调用我的程序。例如：

foo --xyz1 10 --xyz2 25 --xyz3 31 

我想argparsing的结果是一个xyz = [10,25,31]。

我的参数的名称有一个共同的前缀，但不幸的是必须至少使用不同的数字后缀，它也表示顺序。我也没有固定数量的参数。

有没有办法用argparse来建模？可以通过一些内置功能的组合来实现，或者通过重写/插入一些自定义分析器处理。

Answer 1

我建议有点预处理来实现这一目标：

代码：

def get_xyz_cmd_line(xyz_cmd_line): 
    # build a generator to iterate the cmd_line 
    cmd_line_gen = iter(xyz_cmd_line) 

    # we will separate the xyz's from everything else 
    xyz = [] 
    remaining_cmd_line = [] 

    # go through the command line and extract the xyz's 
    for opt in cmd_line_gen: 
     if opt.startswith('--xyz'): 
      # grab the opt and the arg for it 
      xyz.append((opt, cmd_line_gen.next())) 
     else: 
      remaining_cmd_line.append(opt) 

    # sort the xyz's and return all of them as -xyz # -xyz # ... 
    return list(it.chain(*[ 
     ('--xyz', x[1]) for x in sorted(xyz)])) + remaining_cmd_line 

测试：

import argparse 
import itertools as it 

parser = argparse.ArgumentParser(description='Get my Option') 
parser.add_argument('--an_opt', metavar='N', type=int, 
        help='An option') 
parser.add_argument('--xyz', metavar='N', type=int, action='append', 
        help='An option') 

cmd_line = "--an_opt 1 --xyz1 10 --xyz3 31 --xyz2 25 ".split() 
args = parser.parse_args(get_xyz_cmd_line(cmd_line)) 
print(args) 

输出：

Namespace(an_opt=1, xyz=[10, 25, 31]) 

使用方法：

名义上不是固定cmd_line作为在上面的例子中，这将喜欢的东西被称为：

args = parser.parse_args(get_xyz_cmd_line(sys.argv[1:])) 

更新：如果您需要--xyz = 31（即=分隔符）：

然后您将需要更改：

# grab the opt and the arg for it 
xyz.append((opt, cmd_line_gen.next())) 

要：

if '=' in opt: 
    xyz.append(tuple(opt.split('=', 1))) 
else: 
    # grab the opt and the arg for it 
    xyz.append((opt, cmd_line_gen.next())) 

Answer 2

这里就是我做参考（快速和肮脏的版本），虽然我也很喜欢史蒂芬·劳赫的回答（所以我会标记，作为一个答案 - ESP。因为我用于我的溶液内部的实现细节）：

class CustomArgumentsParser(argparse.ArgumentParser): 

    def _parse_optional(self, arg_string): 
    suffix_index = arg_string.find(':') 
    if suffix_index < 0: 
     return super(CustomArgumentParser, self)._parse_optional(arg_string) 

    original_arg_string = arg_string 
    suffix = arg_string[suffix_index + 1:] 
    arg_string = arg_string[0:suffix_index] 

    option_tuple = super(CustomArgumentParser, self)._parse_optional(arg_string) 
    if not option_tuple: 
     return option_tuple 

    action, option_string, explicit_arg = option_tuple 
    if isinstance(action, BuildListAction): 
     return action, suffix, explicit_arg 
    else: 
     self.exit(-1, message='Unknown argument %s' % original_arg_string) 


class BuildListAction(argparse.Action): 
    def __init__(self, 
       option_strings, 
       dest, 
       nargs=None, 
       const=None, 
       default=None, 
       type=None, 
       choices=None, 
       required=False, 
       help=None, 
       metavar=None): 
    super(BuildListAction, self).__init__(
     option_strings=option_strings, 
     dest=dest, 
     nargs=nargs, 
     const=const, 
     default=default, 
     type=type, 
     choices=choices, 
     required=required, 
     help=help, 
     metavar=metavar) 

    def __call__(self, parser, namespace, values, option_string=None): 
    index = int(option_string) - 1 

    list = getattr(namespace, self.dest) 
    if list is None: 
     list = [] 
     setattr(namespace, self.dest, list) 

    if index >= len(list): 
     list.extend([self.default] * (index + 1 - len(list))) 
    list[index] = values 

用法：

argparser = CustomArgumentsParser() 
argparser.add_argument('--xyz', type=int, action=BuildListAction) 

注 - 这支持形式--xyz的ARGS：1，--xyz：2，..这与原来的问题略有不同。