YAML 以键/值格式构建为 JSON

Question

我在下面有这个 yaml 文件。我需要保留 yaml 文件的这种结构。 在文件中有一些嵌套键/值的特殊性（如personalinfo and more）和一些没有值的嵌套键（如extra）。
键/值的数量是任意的，因为父母的数量也是任意的（在这种情况下，父母是personalInfo, more, extra）。

name: person1
personalInfo:
  - key: address
    value: street 1
  - key: age
    value: 10
  - key: school
    value: XYZ
more:
  - key: mother
    value: Samantha
extra:
 - key: a
 - key: b

我想从这个 yaml 生成一个格式如下的 Json，但我不知道如何实现。

'{"personalInfo" : {"address": "street 1", "age": "10", "school": "XYZ"}, "more":{"mother": "Samantha"}, "extra": ["a", "b"]}' "localhost:8080/person1"

Answer 1

最简单最纯粹的方法是 PyYaml，可以通过 pip install pyyaml 安装。存在一个普通的 yaml.load() 函数，但始终应首选 yaml.safe_load() ，除非您明确需要提供任意对象序列化/反序列化以避免引入任意代码执行的可能性。

import yaml
with open("test.yml", 'r') as stream:
    try:
        data = yaml.safe_load(stream)
    except yaml.YAMLError as exc:
        print(exc)

这会返回：

{'name': 'person1',
 'personalInfo': [{'key': 'address', 'value': 'street 1'},
  {'key': 'age', 'value': 10},
  {'key': 'school', 'value': 'XYZ'}],
 'more': [{'key': 'mother', 'value': 'Samantha'}],
 'extra': [{'key': 'a'}, {'key': 'b'}]}

你可以试试这个来得到你想要的结果：

import json

new_data = {}

for i in data:
    if i!='name':    
        temp = {}
        val = []
        for k in data[i]:
            if 'key' in k:
                try:
                    temp[k['key']] = k['value']
                except:
                    val.append(k['key'])
        if val:
            new_data[i] = val
        elif temp:
            new_data[i] = temp
        else:
            new_data[i] = data[i]

str1 = json.dumps(new_data)
str2 = "localhost:8080/"+data['name']

with open("sample.json", "w") as outfile:
    json.dump(str1.replace('"', "'"), outfile)
    json.dump(str2.replace('"', "'"), outfile)

sample.json 中的结果

"{'personalInfo': {'address': 'street 1', 'age': 10, 'school': 'XYZ'}, 'more': {'mother': 'Samantha'}, 'extra': ['a', 'b']}""localhost:8080/person1"