mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-11-15 01:05:42 +00:00
94 lines
3.9 KiB
Python
94 lines
3.9 KiB
Python
from flask_wtf import FlaskForm
|
|
from werkzeug.utils import secure_filename
|
|
from wtforms import (FileField, StringField, SubmitField,
|
|
ValidationError, IntegerField)
|
|
from wtforms.validators import DataRequired, Length
|
|
|
|
|
|
class AddCorpusFileForm(FlaskForm):
|
|
'''
|
|
Form to add a .vrt corpus file to the current corpus.
|
|
'''
|
|
# Required fields
|
|
author = StringField('Author', validators=[DataRequired(), Length(1, 255)])
|
|
file = FileField('File', validators=[DataRequired()])
|
|
publishing_year = IntegerField('Publishing year',
|
|
validators=[DataRequired()])
|
|
title = StringField('Title', validators=[DataRequired(), Length(1, 255)])
|
|
# Optional fields
|
|
address = StringField('Adress', validators=[Length(0, 255)])
|
|
booktitle = StringField('Booktitle', validators=[Length(0, 255)])
|
|
chapter = StringField('Chapter', validators=[Length(0, 255)])
|
|
editor = StringField('Editor', validators=[Length(0, 255)])
|
|
institution = StringField('Institution', validators=[Length(0, 255)])
|
|
journal = StringField('Journal', validators=[Length(0, 255)])
|
|
pages = StringField('Pages', validators=[Length(0, 255)])
|
|
publisher = StringField('Publisher', validators=[Length(0, 255)])
|
|
school = StringField('School', validators=[Length(0, 255)])
|
|
submit = SubmitField()
|
|
|
|
def __init__(self, corpus, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.corpus = corpus
|
|
|
|
def validate_file(self, field):
|
|
if not field.data.filename.lower().endswith('.vrt'):
|
|
raise ValidationError('File does not have an approved extension: '
|
|
'.vrt')
|
|
field.data.filename = secure_filename(field.data.filename)
|
|
for corpus_file in self.corpus.files:
|
|
if field.data.filename == corpus_file.filename:
|
|
raise ValidationError('File already registered to corpus.')
|
|
|
|
|
|
class EditCorpusFileForm(FlaskForm):
|
|
'''
|
|
Form to edit meta data of one corpus file.
|
|
'''
|
|
# Required fields
|
|
author = StringField('Author', validators=[DataRequired(), Length(1, 255)])
|
|
publishing_year = IntegerField('Publishing year',
|
|
validators=[DataRequired()])
|
|
title = StringField('Title', validators=[DataRequired(), Length(1, 255)])
|
|
# Optional fields
|
|
address = StringField('Adress', validators=[Length(0, 255)])
|
|
booktitle = StringField('Booktitle', validators=[Length(0, 255)])
|
|
chapter = StringField('Chapter', validators=[Length(0, 255)])
|
|
editor = StringField('Editor', validators=[Length(0, 255)])
|
|
institution = StringField('Institution', validators=[Length(0, 255)])
|
|
journal = StringField('Journal', validators=[Length(0, 255)])
|
|
pages = StringField('Pages', validators=[Length(0, 255)])
|
|
publisher = StringField('Publisher', validators=[Length(0, 255)])
|
|
school = StringField('School', validators=[Length(0, 255)])
|
|
submit = SubmitField()
|
|
|
|
|
|
class AddCorpusForm(FlaskForm):
|
|
'''
|
|
Form to add a a new corpus.
|
|
'''
|
|
description = StringField('Description',
|
|
validators=[DataRequired(), Length(1, 255)])
|
|
submit = SubmitField()
|
|
title = StringField('Title', validators=[DataRequired(), Length(1, 32)])
|
|
|
|
|
|
class ImportCorpusForm(FlaskForm):
|
|
'''
|
|
Form to import a corpus.
|
|
'''
|
|
description = StringField('Description',
|
|
validators=[DataRequired(), Length(1, 255)])
|
|
file = FileField('File', validators=[DataRequired()])
|
|
submit = SubmitField()
|
|
title = StringField('Title', validators=[DataRequired(), Length(1, 32)])
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def validate_file(self, field):
|
|
if not field.data.filename.lower().endswith('.zip'):
|
|
raise ValidationError('File does not have an approved extension: '
|
|
'.zip')
|
|
field.data.filename = secure_filename(field.data.filename)
|