Small changes in the contribution package

This commit is contained in:
Inga Kirschnick
2022-11-08 14:11:57 +01:00
parent 05340ea7ff
commit 033872718e
10 changed files with 230 additions and 70 deletions

View File

@ -1,4 +1,4 @@
from xml.dom import ValidationErr
from flask import current_app
from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired
from wtforms import (
@ -66,8 +66,9 @@ class TesseractOCRModelContributionForm(CreateContributionBaseForm):
compatible_service_versions = SelectMultipleField(
'Compatible service versions'
)
def validate_traineddata(self, field):
if field.data.mimetype != '.traineddata':
def validate_tesseract_model_file(self, field):
current_app.logger.warning(field.data.filename)
if not field.data.filename.lower().endswith('.traineddata'):
raise ValidationError('traineddata files only!')
def __init__(self, *args, **kwargs):
@ -87,8 +88,9 @@ class SpacyNLPModelContributionForm(CreateContributionBaseForm):
compatible_service_versions = SelectMultipleField(
'Compatible service versions'
)
def validate_spacy(self, field):
if field.data.mimetype != '.tar.gz':
def validate_spacy_model_file(self, field):
current_app.logger.warning(field.data.filename)
if not field.data.filename.lower().endswith('.tar.gz'):
raise ValidationError('.tar.gz files only!')
def __init__(self, *args, **kwargs):