python - 如何使自定义 YAML 标签与 pyyaml 中的序列别名一起使用

Question

我有一个带有如下别名的 yaml 文件:

vars:
  users: &users ['user1', 'user2', 'user3', 'user4']

refs:
  users: *users
  user: !rand ['user1', 'user2', 'user3', 'user4']

!rand 是一个自定义标签，它在 yaml 序列上调用 python 的 random.choice。

我正在使用 !rand 标记的以下实现:

import random
import yaml

def load_yaml(file):
    def rand_constructor(loader, node):
        value = loader.construct_sequence(node)
        return random.choice(value)

    yaml.add_constructor('!rand', rand_constructor)

    with open(file) as f:
        return yaml.load(f)

它按预期工作，并且 user 从 users 获取随机值。现在，当我将 !rand 与别名 *users 一起使用时，它会停止工作:

vars:
  users: &users ['user1', 'user2', 'user3', 'user4']

refs:
  users: *users
  user: !rand *users

错误是:

File ../python3.6/site-packages/yaml/parser.py", line 439, in parse_block_mapping_key
    "expected <block end>, but found %r" % token.id, token.start_mark)
yaml.parser.ParserError: while parsing a block mapping
  in "/temp/config.yml", line 5, column 3
expected <block end>, but found '<alias>'
  in "/temp/config.yml", line 6, column 18

如何使其与序列别名一起使用？

Answer 1

在 YAML 中，标签旨在定义内容类型。它们不是设计来处理内容的。

因此，您无法标记别名，因为别名只是对已在其定义位置标记的内容的引用。在您的情况下，您的序列将根据 YAML core schema 获得 !!seq 标签。 。 YAML 不提供重新标记的功能。

话虽如此，您当然可以采取解决方法:将 !rand 参数包装在序列中:

vars:
  users: &users ['user1', 'user2', 'user3', 'user4']

refs:
  users: *users
  user: !rand [*users]

然后您只需将代码更改为:

import random
import yaml

def load_yaml(file):
    def rand_constructor(loader, node):
        value = loader.construct_sequence(node)
        return random.choice(value[0])

    yaml.add_constructor('!rand', rand_constructor)

    with open(file) as f:
        return yaml.load(f)

但请注意，直接调用 !rand 看起来像这样:

vars:
  users: &users ['user1', 'user2', 'user3', 'user4']

refs:
  users: *users
  user: !rand [['user1', 'user2', 'user3', 'user4']]

将其视为外部序列是函数 !rand 的参数列表(它采用一个参数)，内部序列是该参数。