Skip to content
Snippets Groups Projects
Commit df7d618b authored by Patrick Jentsch's avatar Patrick Jentsch
Browse files

Add file type validation

parent c958edba
No related branches found
No related tags found
No related merge requests found
from flask_wtf import FlaskForm
from wtforms import MultipleFileField, StringField, SubmitField
from wtforms import MultipleFileField, StringField, SubmitField, ValidationError
from wtforms.validators import DataRequired, Length
......@@ -14,8 +14,10 @@ class CreateCorpusForm(FlaskForm):
'Title',
validators=[DataRequired(), Length(1, 32)]
)
'''
def validate_files(form, field):
for file in form.files.getlist[field.name]:
if file.mimetype
'''
for file in field.data:
if not file.filename.lower().endswith('.vrt'):
raise ValidationError(
'File does not have an approved extension: .vrt'
)
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'
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment