Source code for eventsourcing.domain.model.collection

from __future__ import absolute_import, division, print_function, unicode_literals

from eventsourcing.domain.model.entity import AbstractEntityRepository, TimestampedVersionedEntity


class Collection(TimestampedVersionedEntity):
    class Event(TimestampedVersionedEntity.Event):
        """Supertype for events of collection entities."""

    class Created(Event, TimestampedVersionedEntity.Created):
        """Published when collection is created."""

    class Discarded(Event, TimestampedVersionedEntity.Discarded):
        """Published when collection is discarded."""

    class EventWithItem(Event):
        @property
        def item(self):
            return self.__dict__['item']

    class ItemAdded(EventWithItem):
        def __mutate__(self, obj):
            obj = super(Collection.ItemAdded, self).__mutate__(obj)
            obj._items.add(self.item)
            return obj

    class ItemRemoved(EventWithItem):
        def __mutate__(self, obj):
            obj = super(Collection.ItemRemoved, self).__mutate__(obj)
            obj._items.remove(self.item)
            return obj

    def __init__(self, **kwargs):
        super(Collection, self).__init__(**kwargs)
        self._items = set()

    def __iter__(self):
        return self._items.__iter__()

    @property
    def items(self):
        self.__assert_not_discarded__()
        return self._items

    def add_item(self, item):
        self.__trigger_event__(self.ItemAdded, item=item)

    def remove_item(self, item):
        self.__trigger_event__(self.ItemRemoved, item=item)


def register_new_collection(collection_id=None):
    return Collection.__create__(originator_id=collection_id)


class AbstractCollectionRepository(AbstractEntityRepository):
    pass