Question

我有一段python代码，用于提取命令行参数，例如：

import argparse
if __name__ == "__main__":                # myScript.py
    parser = argparse.ArgumentParser()
    parser.add_argument("-i", "--int", help="Int with default", type=int, default=20)
    parser.add_argument("-swd", "--strwdef", help="string with default", type=str, default="DEFAULTSTRING")
    parser.add_argument("-snd", "--strnodef", help="string without default", type=str)
    parser.add_argument("-f", "--flag", help="binary flag", action='store_true')
    args = parser.parse_args()
    i = args.i
    swd = args.swd
    snd = args.snd
    f = args.flag

所以这适用于这样的命令：

python myScript.py -swd SomeString --int 35 -f

我想从JSON文件中读取这些参数。问题：

python 2.7中是否有内置功能来设置它，而不是编写我自己的解析json.load()结果的自定义代码来处理参数类型，默认值等？
如果有，上面示例的JSON文件的正确格式是什么，以及填充变量i，swd等的方式？

谢谢！

Answer 1

来自解析器的

args是一个argparse.Namespace对象，很容易变成带有

的字典

 vars(args)

-swd SomeString --int 35 -f会产生类似namespace(swd='SomeString', int=35, f=True)和{'swd':'SomeString', 'int':35, 'f':True}的内容。

等效的JSON将解码为类似的字典。

可以使用以下命令将字典转换（返回）到命名空间：

In [2]: dd={'swd':'SomeString', 'int':35, 'f':True}
In [3]: dd
Out[3]: {'f': True, 'int': 35, 'swd': 'SomeString'}
In [4]: argparse.Namespace(**dd)
Out[4]: Namespace(f=True, int=35, swd='SomeString')

据我所知，没有一个软件包可以将argparse类似推理应用于字典（可能由json.loads创建）。

最近有人问过是否有办法根据sys.argv重新创建namespace字符串。你可能会看一下。

我可以使用此代码

重新创建sys.argv之类的列表

In [22]: def foo(k,v):
    if len(k)==1:
        k =  '-'+k
    else:
        k = '--'+k
    if isinstance(v,bool):
        return (k,)
    else:
        return k,str(v)
   ....:     
In [23]: argv=[]
In [24]: for k,v in dd.items():
    argv.extend(foo(k,v))
   ....:     
In [25]: argv
Out[25]: ['-f', '--int', '35', '--swd', 'SomeString']

然后可以通过以下方式将其传递给您的解析器：

args = parser.parse_args(argv)

========================

您可以做的另一件事是创建一个包含默认值的字典，并从JSON字典中更新它：

In [27]: defaultd=dict(int=20, strwdef='DEFAULTSTRING',snd=None,flag=False)
In [28]: defaultd
Out[28]: {'strwdef': 'DEFAULTSTRING', 'snd': None, 'int': 20, 'flag': False}
In [29]: defaultd.update(dd)
In [30]: defaultd
Out[30]: 
{'strwdef': 'DEFAULTSTRING',
 'f': True,
 'flag': False,
 'swd': 'SomeString',
 'snd': None,
 'int': 35}

我刚注意到一个问题。对于defaultd，我使用了长名称'flag'名称，但在JSON dict中我使用了短'f'名称。

我可以选择性地从JSON dict复制：

In [32]: for k in defaultd:
   ....:     if k in dd:
   ....:         defaultd[k]=dd[k]
   ....:         
In [33]: defaultd
Out[33]: {'strwdef': 'DEFAULTSTRING', 'snd': None, 'int': 35, 'flag': False}

处理“坏”值或别名会更复杂。

Answer 2

以下是您可以这样做的方法。我不确定它有多安全。 JSON键应该与解析它们所需的变量相同。

j = json.loads(...)
print j  # {'swd': 'someString', 'i': 35}
locals().update(j)
print swd, i  # someString 35

docs for locals建议不要修改，因此请自行承担风险。

我很快想出的一个更安全的解决方法是在argparsing之后将结果存储到字典中。从JSON导入它们时，只需单独更新该字典。

Python：如何从JSON文件中提取参数？

2 个答案: