mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2025-06-13 09:30:40 +00:00
Add file type validation
This commit is contained in:
@ -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'
|
||||
)
|
||||
|
Reference in New Issue
Block a user