现在,我有一个可以使用 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!
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
函数运行剩余的默认值。
如果您尝试将
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
import argparse
config_dict = {'width': 1920, 'height': 1080}
namespace = argparse.Namespace(**config_dict)
来源:https://blog.finxter.com/converting-python-dictionaries-to-argparse-namespaces/