如何修复 Python YAML namedtuple 错误: yaml.representer.RepresenterError: ('cannot represent an object', ...

问题:

你正在尝试 yaml.safe_dump() 一个是(或包含)namedtuple 实例的对象,例如:

yaml_namedtuple_fix.py
import yaml
import collections

MyTuple = collections.namedtuple("MyTuple", ["a", "b", "c"])
mytuple = MyTuple(1,2,3)

yaml.safe_dump(mytuple)

这会导致以下异常:

yaml_namedtuple_error.txt
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/uli/.local/lib/python3.10/site-packages/yaml/__init__.py", line 269, in safe_dump
    return dump_all([data], stream, Dumper=SafeDumper, **kwds)
  File "/home/uli/.local/lib/python3.10/site-packages/yaml/__init__.py", line 241, in dump_all
    dumper.represent(data)
  File "/home/uli/.local/lib/python3.10/site-packages/yaml/representer.py", line 27, in represent
    node = self.represent_data(data)
  File "/home/uli/.local/lib/python3.10/site-packages/yaml/representer.py", line 58, in represent_data
    node = self.yaml_representers[None](self, data)
  File "/home/uli/.local/lib/python3.10/site-packages/yaml/representer.py", line 231, in represent_undefined
    raise RepresenterError("cannot represent an object", data)
yaml.representer.RepresenterError: ('cannot represent an object', MyTuple(a=1, b=2, c=3))

解决方案

你需要添加自定义表示器来隐式将 namedtuple 转换为 dict。

在运行 yaml.safe_dump() 之前,添加以下行:

represent_namedtuple_fix.py
import collections

def represent_namedtuple(dumper, data):
    return dumper.represent_dict(data._asdict())

yaml.SafeDumper.add_representer(MyTuple, represent_namedtuple)

需要为每个使用的 namedtuple add_representer()

现在,yaml.safe_dump() 调用应该完美工作:

yaml_safe_dump_fixed.py
yaml.safe_dump(mytuple) # Returns 'a: 1\nb: 2\nc: 3\n'

Check out similar posts by category: Python