Python 如何使自定义YAML标记与pyyaml中的序列别名一起工作

我有一个yaml文件，别名如下：

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

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

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

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

！rand

是一个自定义标记，它在yaml序列上调用python的

random.choice

我正在使用

的以下实现！兰德

标签：

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

和别名

*用户

停止工作：

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

File../python3.6/site packages/yaml/parser.py”，第439行，在parse_block_mapping_key中
应为，但找到%r“%token.id，token.start\u标记）
yaml.parser.ParserError:解析块映射时
在“/temp/config.yml”中，第5行第3列
应为，但找到“”
在“/temp/config.yml”中，第6行第18列

---

如何使用序列别名？

在YAML中，标记用于定义内容类型，而不是用于处理内容

因此，您不能标记别名，因为别名只是对已在其定义位置标记的内容的引用。在您的情况下，您的序列将根据。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']

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

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

---

将其视为函数

！rand

（取一个参数）的参数列表的外部序列和该参数的内部序列。

您还可以使列表可调用，并在每次需要随机用户时调用它：

YAML = """
user: &users !rand 
    - user1
    - user2
    - user3
    - user4
"""

import random
import yaml
from collections.abc import Sequence

class RandomizableList(Sequence):
    def __init__(self, items):
        self.items = items
    def __len__(self):
        return len(self.items)
    def __getitem__(self, value):
        return self.items[value]
    def __call__(self):
        return random.choice(self.items)
    def __repr__(self):
        return repr(self.items)

def rand_constructor(loader, node):
    return RandomizableList(loader.construct_sequence(node))

yaml.add_constructor('!rand', rand_constructor)
result = yaml.load(YAML)
for i in range(4):
    print(result['user']())
print(result['user'])

---

根据您建议的更改，当我将其更改为

user2:！rand[*users]

时，生成的

值

被设置为空列表

[[]

，回溯显示：“indexer-ror:无法从空序列中选择“对不起，我没有测试这个。可能需要一些调整；也可能是别名解析太晚，无法正常工作。我只是假设这是基于我知道的其他YAML加载库实现的。