现在,我有一个可以使用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
来访问命名空间,从而最大限度地减少了对属性名称的假设。
使用子分析器时,子分析器以"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
函数运行剩余的默认值。
如果您试图将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