mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-12-25 02:44:18 +00:00
Add file type validation
This commit is contained in:
parent
c958edbae1
commit
df7d618b2a
@ -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'
|
||||||
|
)
|
||||||
|
@ -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'
|
||||||
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user