Python 将字典转换为 argparse

Python convert dictionary to argparse

现在,我有一个脚本可以使用 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!

如果您尝试将 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.Namespace 是一个相对简单的对象子类,其大部分代码用于显示属性(如 print(args) 所示)。在内部 parse_args 使用 get_attrset_attr 访问命名空间,最小化对属性名称的假设。

使用子解析器时,子解析器以 'blank' 命名空间开始,并使用以下代码将其值复制到主命名空间。

    # 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 函数运行剩余的默认值。