Source code for landsgemeinde.views.votum

from morepath import redirect
from onegov.core.elements import BackLink, Link
from onegov.core.security import Private
from onegov.landsgemeinde import _
from onegov.landsgemeinde import LandsgemeindeApp
from onegov.landsgemeinde.collections import VotumCollection
from onegov.landsgemeinde.forms import VotumForm
from onegov.landsgemeinde.layouts import VotumCollectionLayout
from onegov.landsgemeinde.layouts import VotumLayout
from onegov.landsgemeinde.models import Votum
from onegov.landsgemeinde.utils import ensure_states
from onegov.landsgemeinde.utils import update_ticker


from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from onegov.core.types import RenderData
    from onegov.landsgemeinde.request import LandsgemeindeRequest
    from webob import Response


@LandsgemeindeApp.form(
    model=VotumCollection,
    name='new',
    template='form.pt',
    permission=Private,
    form=VotumForm
)


def add_votum(
    self: VotumCollection,
    request: 'LandsgemeindeRequest',
    form: VotumForm
) -> 'RenderData | Response':

    if form.submitted(request):
        votum = self.add(**form.get_useful_data())
        updated = ensure_states(votum)
        updated.add(votum)
        update_ticker(request, updated)
        request.success(_('Added a new votum'))

        return redirect(
            request.link(votum.agenda_item, fragment=f'votum-{votum.number}'),
        )

    form.number.data = form.next_number

    layout = VotumCollectionLayout(self, request)
    layout.breadcrumbs.append(Link(_('New'), '#'))
    layout.include_editor()
    layout.edit_mode = True
    layout.editmode_links[1] = BackLink(attrs={'class': 'cancel-link'})

    return {
        'layout': layout,
        'title': _('New votum'),
        'form': form,
    }



@LandsgemeindeApp.form(
    model=Votum,
    name='edit',
    template='form.pt',
    permission=Private,
    form=VotumForm
)


def edit_votum(
    self: Votum,
    request: 'LandsgemeindeRequest',
    form: VotumForm
) -> 'RenderData | Response':

    if form.submitted(request):
        form.populate_obj(self)
        updated = ensure_states(self)
        updated.add(self)
        update_ticker(request, updated)
        request.success(_('Your changes were saved'))
        return request.redirect(
            request.link(self.agenda_item, fragment=f'votum-{self.number}')
        )

    form.process(obj=self)

    layout = VotumLayout(self, request)
    layout.breadcrumbs.append(Link(_('Edit'), '#'))
    layout.edit_mode = True
    layout.editmode_links[1] = BackLink(attrs={'class': 'cancel-link'})

    return {
        'layout': layout,
        'title': layout.title,
        'form': form,
        'form_width': 'large'
    }



@LandsgemeindeApp.view(
    model=Votum,
    request_method='DELETE',
    permission=Private
)


def delete_votum(self: Votum, request: 'LandsgemeindeRequest') -> None:

    request.assert_valid_csrf_token()

    collection = VotumCollection(request.session)
    collection.delete(self)
    ensure_states(
        self.agenda_item.vota[-1]
        if self.agenda_item.vota else self.agenda_item
    )

    update_ticker(request, {self.agenda_item})