Класс синтаксического анализа Python из YAML

Вместо того

nodeMap = loader.construct_mapping(node)

попробуй это:

nodeMap = loader.construct_mapping(node, deep=True)

Кроме того, у вас есть небольшая ошибка в вашем YAML-файле:

!MyClass:

Двоеточие в конце здесь не место.

С вашим подходом возникает множество проблем, даже если не учитывать, что вы должны прочитать PEP 8, руководство по стилю кода Python, в частности часть на Имена методов и переменные экземпляра

1. Поскольку вы указываете, что долго просматривали документацию Python, вы не могли не заметить, что yaml.load() небезопасно. Его также почти никогда не нужно использовать, особенно если вы пишете свои собственные репрезентаторы и конструкторы.

2. Вы используете dumper.represent_dict({data.YAMLTag : data.toDict()}), который выгружает объект как пару "ключ-значение". Что вы хотите сделать, по крайней мере, если вы хотите, чтобы в выходном YAML был тег: dumper.represent_mapping(data.YAMLTag, data.toDict()). Это даст вам вывод формы:

   !MyClass
   name: InstanceId
   times: [0.0, 0.25, 0.5, 1.0, 2.0, 5.0, 10.0]
   zeros: [0.03, 0.03, 0.04, 0.03, 0.03, 0.02, 0.03]
   

   то есть сопоставление с тегами вместо пары "ключ-значение", где значением является сопоставление. (И я ожидал, что первая строка будет '!MyClass':, чтобы убедиться, что скаляр, начинающийся с восклицательного знака, не интерпретируется как тег).

3. Создание сложного объекта, потенциально самореферентного (прямо или косвенно), необходимо выполнить за два шага с использованием генератора. (код PyYAML называет это правильно для вас). В вашем коде вы предполагаете, что у вас есть все параметры для создания экземпляра MyClass. Но если есть ссылка на себя, эти параметры должны включать сам этот экземпляр, и он еще не создан. Правильный пример кода в базе кода YAML для этого - construct_yaml_object() в constructor.py:

   def construct_yaml_object(self, node, cls):
       data = cls.__new__(cls)
       yield data
       if hasattr(data, '__setstate__'):
           state = self.construct_mapping(node, deep=True)
           data.__setstate__(state)
       else:
           state = self.construct_mapping(node)
           data.__dict__.update(state)
   

   Вам не обязательно использовать .__new__(), но вы должны принять во внимание deep=True, как объяснено здесь

В общем, также полезно иметь __repr__(), который позволяет вам проверять загружаемый объект с помощью чего-то более выразительного, чем <__main__.MyClass object at 0x12345>

Импорт:

from __future__ import print_function

import sys
import yaml
from cStringIO import StringIO
import numpy as np

Чтобы проверить правильность работы самореферентных версий, я добавил в класс атрибут self._ref:

class MyClass(object):
    YAMLTag = u'!MyClass'

    def __init__(self, name=None, times=[], zeros=[], ref=None):
        self.update(name, times, zeros, ref)

    def update(self, name, times, zeros, ref):
        self.name = name
        self._T = np.array(times)
        self._zeros = np.array(zeros)
        self._ref = ref

    def toDict(self):
        return dict(name=self.name,
                    times=self._T.tolist(),
                    zeros=self._zeros.tolist(),
                    ref=self._ref,
        )

    def __repr__(self):
        return "{}(name={}, times={}, zeros={})".format(
            self.__class__.__name__,
            self.name,
            self._T.tolist(),
            self._zeros.tolist(),
        )

    def update_self_ref(self, ref):
        self._ref = ref

"Методы" представителя и конструктора:

    @staticmethod
    def to_yaml(dumper, data):
        return dumper.represent_mapping(data.YAMLTag, data.toDict())

    @staticmethod
    def from_yaml(loader, node):
        value = MyClass()
        yield value
        node_map = loader.construct_mapping(node, deep=True)
        value.update(**node_map)


yaml.add_representer(MyClass, MyClass.to_yaml, Dumper=yaml.SafeDumper)
yaml.add_constructor(MyClass.YAMLTag, MyClass.from_yaml, Loader=yaml.SafeLoader)

И как им пользоваться:

instance = MyClass('InstanceId',
                   [0.0, 0.25, 0.5, 1.0, 2.0, 5.0, 10.0],
                   [0.03, 0.03, 0.04, 0.03, 0.03, 0.02, 0.03])
instance.update_self_ref(instance)

buf = StringIO()
yaml.safe_dump(instance, buf)

yaml_str = buf.getvalue()
print(yaml_str)


data = yaml.safe_load(yaml_str)
print(data)
print(id(data), id(data._ref))

комбинация выше дает:

&id001 !MyClass
name: InstanceId
ref: *id001
times: [0.0, 0.25, 0.5, 1.0, 2.0, 5.0, 10.0]
zeros: [0.03, 0.03, 0.04, 0.03, 0.03, 0.02, 0.03]

MyClass(name=InstanceId, times=[0.0, 0.25, 0.5, 1.0, 2.0, 5.0, 10.0], zeros=[0.03, 0.03, 0.04, 0.03, 0.03, 0.02, 0.03]) 
139737236881744 139737236881744

Как видите, значения id для data и data._ref после загрузки одинаковы.

Вышеупомянутое вызывает ошибку, если вы используете упрощенный подход в своем конструкторе, просто используя loader.construct_mapping(node, deep=True)