python - 一起使用 Argparse 和 Json

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

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

argparse_dict = vars(args)

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

argparse_dict.update(json_dict)

这样 json_dict 值就会覆盖 argparse 值。

如果你想保留两者，你要么需要有不同的参数(键)名称，要么值必须是列表，你可以追加或扩展。这需要做更多的工作，首先是在 argparse 中使用正确的 nargs 值。

---

修改后的解析器产生了一个测试输入:

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 session 中来生成此内容，但我认为 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 值添加到“test2”的未定义值的一种方法是:

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 优先处理 JSON 字典中的 'cc' 值。