from flask_wtf import FlaskForm from werkzeug.utils import secure_filename from wtforms import (BooleanField, FileField, StringField, SubmitField, ValidationError, IntegerField, SelectField) from wtforms.validators import DataRequired, Length, NumberRange class AddCorpusFileForm(FlaskForm): address = StringField('Adress', validators=[Length(0, 255)]) author = StringField('Author', validators=[DataRequired(), Length(1, 255)]) booktitle = StringField('Booktitle', validators=[Length(0, 255)]) chapter = StringField('Chapter', validators=[Length(0, 255)]) editor = StringField('Editor', validators=[Length(0, 255)]) file = FileField('File', validators=[DataRequired()]) 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)]) publishing_year = IntegerField('Publishing year', validators=[DataRequired()]) school = StringField('School', validators=[Length(0, 255)]) submit = SubmitField() title = StringField('Title', validators=[DataRequired(), Length(1, 255)]) def __init__(self, corpus, *args, **kwargs): super(AddCorpusFileForm, self).__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') filename = secure_filename(field.data.filename) for corpus_file in self.corpus.files: if filename == corpus_file.filename: raise ValidationError('File already registered to this corpus.') class EditCorpusFileForm(FlaskForm): address = StringField('Adress', validators=[Length(0, 255)]) author = StringField('Author', validators=[DataRequired(), Length(1, 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)]) publishing_year = IntegerField('Publishing year', validators=[DataRequired()]) school = StringField('School', validators=[Length(0, 255)]) submit = SubmitField() title = StringField('Title', validators=[DataRequired(), Length(1, 255)]) class AddCorpusForm(FlaskForm): description = StringField('Description', validators=[DataRequired(), Length(1, 255)]) submit = SubmitField() title = StringField('Title', validators=[DataRequired(), Length(1, 32)]) class QueryForm(FlaskForm): query = StringField('Query', validators=[DataRequired(), Length(1, 1024)]) submit = SubmitField('Send query') class DisplayOptionsForm(FlaskForm): expert_mode = BooleanField('Expert mode') result_context = SelectField('Result context', choices=[('', 'Choose your option'), ('10', '10'), ('20', '20'), ('30', '30'), ('40', '40'), ('50', '50'), ('60', '60'), ('70', '70'), ('80', '80'), ('90', '90'), ('100', '100')]) results_per_page = SelectField('Results per page', choices=[('', 'Choose your option'), ('10', '10'), ('20', '20'), ('30', '30'), ('40', '40'), ('50', '50'), ('60', '60'), ('70', '70'), ('80', '80'), ('90', '90'), ('100', '100')]) class InspectDisplayOptionsForm(FlaskForm): expert_mode_inspect = BooleanField('Expert mode') highlight_sentences = BooleanField('Highlight sentences') context_sentences = IntegerField('Context sentences', validators=[NumberRange(min=0, max=10)], default=3) class QueryDownloadForm(FlaskForm): file_type = SelectField('File type', choices=[('', 'Choose file type'), ('csv', 'csv'), ('json', 'json'), ('excel', 'excel'), ('html', 'html-table')], validators=[DataRequired()])