Move route, cleanup and first fixes

This commit is contained in:
Patrick Jentsch 2022-10-26 10:23:22 +02:00
parent 36d3ea2ce0
commit 24851b38d9

View File

@ -1,11 +1,10 @@
from apifairy import authenticate
from flask import render_template, flash, abort
from flask_login import login_required, current_user
from flask import abort, flash, Markup, render_template, url_for
from flask_login import login_required
from app import db
from app.decorators import permission_required
from app.models import Permission, TesseractOCRModel, JobStatus
from app.models import TesseractOCRModel, Permission
from . import bp
from .forms import ContributionForm
from .. import db
@bp.before_request
@ -15,56 +14,39 @@ def before_request():
pass
@bp.route('', methods=['GET', 'POST'])
@login_required
@bp.route('')
def contributions():
form = ContributionForm(prefix='contribution-form')
pass
# Is the form submitted?
@bp.route('/tesseract-ocr-pipeline-models', methods=['GET', 'POST'])
def tesseract_ocr_pipeline_models():
form = ContributionForm(
prefix='contribute-tesseract-ocr-pipeline-model-form'
)
if form.is_submitted():
# Are their any false inputs?
if not form.validate():
response = {'errors': form.errors}
return response, 400
# At this point we can be sure that the form data is okay
# example
# compatible_service_versions = ['0.1.0', '0.1.1']
compatible_service_versions=form.compatible_service_versions.data
description=form.description
publisher=form.publisher
publisher_url=form.publisher_url
publishing_url=form.publishing_url
publishing_year=form.publishing_year
shared=False
title=form.title
version=form.version
try:
TesseractOCRModel.create(form.file.data,
compatible_service_versions,
description,
publisher,
publisher_url,
publishing_url,
publishing_year,
shared,
title,
version)
tesseract_ocr_model = TesseractOCRModel.create(
form.file.data,
compatible_service_versions=form.compatible_service_versions.data,
description=form.description.data,
publisher=form.publisher.data,
publisher_url=form.publisher_url.data,
publishing_url=form.publishing_url.data,
publishing_year=form.publishing_year.data,
shared=form.shared.data,
title=form.title.data,
version=form.version.data
)
except OSError:
abort(500)
# job.status = JobStatus.SUBMITTED
db.session.commit()
# message = Markup(f'Job "<a href="{job.url}">{job.title}</a>" created')
flash("Model has been added.")
return {}, 201, {'message': 'Model has been created'}
message = Markup(f'Model "{tesseract_ocr_model.title}" created')
flash(message)
return {}, 201, {'Location': url_for('contributions.contributions')}
return render_template(
'contributions/contribute.html.j2',
form=form,