如何将 argparse 参数解析为字符串列表，在预定义的标志参数处停止？

Question

如何解析由特殊预定义语法分隔的可变长度参数列表。一个例子：

   ./script --arg1 --cmdname otherscript --a1 --a2 --cmdname-- --arg3

用argparse解析后脚本应该有三个参数：arg1、cmdname、arg3。参数cmdname 应包含三个值的列表otherscript、a1、a2。

拥有这样的配方对于能够将cmdname 中的所有内容传递到subprocess.popen(cmdname, ...) 调用中会很有用。

我在考虑子解析器。但我相信子解析器无法停止，并且与其他子解析器确实是互斥的。还有其他简单的，已经提供的方法吗？将Action 子类化是一种方法吗？

Answer 1

如果我们区分以-- 开头的参数和同样以-- 开头的命令部分，这将有助于 argparse。

所以如果./script ++arg1 ++cmdname otherscript --a1 --a2 ++arg3 是可以接受的，那么：

import argparse
import shlex

parser = argparse.ArgumentParser(prefix_chars = '+')
parser.add_argument('++arg1', action = 'store_true')
parser.add_argument('++arg3', action = 'store_true')
parser.add_argument('++cmdname', nargs = '*')
args = parser.parse_args(shlex.split("++arg1 ++cmdname otherscript --a1 --a2 ++arg3")) 
print(args)

产量

Namespace(arg1=True, arg3=True, cmdname=['otherscript', '--a1', '--a2'])

Answer 2

正如您在帖子中指出的那样，子类化 Action 可能是这样做的方法——尽管如果 argparse 不知道 otherscript 的参数，这会变得非常棘手。您可能可以使用parse_known_args 解决此问题，但您可能无法解决。老实说，我真的认为最简单的方法是自己预处理sys.argv。

import shlex
s = shlex.split("./script --arg1 --cmdname otherscript --a1 --a2 --cmdname-- --arg3")
def preprocess(lst):
    """
    process an iterable into 2 lists.
    The second list contains the portion bracketed by '--cmdname' and '--cmdname--'
    whereas the first portion contains the rest of it.
    """
    argv1,argv2 = [],[]
    current = argv1
    for i in lst:
        if i == '--cmdname':
           current = argv2
        elif i == '--cmdname--':
           current = argv1
        else:
           current.append(i)
    return argv1,argv2

l1,l2 = preprocess(s)
print l1
print l2

preprocess 的替代实现适用于具有 .index 方法的可切片对象 -- sys.argv 可以正常工作：

def preprocess(lst):
    """
    process an iterable into 2 lists.
    The second list contains the portion bracketed by '--cmdname' and '--cmdname--'
    whereas the first portion contains the rest of it.
    """
    try:
        i1 = lst.index('--cmdname')
        i2 = lst.index('--cmdname--')
        argv1 = lst[i1+1:i2]
        argv2 = lst[:i1]+lst[i2+1:]
    except ValueError:
        argv1 = lst
        argv2 = []

    return argv1,argv2

---

另一个选择（在@unutbu 的出色评论中指出）是将命令行语法更改为更标准的东西，这大大简化了问题：

./script --arg1 --cmd "otherscript --a1 --a2" --arg3

然后您可以像通常使用argparse 一样解析cmd（为此参数指定type=shlex.split 以从字符串转换为参数列表）。