我有一个带有如下别名的 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
如何使其与序列别名一起使用?
最佳答案
在 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
的参数列表(它采用一个参数),内部序列是该参数。
关于python - 如何使自定义 YAML 标签与 pyyaml 中的序列别名一起使用,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53776602/