python - ruamel.yaml:如何在顶层和其他元素内添加新元素

Question

ruamel.yaml 文档很少，我在这里没有找到我需要的响应。

my_file.yaml 的内容:

desc: "blahblahblah"

问题1:如何(以编程方式)将“size”添加为与“desc”同一级别的元素(即在树的顶层)

问题2:如何(以编程方式)添加子元素深度n​​倍

生成的文件应如下所示:

desc: 
    - "blahblahblah"
    - "desc_trans":
        - "Chinese": 
          - "Mandarin": "blahblahblah"
          - "Uyghur": "blahblahblah"
        - "Spanish": "blahblahblah"
size: "40k"

Answer 1

ruamel.yaml 文档很少，但您想做的主要是在Python级别。您不更改 YAML，而是从文件加载 YAML并将其解析为数据结构，更改该数据结构(使用Python，因为 ruamel.yaml 是一个 Python 包)，然后转储数据结构返回 YAML。

首先:你无法得到你想要的东西，因为你的缩进的序列不一致。 desc 的值的缩进为 6，以及 block 序列指示符 (-) 内的偏移量四岁。 "desc_trans" 的值缩进为 4，并带有偏移量为 2，"Chinese" 的最小缩进量为二。 ruamel.yaml 只有一对序列缩进和偏移应用于所有( block )序列的值。

您正在尝试实现一些棘手的事情:

• 加载键 desc 的标量值并神奇地进行翻译将其放入序列中(旧标量值是第一个元素)。当然，代码应该仅在以下情况下进行此更改:必要的。

• 将普通标量与带有多余标量的标量相结合双引号。如果我没看错的话，任何不是的标量字符串根级映射的键应该用双引号引起来。你可以实现以编程方式实现，但在示例中我将手动实现。

假设Python3:

from pathlib import Path
import sys
import ruamel.yaml
from ruamel.yaml.scalarstring import DoubleQuotedScalarString as dq

yaml = ruamel.yaml.YAML()
yaml.indent(mapping=2, sequence=4, offset=2)
yaml.preserve_quotes = True
data = yaml.load(Path('my_file.yaml'))

assert isinstance(data, dict)  # check that the root-level is a dictionary like object 
data['size'] = dq('40k')

if not isinstance(data['desc'], list):
    data['desc'] = [data['desc']]  # change the non-list into a list
l = data['desc']
# make the desc_trans dict first, cannot use dict(desc_trans=[dict(Chinese=[Mandarin=....
desc_trans = []
l.append({dq('desc_trans'): desc_trans})
Chinese = {dq('Chinese'): [{dq('Mandarin'): dq('blahblahblah')}]}
Spanish = {dq('Spanish'): dq('blahblahblah')}
desc_trans.append(Chinese)
desc_trans.append(Spanish)
# the next line also could be simplified using `desc_trans.append()`
data['desc'][1]['desc_trans'][0]['Chinese'].append({dq('Uygur'): dq('blahblahblah')})

# if you want to write to a file use: out = Path('output.yaml')
out = sys.stdout
yaml.dump(data, out)

给出:

desc:
  - "blahblahblah"
  - "desc_trans":
      - "Chinese":
          - "Mandarin": "blahblahblah"
          - "Uygur": "blahblahblah"
      - "Spanish": "blahblahblah"
size: "40k"

一般来说，我建议将深层元素分配给变量，然后向其附加或分配新键。

由于 ruamel.yaml 实际上加载有序字典子类中的映射，没有必要拥有这些单键值序列对映射，您将其作为 "dest_trans" 和“中文”。省略列表可以简化代码，并且子类具有一个 .insert() 方法来对按键顺序进行必要的控制。