pyYAML，预期 NodeEvent，但得到 DocumentEndEvent

Question

我正在尝试转储一个自定义对象，它是一种对象列表。所以我覆盖了YAMLOBject 类的to_yaml 方法，我将我的类设置为继承自：

@classmethod
def to_yaml(cls, dumper, data):
    """ This methods defines how to save this class to a yml
    file """

    passage_list = []

    for passage in data:
        passage_dict = {
            'satellite': passage.satellite.name,
            'ground_station': passage.ground_station.name,
            'aos': passage.aos,
            'los': passage.los,
            'tca': passage.tca,

        }
        passage_list.append(passage_dict)

    passage_list_dict = {
        'passages': passage_list
    }

    return dumper.represent(passage_list_dict)

当我调用yaml.dump 方法时，使用正确的数据正确创建了输出文件：

if save_to_file:
    with open(save_to_file, 'w') as f:
        yaml.dump(all_passages, f, default_flow_style=False)

但在执行结束时我得到一个EmitterError: expected NodeEvent, but got DocumentEndEvent()

我认为这与未正确关闭 YAML 文档有关，因为当我调试代码时，我得到了 save_to_file 文件，这些文件在文档末尾缺少新行。可以吗？或者是别的什么？

Answer 1

您的代码不起作用，因为dumper.represent 没有返回任何内容。您想改用dumper.represent_data。