62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
from collections.abc import MutableMapping, MutableSequence
|
|
from typing import TypeVar
|
|
|
|
from . import events as e
|
|
from .container import HookedContainer, HookFunction, MutableNesting
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
class HookedMapping(HookedContainer[T], MutableMapping[T, MutableNesting[T]]):
|
|
_data: MutableMapping[T, MutableNesting[T]]
|
|
|
|
def __init__(
|
|
self,
|
|
existing: MutableMapping[T, MutableNesting[T]],
|
|
hook: HookFunction | None = None,
|
|
path: MutableSequence[int] | None = None,
|
|
):
|
|
self.hook = hook
|
|
match existing:
|
|
case HookedMapping(_data=seq):
|
|
self._data = seq
|
|
case MutableMapping() as seq:
|
|
self._data = seq
|
|
case _ as it:
|
|
self._data = dict(it)
|
|
self._data = existing
|
|
self._path = list(path) if path is not None else []
|
|
|
|
# MutableMapping Methods
|
|
|
|
def __iter__(self):
|
|
return iter(self._data)
|
|
|
|
def insert(self, key: T, value: MutableNesting[T]) -> None:
|
|
self._data[key] = value
|
|
if self.hook:
|
|
self.hook(e.AddItemEvent(index=key, item=value, path=self.new_path(key)))
|
|
|
|
# HookedContainer methods
|
|
|
|
def __getitem__(self, key: T) -> MutableNesting[T]:
|
|
value = self._data.get(key, {})
|
|
match value:
|
|
case HookedMapping(_data=seq):
|
|
return type(self)(seq, hook=self.hook, path=self.new_path(key))
|
|
case MutableMapping() as mapping:
|
|
return HookedMapping(mapping, hook=self.hook, path=self.new_path(key))
|
|
case _ as item:
|
|
return item
|
|
|
|
def __setitem__(self, key: T, value: MutableNesting[T]) -> None:
|
|
self._data[key] = value
|
|
if self.hook:
|
|
self.hook(e.SetItemEvent(index=key, item=value, path=self.new_path(key)))
|
|
|
|
def __delitem__(self, key: T) -> None:
|
|
item = self._data[key]
|
|
del self._data[key]
|
|
if self.hook:
|
|
self.hook(e.RemoveItemEvent(index=key, item=item, path=self.new_path(key)))
|