Source code for org.forms.form_export

from onegov.form import merge_forms
from onegov.form.fields import MultiCheckboxField
from onegov.org import _
from onegov.org.forms.generic import DateRangeForm, ExportForm
from operator import attrgetter
from wtforms.fields import RadioField
from wtforms.validators import InputRequired
from wtforms.fields import DateField
from wtforms.fields import SelectChoice


from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from onegov.org.request import OrgRequest


[docs]
    class FormSubmissionsExportBase(DateRangeForm, ExportForm):
        pass

else:
    FormSubmissionsExportBase = merge_forms(DateRangeForm, ExportForm)



[docs]
def registration_window_choices(
    form: FormSubmissionsExport,
    field: MultiCheckboxField
) -> list[SelectChoice]:
    # FIXME: circular import - a layout class just to get a date format?
    from onegov.org.layout import DefaultLayout

    layout = DefaultLayout(form.model, form.request)
    return [
        SelectChoice(
            window.id.hex,
            layout.format_date_range(window.start, window.end)
        )
        for window in sorted(
            form.model.registration_windows,
            key=attrgetter('start')
        )
    ]




[docs]
class FormSubmissionsExport(FormSubmissionsExportBase):

    if TYPE_CHECKING:

[docs]
        request: OrgRequest



[docs]
    selection = RadioField(
        label=_('Selection'),
        default='date',
        choices=[
            SelectChoice('date', _('By date')),
            SelectChoice('window', _('By registration window'))
        ]
    )



[docs]
    registration_window = MultiCheckboxField(
        label=_('Registration Window'),
        choices=registration_window_choices,
        depends_on=('selection', 'window')
    )



[docs]
    start = DateField(
        label=_('Start'),
        validators=[InputRequired()],
        depends_on=('selection', 'date')
    )



[docs]
    end = DateField(
        label=_('End'),
        validators=[InputRequired()],
        depends_on=('selection', 'date')
    )



[docs]
    def on_request(self) -> None:
        if not self.registration_window.choices:
            self.hide(self.selection)