from flask_wtf import FlaskForm from flask_wtf.file import FileField, FileAllowed, FileRequired, FileSize from wtforms import validators from wtforms.validators import ( Length, NumberRange, DataRequired, ValidationError, ) from wtforms import ( SubmitField, StringField, IntegerField, SelectField, ) class UploadForm(FlaskForm): """File upload class for a new site in distribusi-verse""" def distribusiname(form, field): if field.data.lower() == "new": raise ValidationError("Name has to be unique and not just new.") sitename = StringField( "Name of your website:", validators=[validators.InputRequired(), Length(2, 100), distribusiname], ) academicyear = StringField( "Academic year:", validators=[validators.InputRequired(), Length(9, 10)], ) term = SelectField( "Term:", validate_choice=True, coerce=str, choices=[ (u'1.2', u'1.2'), (u'2.3', u'2.3'), (u'3.1', u'3.1'), (u'4.1', u'4.1'), (u'4.2', u'4.2'), ], option_widget=None, validators=[DataRequired()] ) tags = StringField( "Add tags, press enter or add a comma after each tag:", validators=[validators.InputRequired(), Length(2, 500)], ) course = SelectField( u'Course:', validate_choice=True, coerce=str, choices=[ ('hacking', u'Autonomous - Hacking'), ('digitalcraft', u'Autonomous - Digital Craft'), ('criticalstudies', u'Autonomous - Critical Studies'), ('publicprivate', u'Autonomous - Public&Private'), ], option_widget=None, validators=[DataRequired()] ) zipfile = FileField( "Upload your zip file with content here:", validators=[ FileAllowed(["zip"], "Zip archives only!"), FileRequired(), FileSize( max_size=104857600, message="Zipfile size must be smaller than 100MB", ), ], ) submit = SubmitField("Upload")