python - 在 json 文件中使用变量

Question

我需要定义一个变量并在整个 json 文件中使用它。

这是我的 MWE:(改编自 here)

{
  "variables": {
    "my_access_key": "abc",
    "my_secret_key": "def"
  },
  "objectB": {
    "type": "1",
    "access_key": "{{user `my_access_key`}}",
    "secret_key": "{{user `my_secret_key`}}"
  },
  "objectA": {
    "type": "2",
    "access_key": "{{user `my_access_key`}}",
    "secret_key": "{{user `my_secret_key`}}"
  }
}

对象objectB和objectA的access_key字段必须等于“abc”，即由我在文件开头定义。

我怎样才能在 python 中实现这个目标？

Answer 1

JSON 不允许变量引用

在 YAML 中，您可以定义变量，为它们设置引用名称，然后稍后在文件中重用它们。

JSON 不提供这种功能，您必须自己逐个设置这些值。

以编程方式构建 JSON

import json

access = "AAA"
secret = "XXX"

dct = {"variables": {"my_access_key": access, "my_secret_key": secret},
       "objectB": {"type": "1", "access_key": access, "secret_key": secret},
       "objectA": {"type": "2", "access_key": access, "secret_key": secret}
      }
json_str = json.dumps(dct, indent=True)
print json_str

打印什么

{
 "objectA": {
  "access_key": "AAA", 
  "secret_key": "XXX", 
  "type": "2"
 }, 
 "variables": {
  "my_secret_key": "XXX", 
  "my_access_key": "AAA"
 }, 
 "objectB": {
  "access_key": "AAA", 
  "secret_key": "XXX", 
  "type": "1"
 }
}

使用 YAML anchor 和引用

您可以为此目的使用 YAML 功能。由于 YAML 相当容易编辑，因此它可能是配置文件的不错选择。

在使用它之前，请确保安装 pyyaml:

$ pip install pyyaml

然后是代码(在 variables 中修改名称以满足我们的需要):

import json
import yaml
yaml_str = """
variables: &keys
    access_key: abc
    secret_key: def
objectB:
    <<: *keys
    type: "1"
objectA:
    <<: *keys
    type: "2"
"""

dct = yaml.load(yaml_str)
json_str = json.dumps(dct, indent=True)
print json_str

打印

{
 "objectA": {
  "access_key": "abc", 
  "secret_key": "def", 
  "type": "2"
 }, 
 "variables": {
  "access_key": "abc", 
  "secret_key": "def"
 }, 
 "objectB": {
  "access_key": "abc", 
  "secret_key": "def", 
  "type": "1"
 }
}