Source code for pas.forms.parliamentarian_role

from __future__ import annotations

from datetime import date
from onegov.form import Form
from onegov.form.fields import ChosenSelectField
from onegov.form.fields import TranslatedSelectField
from onegov.pas import _
from onegov.pas.collections import ParliamentarianCollection
from onegov.pas.collections import ParliamentaryGroupCollection
from onegov.pas.collections import PartyCollection
from onegov.pas.models.parliamentarian_role import PARLIAMENTARIAN_ROLES
from onegov.pas.models.parliamentarian_role import PARLIAMENTARY_GROUP_ROLES
from onegov.pas.models.parliamentarian_role import PARTY_ROLES
from wtforms.fields import DateField
from wtforms.validators import InputRequired
from wtforms.validators import Optional

from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from collections.abc import Collection
    from onegov.pas.models import ParliamentarianRole
    from typing import Any



[docs]
class ParliamentarianRoleForm(Form):


[docs]
    parliamentarian_id = ChosenSelectField(
        label=_('Parliamentarian'),
        validators=[InputRequired()],
    )



[docs]
    role = TranslatedSelectField(
        label=_('Role'),
        choices=list(PARLIAMENTARIAN_ROLES.items()),
        validators=[InputRequired()],
        default='member'
    )



[docs]
    start = DateField(
        label=_('Start'),
        validators=[InputRequired()],
        default=date.today
    )



[docs]
    end = DateField(
        label=_('End'),
        validators=[Optional()],
    )



[docs]
    party_id = ChosenSelectField(
        label=_('Party'),
    )



[docs]
    party_role = TranslatedSelectField(
        label=_('Party role'),
        choices=list(PARTY_ROLES.items()),
        default='none'
    )



[docs]
    parliamentary_group_id = ChosenSelectField(
        label=_('Parliamentary group'),
    )



[docs]
    parliamentary_group_role = TranslatedSelectField(
        label=_('Parliamentary group role'),
        choices=list(PARLIAMENTARY_GROUP_ROLES.items()),
        default='none'
    )



[docs]
    def on_request(self) -> None:
        self.parliamentarian_id.choices = [
            (str(parliamentarian.id), parliamentarian.title)
            for parliamentarian
            in ParliamentarianCollection(self.request.session).query()
        ]
        self.parliamentary_group_id.choices = [
            (str(parliamentary_group.id), parliamentary_group.title)
            for parliamentary_group
            in ParliamentaryGroupCollection(self.request.session).query()
        ]
        self.parliamentary_group_id.choices.insert(0, ('', '-'))
        self.party_id.choices = [
            (str(party.id), party.title)
            for party
            in PartyCollection(self.request.session).query()
        ]
        self.party_id.choices.insert(0, ('', '-'))



[docs]
    def populate_obj(  # type: ignore[override]
        self,
        obj: ParliamentarianRole,  # type: ignore[override]
        exclude: Collection[str] | None = None,
        include: Collection[str] | None = None
    ) -> None:
        super().populate_obj(obj, exclude, include)
        obj.parliamentary_group_id = obj.parliamentary_group_id or None
        obj.party_id = obj.party_id or None



[docs]
    def get_useful_data(self) -> dict[str, Any]:  # type:ignore[override]
        result = super().get_useful_data()
        result['parliamentary_group_id'] = result.get(
            'parliamentary_group_id') or None
        result['party_id'] = result.get('party_id') or None
        return result