Невозможно записать пользовательские словари в YAML

У меня есть рабочий код Python, который использует три вложенных класса:

@dataclass
class PanelGroup:
    num_panels: int
    panel_efficiency: float
    panel_surface: float
    elevation: int


class PanelString(dict[str, PanelGroup]):
    pass


class SolarInstallation(dict[str, PanelString]):
    pass

Я создаю экземпляры этих классов статически из своего кода:

solar_installation = SolarInstallation(
        SSW_12=PanelString(G1=PanelGroup(4, 0.191, 1.705, 50),
                           G2=PanelGroup(5, 0.191, 1.705, 50)))

Я хочу, чтобы эта конфигурация исходила из файла YAML, поэтому я пытаюсь заставить работать следующий фрагмент кода, чтобы вывести правильное содержимое YAML в файл:

yaml = ruamel.yaml.YAML()
yaml.register_class(SolarInstallation)
yaml.register_class(PanelString)
yaml.register_class(PanelGroup)
yaml.dump(solar_installation, sys.stdout)

Но единственное, что я получаю, это пустой словарь:

!SolarInstallation {}

Что я делаю не так?

Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения текстовых сообщений может быть настолько сложным или простым, насколько вы его сделаете. Как и в любом ML-проекте, вы можете выбрать...
7 лайфхаков для начинающих Python-программистов
7 лайфхаков для начинающих Python-программистов
В этой статье мы расскажем о хитростях и советах по Python, которые должны быть известны разработчику Python.
Установка Apache Cassandra на Mac OS
Установка Apache Cassandra на Mac OS
Это краткое руководство по установке Apache Cassandra.
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
В одном из недавних постов я рассказал о том, как я использую навыки количественных исследований, которые я совершенствую в рамках программы TPQ...
Создание персонального файлового хранилища
Создание персонального файлового хранилища
Вы когда-нибудь хотели поделиться с кем-то файлом, но он содержал конфиденциальную информацию? Многие думают, что электронная почта безопасна, но это...
Создание приборной панели для анализа данных на GCP - часть I
Создание приборной панели для анализа данных на GCP - часть I
Недавно я столкнулся с интересной бизнес-задачей - визуализацией сбоев в цепочке поставок лекарств, которую могут просматривать врачи и...
1
0
33
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Вам нужно будет рассказать ruamel.yaml как сбросить ваши классы, просто регистрация не довольно:

import sys
from dataclasses import dataclass
import ruamel.yaml


@dataclass
class PanelGroup:
    num_panels: int
    panel_efficiency: float
    panel_surface: float
    elevation: int

    @classmethod
    def to_yaml(cls, representer, node):
        d = {}
        for k in node.__annotations__:
            d[k] = node.__getattribute__(k)
        return representer.represent_mapping('!' + cls.__name__, d)

class PanelString(dict[str, PanelGroup]):
    @classmethod
    def to_yaml(cls, representer, node):
        return representer.represent_mapping('!' + cls.__name__, node)


class SolarInstallation(dict[str, PanelString]):
    @classmethod
    def to_yaml(cls, representer, node):
        return representer.represent_mapping('!' + cls.__name__, node)

solar_installation = SolarInstallation(
        SSW_12=PanelString(G1=PanelGroup(4, 0.191, 1.705, 50),
                           G2=PanelGroup(5, 0.191, 1.705, 50)))
yaml = ruamel.yaml.YAML()
yaml.register_class(SolarInstallation)
yaml.register_class(PanelString)
yaml.register_class(PanelGroup)
yaml.dump(solar_installation, sys.stdout)

который дает:

!SolarInstallation
SSW_12: !PanelString
  G1: !PanelGroup
    num_panels: 4
    panel_efficiency: 0.191
    panel_surface: 1.705
    elevation: 50
  G2: !PanelGroup
    num_panels: 5
    panel_efficiency: 0.191
    panel_surface: 1.705
    elevation: 50

После ответа я понимаю, что подклассы dict (такие как PanelString и SolarInstallation могут быть автоматически сброшены (с тегом) с небольшим изменением кода в ruamel.yaml. Я постараюсь запомнить это для будущих обновлений. Когда вы захотите загрузить это назад, вам нужно определить from_yaml class_methods.

Anthon 23.04.2022 15:07

Другие вопросы по теме