Python将字典转换为argparse

Posted 2023-03-25

【中文标题】Python将字典转换为argparse

【英文标题】：Python convert dictionary to argparse

【发布时间】：2021-09-24 16:06:08

【问题描述】：

现在，我有一个可以使用 argparse 接受命令行参数的脚本。例如，像这样：

#foo.py
def function_with_args(optional_args=None):
  parser = argparse.ArgumentParser()
  # add some arguments
  args = parser.parse_args(optional_args)
  # do something with args

但是，我希望能够将此函数与字典一起使用，例如：

def function_using_dict(**kwargs):
  # define parser and add some arguments
  args = parser.parse_dict_args(kwargs)
  # everything else is the same

请注意，我想使用 argparse 中有很多带有默认值的参数，因此以下方法不起作用：

def function_no_default_args(**kwargs):
  args = kwargs # not using default values that we add to the parser!

【问题讨论】：

您是否只想将一组值注入Namespace 对象？您可以直接这样做：args = argparse.Namespace(**kwargs)。 parse_args的重点是将字符串列表转换为Namespace对象；目前尚不清楚您的 dict 中有什么，或者您希望 argparse 以其他方式处理它。

那是个好主意。有没有办法使用解析器获取默认值，比如ns = parser.parse_args([]); ns.update(kwargs)？ （但问题是命名空间似乎没有“更新”功能）

您可以使用parser.parse_args([], ns)，尽管parse_args 不会考虑ns 中已有的值。如果您希望您的kwargs 优先于parse_args，您可以执行n = Namespace(** **vars(p.parse_args([])), **kwargs) 之类的操作。

【参考方案1】：

argparse.Namespace 是一个相对简单的对象子类，它的大部分代码都用于显示属性（如print(args) 所示）。在内部 parse_args 使用 get_attr 和 set_attr 访问命名空间，最大限度地减少对属性名称的假设。

使用子解析器时，子解析器以“空白”命名空间开始，并使用以下代码将其值复制到主命名空间。

    # In case this subparser defines new defaults, we parse them
    # in a new namespace object and then update the original
    # namespace for the relevant parts.
    subnamespace, arg_strings = parser.parse_known_args(arg_strings, None)
    for key, value in vars(subnamespace).items():
        setattr(namespace, key, value)

最初主命名空间被传递给子解析器，例如。 parser.parse_known_args(arg_strings, namespace)，但当前版本让子解析器默认优先。

处理默认值有点复杂。如果您没有任何必需的参数，那么

 args = parser.parse_args([])

将设置所有默认值。或者您可以查看parse.parse_known_args 的开头，以了解在解析开始时如何将默认值插入到命名空间中。请注意，在解析结束时有一个附加步骤，通过它们各自的 type 函数运行剩余的默认值。

【参考方案2】：

如果您尝试将parse_args 的结果转换为dict，您可以这样做：

kwargs = vars(args)

在你的评论之后，我考虑了一下。将使用您现有的功能。

#foo.py
def function_with_args_and_default_kwargs(optional_args=None, **kwargs):
  parser = argparse.ArgumentParser()
  # add some arguments
  # add the other arguments
  for k, v in kwargs.items():
    parser.add_argument('--' + k, default=v)
  args = parser.parse_args(optional_args)
  # do something with args

【讨论】：

嗨，本，感谢您的快速回复。我的意思是反其道而行之。我有一本字典，但我仍然希望能够使用 argparse 对其进行“解析”，这样我也可以获取具有默认值的参数。

您不想要描述，还是只想让参数（假设所有字符串）具有默认值？

你说得对，在这种情况下，我不需要为我的论点提供 argparse 描述。我只是希望它们有一个我已经用 argparse 指定的默认值。

不得不做一些修复，但我认为迭代 kwargs 并将它们添加到您的解析器会起作用，尽管您不知道描述等。