eventcally/project/forms/reference_request.py

from flask_babelex import lazy_gettext
from flask_wtf import FlaskForm
from wtforms import SelectField, StringField, SubmitField
from wtforms.validators import DataRequired
from project.forms.common import event_rating_choices
from project.models import (
    EventReferenceRequestRejectionReason,
    EventReferenceRequestReviewStatus,
)


class CreateEventReferenceRequestForm(FlaskForm):
    admin_unit_id = SelectField(
        lazy_gettext("Admin unit"), validators=[DataRequired()], coerce=int
    )
    submit = SubmitField(lazy_gettext("Save request"))


class DeleteReferenceRequestForm(FlaskForm):
    submit = SubmitField(lazy_gettext("Delete request"))
    name = StringField(lazy_gettext("Name"), validators=[DataRequired()])


class ReferenceRequestReviewForm(FlaskForm):
    review_status = SelectField(
        lazy_gettext("Review status"),
        coerce=int,
        choices=[
            (
                int(EventReferenceRequestReviewStatus.inbox),
                lazy_gettext("EventReferenceRequestReviewStatus.inbox"),
            ),
            (
                int(EventReferenceRequestReviewStatus.verified),
                lazy_gettext("EventReferenceRequestReviewStatus.verified"),
            ),
            (
                int(EventReferenceRequestReviewStatus.rejected),
                lazy_gettext("EventReferenceRequestReviewStatus.rejected"),
            ),
        ],
    )

    rejection_reason = SelectField(
        lazy_gettext("Rejection reason"),
        coerce=int,
        choices=[
            (0, ""),
            (
                int(EventReferenceRequestRejectionReason.duplicate),
                lazy_gettext("EventReferenceRequestRejectionReason.duplicate"),
            ),
            (
                int(EventReferenceRequestRejectionReason.untrustworthy),
                lazy_gettext("EventReferenceRequestRejectionReason.untrustworthy"),
            ),
            (
                int(EventReferenceRequestRejectionReason.irrelevant),
                lazy_gettext("EventReferenceRequestRejectionReason.irrelevant"),
            ),
            (
                int(EventReferenceRequestRejectionReason.illegal),
                lazy_gettext("EventReferenceRequestRejectionReason.illegal"),
            ),
        ],
    )

    rating = SelectField(
        lazy_gettext("Rating"), default=50, coerce=int, choices=event_rating_choices
    )
    submit = SubmitField(lazy_gettext("Save review"))