如何在没有 yaml.YAMLObject 的情况下执行 Python 对象的 yaml.safe_dump() 和 .safe_load()？

Question

我想用yaml.safe_dump() 序列化一些对象。如何使用 add_representer() 和 add_constructor() 序列化 Python 对象...

我无法将 yaml.YAMLObject 添加到 Thing（第三方模块）并且不想使用。

我做这样的转储：

import yaml

class Thing(object):
  def __init__(self, name):
    self.name = name

def Thing_representer(dumper, data):
  return dumper.represent_mapping('!Thing', data.__dict__)

yaml.SafeDumper.add_representer(Thing, Thing_representer)
safe_dump = yaml.safe_dump(t)
print safe_dump

它工作正常，但我不知道如何做构造函数？

def Thing_constructor(loader, data):
  thing = Thing()
  return thing.__dict__.update(loader.construct_mapping(data))

yaml.SafeLoader.add_constructor('!Thing', Thing_constructor)
yaml.safe_load(safe_dump)

它抛出异常TypeError: __init__() takes exactly 2 arguments (1 given) 并且应该抛出异常，因为构造函数需要参数。也许还有另一种选择来构造对象跳过构造函数？

Answer 1

你不能在不提交名字的情况下构造Thing()。你可以解决 以各种方式，但以下应该有效。

def thing_constructor(self, node):
    name = None
    for x in node.value:
        if x[0].value == 'name':
            name = x[1].value
    return Thing(name)


yaml.SafeLoader.add_constructor('!Thing', thing_constructor)

res = yaml.safe_load(safe_dump)
print res.name

你可以简化name参数的设置，但是如果Thing需要更多的参数，这样扩展性会更好。