Source code for town6.views.allocation

from __future__ import annotations

from onegov.core.security import Private
from onegov.org.forms import BatchCopyAllocationRulesForm
from onegov.org.views.allocation import (
    get_allocation_rule_form_class,
    get_edit_allocation_form_class,
    handle_allocation_rule,
    handle_copy_rules,
    handle_edit_allocation,
    handle_edit_rule,
    view_allocation_rules,
)
from onegov.town6 import TownApp
from onegov.reservation import Allocation
from onegov.reservation import Resource
from onegov.town6.layout import (
    AllocationEditFormLayout,
    AllocationRulesLayout,
)


from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from onegov.core.types import RenderData
    from onegov.org.forms import AllocationRuleForm
    from onegov.org.views.allocation import AllocationEditForm
    from onegov.town6.request import TownRequest
    from webob import Response


@TownApp.html(model=Resource, name='rules', permission=Private,
              template='allocation_rules.pt')

[docs]
def town_view_allocation_rules(
    self: Resource,
    request: TownRequest
) -> RenderData:
    return view_allocation_rules(
        self, request, AllocationRulesLayout(self, request))



@TownApp.form(model=Allocation, template='form.pt', name='edit',
              permission=Private, form=get_edit_allocation_form_class)

[docs]
def town_handle_edit_allocation(
    self: Allocation,
    request: TownRequest,
    form: AllocationEditForm
) -> RenderData | Response:
    """ Handles edit allocation for differing form classes. """

    return handle_edit_allocation(
        self, request, form, AllocationEditFormLayout(self, request))



@TownApp.form(model=Resource, template='form.pt', name='new-rule',
              permission=Private, form=get_allocation_rule_form_class)

[docs]
def town_handle_allocation_rule(
    self: Resource,
    request: TownRequest,
    form: AllocationRuleForm
) -> RenderData | Response:
    return handle_allocation_rule(
        self, request, form, AllocationRulesLayout(self, request))



@TownApp.form(model=Resource, template='form.pt', name='edit-rule',
              permission=Private, form=get_allocation_rule_form_class)

[docs]
def town_handle_edit_rule(
    self: Resource,
    request: TownRequest,
    form: AllocationRuleForm,
) -> RenderData | Response:
    return handle_edit_rule(
        self, request, form, AllocationRulesLayout(self, request))



@TownApp.form(
    model=Resource,
    template='form.pt',
    name='copy-rules',
    permission=Private,
    form=BatchCopyAllocationRulesForm,
)

[docs]
def town_handle_copy_rules(
    self: Resource,
    request: TownRequest,
    form: BatchCopyAllocationRulesForm,
) -> RenderData | Response:
    return handle_copy_rules(
        self, request, form, AllocationRulesLayout(self, request)
    )