eventcally/forms/event.py

from flask import request
from flask_babelex import lazy_gettext, gettext
from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileAllowed
from wtforms import FieldList, RadioField, DateTimeField, StringField, SubmitField, TextAreaField, SelectField, BooleanField, IntegerField, FormField
from wtforms.fields.html5 import DateTimeLocalField, EmailField, URLField
from wtforms.validators import DataRequired, Optional
from wtforms.widgets import html_params, HTMLString
from models import EventPlace, EventTargetGroupOrigin, EventAttendanceMode, EventStatus, Location, EventOrganizer, EventRejectionReason, EventReviewStatus, Image
from .common import event_rating_choices, Base64ImageForm
from .widgets import CustomDateTimeField, CustomDateField

class EventPlaceLocationForm(FlaskForm):
    street = StringField(lazy_gettext('Street'), validators=[Optional()])
    postalCode = StringField(lazy_gettext('Postal code'), validators=[Optional()])
    city = StringField(lazy_gettext('City'), validators=[Optional()])

class EventPlaceForm(FlaskForm):
    name = StringField(lazy_gettext('Name'), validators=[Optional()])
    location = FormField(EventPlaceLocationForm, default=lambda: Location())

    def populate_obj(self, obj):
        for name, field in self._fields.items():
            if name == 'location' and not obj.location:
                obj.location = Location()
            field.populate_obj(obj, name)

class OrganizerForm(EventPlaceForm):
    pass

class EventOrganizerForm(FlaskForm):
    name = StringField(lazy_gettext('Organizator'), validators=[Optional()])
    url = URLField(lazy_gettext('Link URL'), validators=[Optional()])
    email = EmailField(lazy_gettext('Email'), validators=[Optional()])
    phone = StringField(lazy_gettext('Phone'), validators=[Optional()])
    fax = StringField(lazy_gettext('Fax'), validators=[Optional()])

class BaseEventForm(FlaskForm):
    name = StringField(lazy_gettext('Name'), validators=[DataRequired()])
    external_link = URLField(lazy_gettext('Link URL'), validators=[Optional()])
    ticket_link = StringField(lazy_gettext('Ticket Link URL'), validators=[Optional()])
    description = TextAreaField(lazy_gettext('Description'), validators=[DataRequired()])
    recurrence_rule = TextAreaField(lazy_gettext('Recurrence rule'), validators=[Optional()])
    start = CustomDateTimeField(lazy_gettext('Start'), validators=[DataRequired()])
    end = CustomDateTimeField(lazy_gettext('End'), validators=[Optional()])
    previous_start_date = CustomDateTimeField(lazy_gettext('Previous start date'), validators=[Optional()])
    tags = StringField(lazy_gettext('Tags'), validators=[Optional()])

    category_id = SelectField(lazy_gettext('Category'), validators=[DataRequired()], coerce=int)

    kid_friendly = BooleanField(lazy_gettext('Kid friendly'), validators=[Optional()])
    accessible_for_free = BooleanField(lazy_gettext('Accessible for free'), validators=[Optional()])
    age_from = IntegerField(lazy_gettext('Typical Age from'), validators=[Optional()])
    age_to = IntegerField(lazy_gettext('Typical Age to'), validators=[Optional()])
    registration_required = BooleanField(lazy_gettext('Registration required'), validators=[Optional()])
    booked_up = BooleanField(lazy_gettext('Booked up'), validators=[Optional()])
    expected_participants = IntegerField(lazy_gettext('Expected number of participants'), validators=[Optional()])
    price_info = TextAreaField(lazy_gettext('Price info'), validators=[Optional()])

    target_group_origin = SelectField(lazy_gettext('Target group origin'), coerce=int, choices=[
        (int(EventTargetGroupOrigin.both), lazy_gettext('EventTargetGroupOrigin.both')),
        (int(EventTargetGroupOrigin.tourist), lazy_gettext('EventTargetGroupOrigin.tourist')),
        (int(EventTargetGroupOrigin.resident), lazy_gettext('EventTargetGroupOrigin.resident'))])

    attendance_mode = SelectField(lazy_gettext('Attendance mode'), coerce=int, choices=[
        (int(EventAttendanceMode.offline), lazy_gettext('EventAttendanceMode.offline')),
        (int(EventAttendanceMode.online), lazy_gettext('EventAttendanceMode.online')),
        (int(EventAttendanceMode.mixed), lazy_gettext('EventAttendanceMode.mixed'))])

    photo = FormField(Base64ImageForm, lazy_gettext('Photo'), default=lambda: Image())
    rating = SelectField(lazy_gettext('Rating'), default=50, coerce=int, choices=event_rating_choices)

class CreateEventForm(BaseEventForm):
    event_place_choice = RadioField(lazy_gettext('Place'), choices=[(1,lazy_gettext('Select existing place')), (2,lazy_gettext('Enter new place'))], default=1, coerce=int)
    event_place_id = SelectField(lazy_gettext('Place'), validators=[Optional()], coerce=int)
    new_event_place = FormField(EventPlaceForm, default=lambda: EventPlace())

    organizer_choice = RadioField(lazy_gettext('Organizer'), choices=[(1,lazy_gettext('Select existing organizer')), (2,lazy_gettext('Enter new organizer'))], default=1, coerce=int)
    organizer_id = SelectField(lazy_gettext('Organizer'), validators=[Optional()], coerce=int)
    new_organizer = FormField(OrganizerForm, default=lambda: EventOrganizer())

    submit = SubmitField(lazy_gettext("Create event"))

    def populate_obj(self, obj):
        for name, field in self._fields.items():
            if name == 'new_event_place':
                if self.event_place_choice.data != 2:
                    continue
                if not obj.event_place:
                    obj.event_place = EventPlace()
                field.populate_obj(obj, 'event_place')
            elif name == 'new_organizer':
                if self.organizer_choice.data != 2:
                    continue
                if not obj.organizer:
                    obj.organizer = EventOrganizer()
                field.populate_obj(obj, 'organizer')
            elif name == 'photo' and not obj.photo:
                obj.photo = Image()
            field.populate_obj(obj, name)

    def validate(self):
        if not super(BaseEventForm, self).validate():
            return False
        if self.event_place_id.data == 0 and not self.new_event_place.form.name.data:
            msg = gettext('Select existing place or enter new place')
            self.event_place_id.errors.append(msg)
            self.new_event_place.form.name.errors.append(msg)
            return False
        if self.organizer_id.data == 0 and not self.new_organizer.form.name.data:
            msg = gettext('Select existing organizer or enter new organizer')
            self.organizer_id.errors.append(msg)
            self.new_organizer.form.name.errors.append(msg)
            return False
        return True

class UpdateEventForm(BaseEventForm):
    event_place_id = SelectField(lazy_gettext('Place'), validators=[DataRequired()], coerce=int)
    organizer_id = SelectField(lazy_gettext('Organizer'), validators=[DataRequired()], coerce=int)

    status = SelectField(lazy_gettext('Status'), coerce=int, choices=[
        (int(EventStatus.scheduled), lazy_gettext('EventStatus.scheduled')),
        (int(EventStatus.cancelled), lazy_gettext('EventStatus.cancelled')),
        (int(EventStatus.movedOnline), lazy_gettext('EventStatus.movedOnline')),
        (int(EventStatus.postponed), lazy_gettext('EventStatus.postponed')),
        (int(EventStatus.rescheduled), lazy_gettext('EventStatus.rescheduled'))])

    submit = SubmitField(lazy_gettext("Update event"))

    def populate_obj(self, obj):
        for name, field in self._fields.items():
            if name == 'photo' and not obj.photo:
                obj.photo = Image()
            field.populate_obj(obj, name)

class DeleteEventForm(FlaskForm):
    submit = SubmitField(lazy_gettext("Delete event"))
    name = StringField(lazy_gettext('Name'), validators=[DataRequired()])

class FindEventForm(FlaskForm):
    class Meta:
        csrf = False
    date_from = CustomDateField(lazy_gettext('From'), validators=[Optional()])
    date_to = CustomDateField(lazy_gettext('to'), validators=[Optional()])
    keyword = StringField(lazy_gettext('Keyword'), validators=[Optional()])
    category_id = SelectField(lazy_gettext('Category'), validators=[Optional()], coerce=int)
    organizer_id = SelectField(lazy_gettext('Organizer'), validators=[Optional()], coerce=int)

    submit = SubmitField(lazy_gettext("Find events"))

    def is_submitted(self):
        return 'submit' in request.args