Исходный код EJIO.utils.serialization.implementations.yaml_serializer

## /utils/serialization/implementations/yaml_serializer.py

from yaml import dump, safe_load, YAMLError
from os.path import exists
from typing import Any
from ..serializer import Serializer
from ..serializer_registry import register_serializer
from ..err import SerializationError

__all__ = [
    'YAMLSerializer'
]




@register_serializer('yaml')
class YAMLSerializer(Serializer):
    """
    Реализация интерфейса Serializer для формата YAML.
    """
    __slots__ = ()



    @staticmethod
    def serialize[T](data: dict[str, T], /, **kwargs: Any) -> str:
        """
        Преобразовать словарь в YAML-строку.

        :param data: Сериализуемый словарь.
        :param kwargs: Ключевые параметры для сереализатора.
        :return: YAML-строка.
        :raises SerializationError: При ошибках сериализации.
        """
        try:
            return dump(data, **kwargs)
        except YAMLError as e:
            raise SerializationError("Serialization to yaml failed") from e




    @staticmethod
    def deserialize[T](data: str, /, **kwargs: Any) -> dict[str, T]:
        """
        Преобразовать YAML-строку обратно в словарь.

        :param data: YAML-строка.
        :param kwargs: Ключевые параметры для сереализатора.
        :return: Словарь, полученный из строки.
        :raises SerializationError: При ошибках десериализации.
        """
        try:
            result = safe_load(data)
            if not isinstance(result, dict):
                raise SerializationError("YAML data must represent a dictionary")
            return result
        except YAMLError as e:
            raise SerializationError("Deserialization from yaml failed") from e




    @staticmethod
    def save[T](data: dict[str, T], filename: str, /, **kwargs: Any) -> None:
        """
        Сохранить словарь в YAML-файл.

        :param data: Сериализуемый словарь.
        :param filename: Путь к файлу для сохранения.
        :param kwargs: Ключевые параметры для сереализатора.
        :raises SerializationError: При ошибках сериализации.
        """
        try:
            with open(filename, 'w', encoding='utf-8') as f:
                dump(data, f, **kwargs)
        except (IOError, OSError, YAMLError) as e:
            raise SerializationError("Failed to serialize data to file") from e




    @staticmethod
    def load[T](filename: str, /, **kwargs: Any) -> dict[str, T]:
        """
        Загрузить словарь из YAML-файла.

        :param filename: Путь к файлу для загрузки.
        :param kwargs: Ключевые параметры для сереализатора.
        :return: Словарь, полученный из файла.
        :raises SerializationError: При ошибках десериализации.
        """
        try:
            if not exists(filename):
                raise FileNotFoundError(f"File {filename} not found")
            with open(filename, 'r', encoding='utf-8') as f:
                result = safe_load(f)
            if not isinstance(result, dict):
                raise SerializationError("YAML file data must represent a dictionary")
            return result
        except (IOError, OSError, FileNotFoundError, YAMLError) as e:
            raise SerializationError("Failed to deserialize data from file") from e