4

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

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

...对此：

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

（更准确地说，我会将论点作为列表处理。）

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

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

4

2 回答 2

2

命名空间对象只是一个简单的对象子类，因此您可以使用以下命令将值作为 dict 取出vars：

>>> 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.

于 2013-02-12T00:38:07.383 回答
0

我知道这是一个老问题，但我刚刚遇到了同样的问题。我意识到我所需要的只是一种迭代 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元组单独添加它们。

于 2021-07-23T14:25:34.710 回答