Question

我试图在我的Python脚本中使用YAML配置文件，我发现的一个问题是我无法访问其他属性，所以我需要复制很多内容。

例如

root_path: /root
script_path: root_path + /scripts

当然，这并不存在，但有没有办法实现这一目标？因为有很多内容我无法复制，因为当一个内容发生变化时，我需要在任何地方进行更改......

我也考虑过创建自己的连接功能

root_path: &ROOT /root
script_path: !join [*ROOT, '/scripts']


def join(loader, node):
    seq = loader.construct_sequence(node)
    return ''.join([str(i) for i in seq])

yaml.add_constructor('!join', join)

但我每次都需要设置&VARNAME，自动将参数键设置为可引用对象会很好...

Answer 1

好吧，我一直在研究解决方案，因为没有什么是我想要的，而且一切都很复杂，我想死...

此解决方案将%something%字符串转换为something的值。

它完美无缺，这是一个例子

root_path: /root
script_path: "%root%/scripts"

使用此方法，script_path将变为/root/scripts。

def replace_values(yaml_file):
    def _get(dict, list):
        return reduce(lambda d, k: d[k], list, dict)

    def _replace(obj):
        for k, v in obj.iteritems():
            if isinstance(v, dict):
                _replace(v)
            if isinstance(v, str):
                match = re.match(r'%(.*)%', v)
                if match:
                    reference = match.group(1).split('.')
                    replace = _get(yaml_file, reference)
                    obj[k] = re.sub(r'%.*%', replace, v)

    _replace(yaml_file)
    return yaml_file

使用很简单，只需按正常方式加载Yaml文件，然后调用replace。

with open(config_file.get(options.env), 'r') as ymlfile:
    config = yaml.load(ymlfile)

config = replace_values(config)

然后我们新的config会存储替换后的值，当然它不会覆盖原始的.yml文件。我希望你发现这很有用，我真的这样做，因为它正是我所需要的。

我使用了一个百分比%令牌，但您可以将其更改为您想要的任何内容，并更改方法以使其与正则表达式一起使用（正则表达式使用了一些令牌，这就是我使用{{{ 1}}）

Answer 2

你无法用YAML真正做到这一点。对于配置文件，您可以改为使用ConfigParser，它允许您插入值，以便您的示例如下所示：

root_path: /root
script_path: %(root_path)s/scripts

Answer 3

我只想稍微调整一下YAML解析器¹：

import ruamel.yaml as yaml
from ruamel.yaml.comments import CommentedMap

yaml_str = """\
root_path: /root   # use this key for expansion of values
script_path: root_path + /scripts
"""

def set_item(self, key, value):
    split_val = value.split(' + ', 1)
    if len(split_val) > 1:
        alt_key = split_val[0].strip()
        if alt_key in self.keys():
            value = self.get(alt_key) + split_val[1]
    self._org__setitem__(key, value)

CommentedMap._org__setitem__ = CommentedMap.__setitem__
CommentedMap.__setitem__ = set_item

data = yaml.load(yaml_str, Loader=yaml.RoundTripLoader)
print yaml.dump(data, Dumper=yaml.RoundTripDumper)

会给你：

root_path: /root   # use this key for expansion of values
script_path: /root/scripts

请注意，使用RoundTripLoader / Dumper组合可以保留注释和键的顺序。 data用作普通的Python dict。

¹_{这是使用ruamel.yaml完成的，我是作者。它是PyYAML的超集，主要用于保存数据往返的评论和其他信息。}

Python YAML参数引用（从其他参数获取值）

3 个答案: