mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-11-15 01:05:42 +00:00
120 lines
4.0 KiB
Python
120 lines
4.0 KiB
Python
from flask import current_app
|
|
from flask_wtf import FlaskForm
|
|
from flask_wtf.file import FileField, FileRequired
|
|
from wtforms import (
|
|
BooleanField,
|
|
StringField,
|
|
SubmitField,
|
|
SelectMultipleField,
|
|
IntegerField,
|
|
ValidationError
|
|
)
|
|
from wtforms.validators import InputRequired, Length
|
|
from app.services import SERVICES
|
|
|
|
class CreateContributionBaseForm(FlaskForm):
|
|
title = StringField(
|
|
'Title',
|
|
validators=[InputRequired(), Length(max=64)]
|
|
)
|
|
description = StringField(
|
|
'Description',
|
|
validators=[InputRequired(), Length(max=255)]
|
|
)
|
|
version = StringField(
|
|
'Version',
|
|
validators=[InputRequired(), Length(max=16)]
|
|
)
|
|
publisher = StringField(
|
|
'Publisher',
|
|
validators=[InputRequired(), Length(max=128)]
|
|
)
|
|
publisher_url = StringField(
|
|
'Publisher URL',
|
|
validators=[InputRequired(), Length(max=512)]
|
|
)
|
|
publishing_url = StringField(
|
|
'Publishing URL',
|
|
validators=[InputRequired(), Length(max=512)]
|
|
)
|
|
publishing_year = IntegerField(
|
|
'Publishing year',
|
|
validators=[InputRequired()]
|
|
)
|
|
shared = BooleanField(
|
|
'Shared'
|
|
)
|
|
submit = SubmitField()
|
|
|
|
class EditForm(CreateContributionBaseForm):
|
|
def prefill(self, model_file):
|
|
''' Pre-fill the form with data of an exististing corpus file '''
|
|
self.title.data = model_file.title
|
|
self.description.data = model_file.description
|
|
self.publisher.data = model_file.publisher
|
|
self.publishing_year.data = model_file.publishing_year
|
|
self.publisher_url.data = model_file.publisher_url
|
|
self.publishing_url.data = model_file.publishing_url
|
|
self.version.data = model_file.version
|
|
self.shared.data = model_file.shared
|
|
|
|
class EditTesseractOCRModelForm(EditForm):
|
|
pass
|
|
|
|
class EditSpaCyNLPPipelineModelForm(EditForm):
|
|
pipeline_name = StringField(
|
|
'Pipeline name',
|
|
validators=[InputRequired(), Length(max=64)]
|
|
)
|
|
def prefill(self, model_file):
|
|
super().prefill(model_file)
|
|
self.pipeline_name.data = model_file.pipeline_name
|
|
|
|
class TesseractOCRModelContributionForm(CreateContributionBaseForm):
|
|
tesseract_model_file = FileField(
|
|
'File',
|
|
validators=[FileRequired()]
|
|
)
|
|
compatible_service_versions = SelectMultipleField(
|
|
'Compatible service versions'
|
|
)
|
|
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):
|
|
service_manifest = SERVICES['tesseract-ocr-pipeline']
|
|
super().__init__(*args, **kwargs)
|
|
self.compatible_service_versions.choices = [('', 'Choose your option')]
|
|
self.compatible_service_versions.choices += [
|
|
(x, x) for x in service_manifest['versions'].keys()
|
|
]
|
|
self.compatible_service_versions.default = ''
|
|
|
|
class SpacyNLPModelContributionForm(CreateContributionBaseForm):
|
|
spacy_model_file = FileField(
|
|
'File',
|
|
validators=[FileRequired()]
|
|
)
|
|
compatible_service_versions = SelectMultipleField(
|
|
'Compatible service versions'
|
|
)
|
|
pipeline_name = StringField(
|
|
'Pipeline name',
|
|
validators=[InputRequired(), Length(max=64)]
|
|
)
|
|
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):
|
|
service_manifest = SERVICES['spacy-nlp-pipeline']
|
|
super().__init__(*args, **kwargs)
|
|
self.compatible_service_versions.choices = [('', 'Choose your option')]
|
|
self.compatible_service_versions.choices += [
|
|
(x, x) for x in service_manifest['versions'].keys()
|
|
]
|
|
self.compatible_service_versions.default = ''
|