Add form and service

This commit is contained in:
Stephan Porada 2020-01-07 13:03:42 +01:00
parent 65b00e47ed
commit 93891a34c9
2 changed files with 18 additions and 2 deletions

View File

@ -65,3 +65,16 @@ class AddOCRJobForm(FlaskForm):
'File does not have an approved extension: ' 'File does not have an approved extension: '
'.pdf | .tif | .tiff' '.pdf | .tif | .tiff'
) )
class AddMergeImagesJobForm(FlaskForm):
files = MultipleFileField('Files', validators=[DataRequired()])
def validate_files(form, filed):
for file in field.data:
if not file.filename.lower().endswith(('.jpeg', '.jpg', '.png',
'.tiff', '.tif')):
raise ValidationError(
'File does not have an approved extension: '
'.jpeg | .jpg | .png | .tiff | .tif'
)

View File

@ -1,5 +1,5 @@
from app import db from app import db
from app.jobs.forms import AddNLPJobForm, AddOCRJobForm from app.jobs.forms import AddNLPJobForm, AddOCRJobForm, AddMergeImagesJobForm
from app.models import Job, JobInput from app.models import Job, JobInput
from flask import (abort, current_app, flash, make_response, render_template, from flask import (abort, current_app, flash, make_response, render_template,
url_for) url_for)
@ -10,7 +10,10 @@ import json
import os import os
SERVICES = {'nlp': {'name': 'Natural Language Processing', SERVICES = {'merge_images': {'name': 'Merge Images',
'resources': {'mem_mb': 4096, 'n_cores': 4},
'add_job_form': AddMergeImagesJobForm},
'nlp': {'name': 'Natural Language Processing',
'resources': {'mem_mb': 4096, 'n_cores': 2}, 'resources': {'mem_mb': 4096, 'n_cores': 2},
'add_job_form': AddNLPJobForm}, 'add_job_form': AddNLPJobForm},
'ocr': {'name': 'Optical Character Recognition', 'ocr': {'name': 'Optical Character Recognition',