如何从Python中删除yaml转储中的键/值对？

Question

假设我有一个天真的类定义：

import yaml
class A:
    def __init__(self):
        self.abc = 1
        self.hidden = 100
        self.xyz = 2

    def __repr__(self):
        return yaml.dump(self)

A()

印花

!!python/object:__main__.A
abc: 1
hidden: 100
xyz: 2

是否有一种干净的方法从yaml dump的打印输出中删除包含hidden: 100的行？ hidden的密钥名称是预先知道的，但其数值可能会更改。

期望的输出：

!!python/object:__main__.A
abc: 1
xyz: 2

仅供参考：此转储仅供显示，不会加载。

我想可以使用yaml.representative来使用key = hidden来抑制键/值对。 另一种方法是查找hidden: [number]在字符串输出中使用RegEx。

Answer 1

我查看了pyyaml的文档，但没有找到实现目标的方法。 解决方法是删除hidden的attribte，调用yaml.dump ，然后将其添加回：

    def __repr__(self):
        hidden = self.hidden
        del self.hidden

        return yaml.dump(self)

        self.hidden = hidden

退后一步，你为什么要使用yaml作为__repr__ ？ 你可以自己滚动而不是依靠yaml吗？

Answer 2

json是成熟的解决方案，并且（在写作的那一刻）有比pyyaml更好的文档;
我会使用它，而pyyaml的文档很难完全理解。 作为奖励，YAML（几乎）是JSON的超集，因此您将能够在不转换数据的情况下将您的数据读取为YAML。
^{但是，要轻松使用YAML的所有好东西，您可能必须将数据转换为YAML}

默认情况下， json模块无法序列化自定义对象，但可以轻松扩展：

import json

def default(o):
    if isinstance(o, A):
        result = vars(o).copy()
        del result['hidden']
        result['__class__'] = o.__class__.__name__
        return result
    else:
        return o

json.dumps(A(), default=default) # => '{"__class__": "A", "xyz": 2, "abc": 1}'

如果您不想在dumps任何位置写入default=default ，则可以创建自定义序列化程序：

dumper = json.JSONEncoder(default=default)
dumper.encode(A()) # => '{"__class__": "A", "xyz": 2, "abc": 1}'

或者，通过子类化可以轻松地进一步扩展它：

class Dumper(json.JSONEncoder):
    __slots__ = ()
    def default(self, o):
        if isinstance(o, A):
            result = vars(o).copy()
            del result['hidden']
            result['__class__'] = o.__class__.__name__
            return result
        else:
            return super().default(o)

dumper = Dumper()
dumper.encode(A()) # => '{"__class__": "A", "xyz": 2, "abc": 1}'

请注意，JSON中的字段是无序的。
另外，如果你想使用它，我建议你不要用密钥__class__序列化dict，因为它可能很难将它与序列化对象区分开来。

看到它在线工作