<reponame>politbuero-kampagnen/onegov-cloud<filename>src/onegov/reservation/collection.py
from onegov.core.utils import normalize_for_url
from onegov.reservation.models import Resource
from uuid import uuid4


any_type = object()


class ResourceCollection(object):
    """ Manages a list of resources.

    """
    def __init__(self, libres_context):
        assert hasattr(libres_context, 'get_service'), """
            The ResourceCollection expected the libres_contex, not the session.
        """

        self.libres_context = libres_context
        self.session = libres_context.get_service('session_provider').session()

    def query(self):
        return self.session.query(Resource)

    def add(self, title, timezone, type=None, name=None, meta={}, content={},
            definition=None, group=None):

        # look up the right class depending on the type (we need to do
        # this a bit akwardly here, because Resource does not use the
        # ModelBase as declarative base)
        resource = Resource.get_polymorphic_class(type, Resource)()
        resource.id == uuid4()
        resource.name = name or normalize_for_url(title)
        resource.title = title
        resource.timezone = timezone
        resource.meta = meta
        resource.content = content
        resource.definition = definition
        resource.group = group
        resource.renew_access_token()

        self.session.add(resource)
        self.session.flush()

        return self.bind(resource)

    def bind(self, resource):
        if resource:
            resource.bind_to_libres_context(self.libres_context)

        return resource

    def by_id(self, id, ensure_type=any_type):
        query = self.query().filter(Resource.id == id)

        if ensure_type is not any_type:
            query = query.filter(Resource.type == type)

        return self.bind(query.first())

    def by_name(self, name, ensure_type=any_type):
        query = self.query().filter(Resource.name == name)

        if ensure_type is not any_type:
            query = query.filter(Resource.type == type)

        return self.bind(query.first())

    def by_allocation(self, allocation):
        return self.by_id(allocation.resource)

    def by_reservation(self, reservation):
        return self.by_id(reservation.resource)

    def delete(self, resource, including_reservations=False,
               handle_reservation=None):
        scheduler = resource.get_scheduler(self.libres_context)

        if not including_reservations:
            assert not scheduler.managed_reserved_slots().first()
            assert not scheduler.managed_reservations().first()

            scheduler.managed_allocations().delete('fetch')
        else:
            if callable(handle_reservation):
                for res in scheduler.managed_reservations():
                    # e.g. create a ticket snapshot
                    handle_reservation(res)
            scheduler.extinguish_managed_records()

        self.session.delete(resource)
        self.session.flush()

from onegov.core.utils import normalize_for_url
from onegov.reservation.models import Resource
from uuid import uuid4


any_type = object()


class ResourceCollection(object):
    """ Manages a list of resources.

    """
    def __init__(self, libres_context):
        assert hasattr(libres_context, 'get_service'), """
            The ResourceCollection expected the libres_contex, not the session.
        """

        self.libres_context = libres_context
        self.session = libres_context.get_service('session_provider').session()

    def query(self):
        return self.session.query(Resource)

    def add(self, title, timezone, type=None, name=None, meta={}, content={},
            definition=None, group=None):

        # look up the right class depending on the type (we need to do
        # this a bit akwardly here, because Resource does not use the
        # ModelBase as declarative base)
        resource = Resource.get_polymorphic_class(type, Resource)()
        resource.id == uuid4()
        resource.name = name or normalize_for_url(title)
        resource.title = title
        resource.timezone = timezone
        resource.meta = meta
        resource.content = content
        resource.definition = definition
        resource.group = group
        resource.renew_access_token()

        self.session.add(resource)
        self.session.flush()

        return self.bind(resource)

    def bind(self, resource):
        if resource:
            resource.bind_to_libres_context(self.libres_context)

        return resource

    def by_id(self, id, ensure_type=any_type):
        query = self.query().filter(Resource.id == id)

        if ensure_type is not any_type:
            query = query.filter(Resource.type == type)

        return self.bind(query.first())

    def by_name(self, name, ensure_type=any_type):
        query = self.query().filter(Resource.name == name)

        if ensure_type is not any_type:
            query = query.filter(Resource.type == type)

        return self.bind(query.first())

    def by_allocation(self, allocation):
        return self.by_id(allocation.resource)

    def by_reservation(self, reservation):
        return self.by_id(reservation.resource)

    def delete(self, resource, including_reservations=False,
               handle_reservation=None):
        scheduler = resource.get_scheduler(self.libres_context)

        if not including_reservations:
            assert not scheduler.managed_reserved_slots().first()
            assert not scheduler.managed_reservations().first()

            scheduler.managed_allocations().delete('fetch')
        else:
            if callable(handle_reservation):
                for res in scheduler.managed_reservations():
                    # e.g. create a ticket snapshot
                    handle_reservation(res)
            scheduler.extinguish_managed_records()

        self.session.delete(resource)
        self.session.flush()
Manages a list of resources. The ResourceCollection expected the libres_contex, not the session. # look up the right class depending on the type (we need to do # this a bit akwardly here, because Resource does not use the # ModelBase as declarative base) # e.g. create a ticket snapshot