是否可以在跨文档的多文档YAML流中使用别名？

Question

这就是我想要做的（代码在Python 3）：

import ruamel.yaml as yaml 
from print import pprint 

yaml_document_with_aliases = """ 
title: test 
choices: &C 
    a: one 
    b: two 
    c: three 
--- 
title: test 2 
choices: *C 
""" 

items = list(yaml.load_all(yaml_document_with_aliases)) 

结果是：

ComposerError: found undefined alias 'C' 

当我使用非基于文件YAML文件，此按预期工作：

import ruamel.yaml as yaml 
from print import pprint 

yaml_nodes_with_aliases = """ 
- 
    title: test 
    choices: &C 
    a: one 
    b: two 
    c: three 
- 
    title: test 2 
    choices: *C 
""" 

items = yaml.load(yaml_nodes_with_aliases) 

pprint(items) 

结果：

[{'choices': {'a': 'one', 'b': 'two', 'c': 'three'}, 'title': 'test'}, 
{'choices': {'a': 'one', 'b': 'two', 'c': 'three'}, 'title': 'test 2'}] 

（我想反正来完成）

---

因为它现在是不可能的，我用以下的脆性解决方法：

def yaml_load_all_with_aliases(yaml_text): 
    if not yaml_text.startswith('---'): 
     yaml_text = '---\n' + yaml_text 
    for pat, repl in [('^', ' '), ('^\s*---\s*$', '-'), ('^\s+\.{3}$\n', '')]: 
     yaml_text = re.sub(pat, repl, yaml_text, flags=re.MULTILINE) 
    yaml_text = yaml_text.strip() 
    return yaml.safe_load(yaml_text) 

Answer 1

这里的问题是：

title: test 
choices: &C 
    a: one 
    b: two 
    c: three 
--- 
title: test 2 
choices: *C 

不是a文件，这些是两个YAML文件中的一个文件。锚定义&C不从一个YAML文档携带到另一个，它只能用到文档分隔符---。

如果你愿意把所有锚“结转”下列文件中的一个YAML流，你可以在Composer类（即猴子补丁吧）移植了新的compose_document方法：

import sys 
import ruamel.yaml 

yaml_str = """\ 
title: test 
choices: &C 
    a: one 
    b: two 
    c: three 
--- 
title: test 2 
choices: *C 
""" 


def my_compose_document(self): 
    self.get_event() 
    node = self.compose_node(None, None) 
    self.get_event() 
    # this prevents cleaning of anchors between documents in **one stream** 
    # self.anchors = {} 
    return node 

ruamel.yaml.composer.Composer.compose_document = my_compose_document 

datas = [] 
for data in ruamel.yaml.safe_load_all(yaml_str): 
    datas.append(data) 

datas[0]['choices']['a'] = 1 
for data in datas: 
    ruamel.yaml.round_trip_dump(data, sys.stdout, explicit_start=True) 

这给：

--- 
title: test 
choices: 
    a: 1 
    b: two 
    c: three 
--- 
title: test 2 
choices: 
    a: one 
    b: two 
    c: three 

注意，这让你有钥匙a，b，并c的字典的副本。

（如果该键排序和评论保存是重要的，使用的round_trip_load_all代替safe_load_all）