"""
Define and provide key-value mappings of :py:class:`betty.config.Configuration` instances.
"""
from __future__ import annotations
from abc import abstractmethod
from contextlib import suppress
from typing import (
Generic,
Iterable,
Iterator,
TypeVar,
TYPE_CHECKING,
)
from typing_extensions import override
from betty.assertion import assert_sequence, assert_mapping
from betty.config import Configuration
from betty.config.collections import ConfigurationCollection, ConfigurationKey
if TYPE_CHECKING:
from betty.serde.dump import Dump, DumpMapping, DumpSequence
from collections.abc import MutableMapping
_ConfigurationT = TypeVar("_ConfigurationT", bound=Configuration)
_ConfigurationKeyT = TypeVar("_ConfigurationKeyT", bound=ConfigurationKey)
class _ConfigurationMapping(
ConfigurationCollection[_ConfigurationKeyT, _ConfigurationT],
Generic[_ConfigurationKeyT, _ConfigurationT],
):
def __init__(
self,
configurations: Iterable[_ConfigurationT] | None = None,
):
self._configurations: MutableMapping[_ConfigurationKeyT, _ConfigurationT] = {}
super().__init__(configurations)
def __contains__(self, configuration_key: _ConfigurationKeyT) -> bool:
return configuration_key in self._configurations
@override
def __getitem__(self, configuration_key: _ConfigurationKeyT) -> _ConfigurationT:
return self._configurations[configuration_key]
@override
def __iter__(self) -> Iterator[_ConfigurationKeyT]:
return (configuration_key for configuration_key in self._configurations)
@override
def keys(self) -> Iterator[_ConfigurationKeyT]:
return (configuration_key for configuration_key in self._configurations)
@override
def values(self) -> Iterator[_ConfigurationT]:
yield from self._configurations.values()
@override
def replace(self, *configurations: _ConfigurationT) -> None:
self.clear()
self.append(*configurations)
@override
def prepend(self, *configurations: _ConfigurationT) -> None:
self.insert(0, *configurations)
@override
def append(self, *configurations: _ConfigurationT) -> None:
for configuration in configurations:
configuration_key = self._get_key(configuration)
with suppress(KeyError):
del self._configurations[configuration_key]
self._configurations[configuration_key] = configuration
@override
def insert(self, index: int, *configurations: _ConfigurationT) -> None:
self.remove(*map(self._get_key, configurations))
existing_configurations = list(self.values())
self._configurations = {
self._get_key(configuration): configuration
for configuration in (
*existing_configurations[:index],
*configurations,
*existing_configurations[index:],
)
}
@abstractmethod
def _get_key(self, configuration: _ConfigurationT) -> _ConfigurationKeyT:
pass
[docs]
class ConfigurationMapping(
_ConfigurationMapping[_ConfigurationKeyT, _ConfigurationT],
Generic[_ConfigurationKeyT, _ConfigurationT],
):
"""
A key-value mapping where values are :py:class:`betty.config.Configuration`.
To test your own subclasses, use :py:class:`betty.test_utils.config.collections.mapping.ConfigurationMappingTestBase`.
"""
@abstractmethod
def _load_key(self, item_dump: Dump, key_dump: str) -> Dump:
pass
@abstractmethod
def _dump_key(self, item_dump: Dump) -> tuple[Dump, str]:
pass
def __load_item_key(self, value_dump: DumpMapping[Dump], key_dump: str) -> Dump:
value_dump = self._load_key(value_dump, key_dump)
return value_dump
[docs]
@override
def load(self, dump: Dump) -> None:
self.clear()
self.replace(
*assert_mapping(self._load_item)(
{
item_key_dump: self.__load_item_key(item_value_dump, item_key_dump)
for item_key_dump, item_value_dump in assert_mapping(
assert_mapping()
)(dump).items()
}
).values()
)
[docs]
@override
def dump(self) -> DumpMapping[Dump]:
dump: DumpMapping[Dump] = {}
for configuration_item in self._configurations.values():
item_dump = configuration_item.dump()
item_dump, configuration_key = self._dump_key(item_dump)
dump[configuration_key] = item_dump
return dump
[docs]
class OrderedConfigurationMapping(
_ConfigurationMapping[_ConfigurationKeyT, _ConfigurationT],
Generic[_ConfigurationKeyT, _ConfigurationT],
):
"""
An ordered key-value mapping where values are :py:class:`betty.config.Configuration`.
To test your own subclasses, use :py:class:`betty.test_utils.config.collections.mapping.OrderedConfigurationMappingTestBase`.
"""
[docs]
@override
def load(self, dump: Dump) -> None:
self.replace(*assert_sequence(self._load_item)(dump))
[docs]
@override
def dump(self) -> DumpSequence[Dump]:
return [
configuration_item.dump()
for configuration_item in self._configurations.values()
]