moment/harmonyqml/backend/model/list_model_map.py

from typing import Any, Callable, DefaultDict, MutableSequence

from PyQt5.QtCore import QObject, pyqtSlot

from .list_model import ListModel


class ListModelMap(QObject):
    def __init__(self,
                 parent:           QObject,
                 models_container: Callable[..., MutableSequence] = list
                ) -> None:
        super().__init__(parent)

        # Set the parent to prevent item garbage-collection on the C++ side
        self.dict: DefaultDict[Any, ListModel] = \
            DefaultDict(lambda: ListModel(parent    = self,
                                          container = models_container))


    @pyqtSlot(str, result="QVariant")
    def get(self, key) -> ListModel:
        return self.dict[key]


    @pyqtSlot(str, result=bool)
    def has(self, key) -> bool:
        return key in self.dict


    def __getitem__(self, key) -> ListModel:
        return self.dict[key]


    def __setitem__(self, key, value: ListModel) -> None:
        value.setParent(self)
        self.dict[key] = value


    def __detitem__(self, key) -> None:
        del self.dict[key]


    def __iter__(self):
        return iter(self.dict)


    def __len__(self) -> int:
        return len(self.dict)