Meshroom/meshroom/common/core.py

class CoreModel:

    def __init__(self, keyAttrName="name", **kwargs):
        self._objects = {}
        self._keyAttrName = keyAttrName

    def __len__(self):
        return len(self._objects)

    def __iter__(self):
        """ Enables iteration over the list of objects. """
        return iter(self._objects.values())

    def keys(self):
        return self._objects.keys()

    @property
    def objects(self):
        return self._objects

    # TODO: operator[]
    def get(self, key):
        """
        Raises a KeyError if key is not in the map.
        :param key:
        :return:
        """
        return self._objects[key]

    def add(self, obj):
        key = getattr(obj, self._keyAttrName, None)
        assert key is not None
        assert key not in self._objects
        self._objects[key] = obj

    def pop(self, key):
        assert key in self._objects
        return self._objects.pop(key)

    def remove(self, obj):
        assert obj in self._objects.values()
        return self._objects.pop(getattr(obj, self._keyAttrName))

    def clear(self):
        self._objects = {}


class CoreSignal:
    """ Simple signal/callback implementation """
    def __init__(self):
        self._callbacks = set()

    def emit(self, *args):
        # TODO: check if we really need this in non-UI mode
        # [cb(*args) for cb in self._callbacks]
        pass

    def connect(self, func):
        self._callbacks.add(func)


def CoreSlot(*args, **kwargs):
    def slot_decorator(func):
        def func_wrapper(*f_args, **f_kwargs):
            return func(*f_args, **f_kwargs)
        return func_wrapper
    return slot_decorator


class CoreProperty(property):
    def __init__(self, ptype, fget=None, fset=None, **kwargs):
        super(CoreProperty, self).__init__(fget, fset)


class CoreObject(object):
    def __init__(self, *args, **kwargs):
        super(CoreObject, self).__init__()


Model = CoreModel
Slot = CoreSlot
Signal = CoreSignal
Property = CoreProperty
BaseObject = CoreObject
Variant = object