Add file type validation

This commit is contained in:
Patrick Jentsch
2019-08-01 14:01:22 +02:00
parent c958edbae1
commit df7d618b2a
2 changed files with 16 additions and 6 deletions

View File

@ -1,5 +1,5 @@
from flask_wtf import FlaskForm
from wtforms import MultipleFileField, SelectField, StringField, SubmitField
from wtforms import MultipleFileField, SelectField, StringField, SubmitField, ValidationError
from wtforms.validators import DataRequired, Length
@ -29,3 +29,11 @@ class CreateOCRJobForm(FlaskForm):
'Title',
validators=[DataRequired(), Length(1, 32)]
)
def validate_files(form, field):
for file in field.data:
if not file.filename.lower().endswith(('.pdf', '.tif', '.tiff')):
raise ValidationError(
'File does not have an approved extension: '
'.pdf | .tif | .tiff'
)