add tables for files

This commit is contained in:
Patrick Jentsch
2019-10-16 16:52:05 +02:00
parent 8158c742d9
commit abe2ed1666
8 changed files with 54 additions and 277 deletions

View File

@ -1,9 +1,10 @@
from flask import abort, current_app, flash, redirect, render_template, url_for
from . import services
from flask_login import current_user, login_required
from werkzeug.utils import secure_filename
from . import services
from .forms import NewNLPJobForm, NewOCRJobForm
from ..models import Job
from .. import db
from ..models import Job, JobInput
import json
import os
@ -53,7 +54,10 @@ def service(service_handle):
else:
for file in new_job_form.files.data:
''' TODO: Use secure filename '''
file.save(os.path.join(dir, file.filename))
filename = secure_filename(file.filename)
file.save(os.path.join(dir, filename))
job_input = JobInput(filename=filename, job=job)
db.session.add(job_input)
job.status = 'submitted'
db.session.commit()
flash('Job created!')