使用 pyyaml/ruamel.yaml 有条件地转储属性

Question

我有一个类似这样的 Python 类：

from dataclasses import dataclass
from ruamel.yaml import yaml_object, YAML

yaml = YAML()

@yaml_object(yaml)
@dataclass
class DataObject:
    normal_attr: str
    normal_attr_2: str
    conditional_attr: str

然后我希望能够根据某些条件使用ruamel.yaml（基于 PyYaml）有条件地转储conditional_attr。理想情况下，它会像这样工作：

data = DataObject()

if verbose_output:
    yaml.dump(data, stream)
else:
    yaml.dump(data, stream, exclude=['conditional_attr'])

当然这实际上不起作用，但是有什么方法可以实现这种行为吗？

Answer 1

我认为排除某些映射键作为转储方法的一个选项并不是一个好主意。

除此之外，ruamel.yaml 已经有一个 documented method to do special dumps 可以用来过滤掉 任何属性，例如对于那些名字以conditional_开头的人：

import sys
from dataclasses import dataclass
from ruamel.yaml import yaml_object, YAML

yaml = YAML()

verbose_output = False

@yaml_object(yaml)
@dataclass
class DataObject:
    normal_attr: str
    normal_attr_2: str
    conditional_attr: str

    @classmethod
    def to_yaml(cls, representer, node):
        d = {}
        for k in node.__annotations__:
            if not verbose_output and k.startswith('conditional_attr'):
                continue
            d[k] = node.__getattribute__(k)
        return representer.represent_mapping('!DataObject', d)

data = DataObject(normal_attr='a', normal_attr_2='b', conditional_attr='c')

yaml.dump(data, sys.stdout)

这给出了：

!DataObject
normal_attr: a
normal_attr_2: b