Source code for election_day.forms.upload.vote

from __future__ import annotations

from onegov.election_day import _
from onegov.election_day.forms.upload.common import ALLOWED_MIME_TYPES
from onegov.election_day.forms.upload.common import MAX_FILE_SIZE
from onegov.form import Form
from onegov.form.fields import UploadField
from onegov.form.validators import FileSizeLimit
from wtforms.fields import RadioField
from wtforms.validators import DataRequired
from wtforms.validators import InputRequired


from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from onegov.election_day.models import Canton
    from onegov.election_day.models import Municipality
    from onegov.election_day.models import Vote




class UploadVoteForm(Form):



    file_format = RadioField(
        _('File format'),
        choices=[('internal', 'OneGov Cloud')],
        validators=[
            InputRequired()
        ],
        default='internal'
    )




    proposal = UploadField(
        label=_('Proposal / Results'),
        validators=[
            DataRequired(),
            FileSizeLimit(MAX_FILE_SIZE)
        ],
        allowed_mimetypes=ALLOWED_MIME_TYPES,
        depends_on=('file_format', '!wabsti_c'),
        render_kw={'force_simple': True}
    )




    sg_gemeinden = UploadField(
        label='SG_Gemeinden.csv',
        validators=[
            DataRequired(),
            FileSizeLimit(MAX_FILE_SIZE)
        ],
        allowed_mimetypes=ALLOWED_MIME_TYPES,
        depends_on=('file_format', 'wabsti_c'),
        render_kw={'force_simple': True}
    )




    sg_geschaefte = UploadField(
        label='SG_Geschaefte.csv',
        validators=[
            DataRequired(),
            FileSizeLimit(MAX_FILE_SIZE)
        ],
        allowed_mimetypes=ALLOWED_MIME_TYPES,
        depends_on=('file_format', 'wabsti_c'),
        render_kw={'force_simple': True}
    )




    def adjust(self, principal: Canton | Municipality, vote: Vote) -> None:
        """ Adjusts the form to the given principal and vote. """

        assert hasattr(vote, 'data_sources')
        if vote.data_sources:
            self.file_format.choices = [
                ('internal', 'OneGov Cloud'),
                ('wabsti_c', 'WabstiCExport')
            ]