如何使用显式引用转储 YAML？

Question

递归引用在ruamel.yaml 或pyyaml 中效果很好：

$ ruamel.yaml.dump(ruamel.yaml.load('&A [ *A ]'))
'&id001
- *id001'

但是它（显然）不适用于普通引用：

$ ruamel.yaml.dump(ruamel.yaml.load("foo: &foo { a: 42 }\nbar: { <<: *foo }"))
bar: {a: 42}
foo: {a: 42}

我想显式创建一个引用：

data = {}
data['foo'] = {'foo': {'a': 42}}
data['bar'] = { '<<': data['foo'], 'b': 43 }

$ ruamel.yaml.dump(data, magic=True)
foo: &foo
    a: 42
bar: 
    <<: *foo
    b: 43

这对于生成具有大量公共键的大型数据结构的 YAML 输出非常有用

在输出上没有有争议的 re.replace 怎么可能？

其实ruamel.yaml.dump(data)的结果是

bar:
  '<<': &id001
    foo:
      a: 42
  b: 43
foo: *id001

所以我需要将'<<' 替换为<<，并可能将id001 替换为foo。

Answer 1

如果你想创建类似的东西，至少在 ruamel.yaml 中，你应该使用往返模式，它也保留了合并。以下不会引发断言错误：

import ruamel.yaml

yaml_str = """\
foo: &xyz
  a: 42
bar:
  <<: *xyz
"""

data = ruamel.yaml.round_trip_load(yaml_str)
assert ruamel.yaml.round_trip_dump(data) == yaml_str

这意味着data 有足够的信息来重新创建输出中的合并。然而，在实践中，在往返模式下，永远不会发生合并。取而代之的是检索值data['foo']['bar']['a'] 意味着data['foo'] 中没有真正的键'bar'，但随后会在附加的“合并映射”中查找该键。

对此没有公共接口（因此情况可能会改变），但通过分析data 并查看ruamel.yaml.comments.CommentedMap()，您会发现有一个merge_attrib（当前为字符串_yaml_merge）等等有用的是有一个方法add_yaml_merge()。后者采用 (int, CommentedMap()) 元组的列表。

baz = ruamel.yaml.comments.CommentedMap()
baz['b'] = 196
baz.yaml_set_anchor('klm')
data.insert(1, 'baz', baz)

你需要在数据的'bar'键之前插入'baz'键，否则映射会反转。在 data['bar'] 的合并中插入新结构后：

data['bar'].add_yaml_merge([(0, baz)])
ruamel.yaml.round_trip_dump(data, sys.stdout)

给出：

foo: &xyz
  a: 42
baz: &klm
  b: 196
bar:
  <<: [*xyz, *klm]

(如果你想看看add_yaml_merge 插入了什么

print(getattr(data['bar'], ruamel.yaml.comments.merge_attrib))

通话前后）

如果你想完全从头开始，你可以这样做：

data = ruamel.yaml.comments.CommentedMap([
    ('foo', ruamel.yaml.comments.CommentedMap([('a', 42)])),
    ])
data['foo'].yaml_set_anchor('xyz')
data['bar'] = bar = ruamel.yaml.comments.CommentedMap()
bar.add_yaml_merge([(0, data['foo'])])

而不是data = ruamel.yaml.round_trip_load(yaml_str)。

---

¹ _{免责声明：我是该软件包的作者。}