是否可以使用 Python 的 argparse 重建命令行？

Question

我有一个Python脚本，它读取一个包含其他工具的命令行调用的文件。在调用该工具之前，我想修改此调用的选项。例如，我可能会转换：

my_util --input file1.txt --option1 red --option2 blue

到此：

my_util --input file1_001.txt --option1 red --option3 green

（更准确地说，我会以列表的形式处理这些论点。）

我认为使用argparse模块是最简单的方法：我可以解析参数，根据需要更改、添加或删除选项，然后重建命令行。

但是我该怎么做最后一步呢？给定parse_args()返回的Namespace对象，我是否可以轻松地重建命令行选项列表，例如可以传递给subprocess.Popen()？

Answer 1

Namespace对象只是一个简单的对象子类，因此您可以使用vars:将值作为dict输出

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo')
>>> args = parser.parse_args(['--foo', 'BAR'])
>>> vars(args)
{'foo': 'BAR'}

或者，您可以直接分配给一个类，并将参数作为类变量输出：

>>> class C(object):
...     pass
...
>>> c = C()
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo')
>>> parser.parse_args(args=['--foo', 'BAR'], namespace=c)
>>> c.foo
'BAR'

使用这两种结构中的任何一种来测试/替换参数并将结果传递给Popen都相当容易。

Answer 2

我知道这是一个老问题，但我刚刚遇到了同样的问题。我意识到我所需要的只是一种迭代Action对象的方法。不幸的是，ArgParser本身并没有公开内部列表。但是，这些对象是由add_argument()返回的，所以我可以构建自己的列表。对我来说，在每个调用周围放置actions.append()似乎太多了，所以我将所有选项存储在一个元组中：

def add_argument(*args, **kwargs):
    return (args, kwargs)
parser = argparse.ArgumentParser()
options = (
    add_argument('--verbose', action='store_true'),
    add_argument('--author'),
    add_argument('--subject', required=True),
    add_argument('--cache', nargs='?'),
    add_argument('files', nargs=''),
)
actions = []
for (args, kwargs) in options:
    actions.append(parser.add_argument(*args, **kwargs))
args = parser.parse_args()

此时，选项在args中解析，并且所有argparse.Action对象都存储在actions列表中。然后我可以迭代这个列表，并重建如下选项：

cmdline = []
for action in actions:
    value = getattr(args, action.dest)
    if action.required or value != action.default:
        if action.option_strings:
            cmdline.append(action.option_strings[0])
        if action.nargs is None:
            cmdline.append(value)
        elif action.nargs == '?':
            if value != action.const:
                cmdline.append(value)
        elif action.nargs != 0:
            cmdline += value

在我的特定情况下，我还想从命令行中删除一些选项。为此，我只是通过调用parser.add_argument()而不是通过options元组将它们单独添加。