Question

我是Python的初学者。

我想知道Argparse和JSON是否可以一起使用。说，我有变量p,q,r

我可以将它们添加到argparse中作为 -

parser.add_argument('-p','--param1',help='x variable', required=True)
parser.add_argument('-q','--param2',help='y variable', required=True)
parser.add_argument('-r','--param3',help='z variable', required=True)

现在假设我想从JSON文件中读取相同的变量，是否可以这样做？所以我可以从命令行或JSON文件输入值。

JSON输入文件 -

{
    "testOwner": "my name",
    "tests": [
        "test1",
        "test2",
        "test3"
    ],

    "testParameters": {
        "test1": {
            "param1": "0",
            "param2": "20",
            "param3" : "True"
        },

        "test2": {
            "param1": "cc"
        }
    }   
}

Answer 1

args中的parse_args命名空间可以转换为字典：

argparse_dict = vars(args)

JSON值也在字典中，比如json_dict。您可以将选定的值从一个字典复制到另一个字典，或者进行整体更新：

argparse_dict.update(json_dict)

这样json_dict值就会写出argparse值。

如果要保留两者，则需要具有不同的参数（键）名称，或者值必须是列表，您可以追加或扩展。这需要更多工作，首先使用nargs中的正确argparse值。

修订后的parser会产生测试输入：

In [292]: args=parser.parse_args('-p one -q two -r three'.split())
In [293]: args
Out[293]: Namespace(param1='one', param2='two', param3='three')
In [295]: args_dict = vars(args)    
In [296]: args_dict
Out[296]: {'param1': 'one', 'param2': 'two', 'param3': 'three'}

JSON字符串在解析时（json.loads？）会生成如下字典：

In [317]: json_dict
Out[317]: 
{'testOwner': 'my name',
 'testParameters': {'test1': {'param1': '0', 'param2': '20', 'param3': 'True'},
  'test2': {'param1': 'cc'}},
 'tests': ['test1', 'test2', 'test3']}

我通过将您的显示粘贴到我的Ipython会话中来制作它，但我认为JSON加载器产生相同的东西

argparse值可以添加：

In [318]: json_dict['testParameters']['test3']=args_dict
In [319]: json_dict
Out[319]: 
{'testOwner': 'my name',
 'testParameters': {'test1': {'param1': '0', 'param2': '20', 'param3': 'True'},
  'test2': {'param1': 'cc'},
  'test3': {'param1': 'one', 'param2': 'two', 'param3': 'three'}},
 'tests': ['test1', 'test2', 'test3']}

在这里，我将其添加为第3个test集，从tests列表中获取（通过巧合）名称。 json_dict['testParameters']['test2']=args_dict会替换test2。

的值

将args值添加到＆＃39; test2＆＃39;的未定义值的一种方法是：

In [320]: args_dict1=args_dict.copy()    
In [322]: args_dict1.update(json_dict['testParameters']['test2'])
In [324]: json_dict['testParameters']['test2']=args_dict1
In [325]: json_dict
Out[325]: 
{'testOwner': 'my name',
 'testParameters': {'test1': {'param1': '0', 'param2': '20', 'param3': 'True'},
  'test2': {'param1': 'cc', 'param2': 'two', 'param3': 'three'},
  'test3': {'param1': 'one', 'param2': 'two', 'param3': 'three'}},
 'tests': ['test1', 'test2', 'test3']}

我使用此版本的update优先考虑＆＃39; cc＆＃39; JSON字典中的值。

Answer 2

发现以下警告很容易

安装程序会使用命令行上的值覆盖配置文件中的值
如果未在命令行或设置文件中未设置选项，则仅使用默认值
它不会检查配置文件中的设置是否有效

import argparse
import json

parser = argparse.ArgumentParser()

parser.add_argument('--save_json',
    help='Save settings to file in json format. Ignored in json file')
parser.add_argument('--load_json',
    help='Load settings from file in json format. Command line options override values in file.')

args = parser.parse_args()

if args.load_json:
    with open(args.load_json, 'rt') as f:
        t_args = argparse.Namespace()
        t_args.__dict__.update(json.load(f))
        args = parser.parse_args(namespace=t_args)

# Optional: support for saving settings into a json file
if args.save_json:
    with open(args.save_json, 'wt') as f:
        json.dump(vars(args), f, indent=4)

Answer 3

鉴于您的JSON文件包含以下形式的字典：

d = {"name": ["-x", "--xvar"], "help": "Help message", "required": True}

创建解析器之后，您可以“解压缩”dict，如下所示：

parser = argparse.ArgumentParser()
parser.add_argument(*(d.pop("name")), **d) 
# Put the 'name' as name/flag and then unpack the rest of
# the dict as the rest of the arguments
parser.parse_args("--xvar 12".split())
>>> Namespace(xvar='12')

但是，这会强制您维护dict键以适合方法add_arguments的参数名称。您也没有简单/直接的方式来使用更高级的行为，例如使用action，type，choices参数。

此外，您还必须更改dict的形式以包含要使用的各种参数。一种解决方案是将名称/标志作为元组中dict的键，参数将是一个字典：

d = {("-x", "--xvar"): {"help": "Help message for x", "required": True}, 
     ("-y", "--yvar"): {"help": "Help message for y", "required": True}}
for names, args in d.iteritems():
    parser.add_argument(*names, **args) # Use a similar unpacking 'magic' as the first example
parser.parse_args("-x 12 --yvar 42".split())
>>> Namespace(xvar='12', yvar='42')

修改鉴于来自OP的评论，看起来他想要解析从JSON文件中获取的值。

d = {"-x": "12", "-y": "42"} args = [] for item in d.items(): args.extend(item) parser.parse_args(args) >>> Namespace(xvar='12', yvar='42')

编辑2

查看argparse文档this paragraph可能有点相关。

一起使用Argparse和Json

3 个答案: