sporada/nopaque
1
0
Fork 0
mirror of https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git synced 2024-12-24 18:34:18 +00:00
Code Issues Projects Releases Wiki Activity

Add file type validation

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
app/main/forms.py
View File

@ -1,5 +1,5 @@
from flask_wtf import FlaskForm from flask_wtf import FlaskForm
from wtforms import MultipleFileField, StringField, SubmitField from wtforms import MultipleFileField, StringField, SubmitField, ValidationError
from wtforms.validators import DataRequired, Length from wtforms.validators import DataRequired, Length
@ -14,8 +14,10 @@ class CreateCorpusForm(FlaskForm):
'Title', 'Title',
validators=[DataRequired(), Length(1, 32)] validators=[DataRequired(), Length(1, 32)]
) )
'''
def validate_files(form, field): def validate_files(form, field):
for file in form.files.getlist[field.name]: for file in field.data:
if file.mimetype if not file.filename.lower().endswith('.vrt'):
''' raise ValidationError(
'File does not have an approved extension: .vrt'
)

10
app/services/forms.py
View File

@ -1,5 +1,5 @@
from flask_wtf import FlaskForm 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 from wtforms.validators import DataRequired, Length
@ -29,3 +29,11 @@ class CreateOCRJobForm(FlaskForm):
'Title', 'Title',
validators=[DataRequired(), Length(1, 32)] 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'
)