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

## /utils/serialization/implementations/toml_serializer.py

from tomllib import loads, load, TOMLDecodeError
from tomli_w import dumps, dump
from os.path import exists
from typing import Any
from ..serializer import Serializer
from ..serializer_registry import register_serializer
from ..err import SerializationError

__all__ = [
    'TOMLSerializer'
]




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



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

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




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

        :param data: TOML-строка.
        :param kwargs: Ключевые параметры для сереализатора.
        :return: Словарь, полученный из строки.
        :raises SerializationError: При ошибках десериализации.
        """
        try:
            return loads(data, **kwargs)  # type: ignore[return-value]
        except TOMLDecodeError as e:
            raise SerializationError("Deserialization from toml failed") from e




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

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




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

        :param filename: Путь к файлу для загрузки.
        :param kwargs: Ключевые параметры для сереализатора.
        :return: Словарь, полученный из файла.
        :raises SerializationError: При ошибках десериализации.
        """
        try:
            if not exists(filename):
                raise FileNotFoundError(f"File {filename} not found")
            with open(filename, 'rb') as f:
                return load(f, **kwargs)  # type: ignore[return-value]
        except (IOError, OSError, FileNotFoundError, TOMLDecodeError) as e:
            raise SerializationError("Failed to deserialize data from file") from e