mirror of
				https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
				synced 2025-11-04 04:12:45 +00:00 
			
		
		
		
	Move jobs namespace back to http routes
This commit is contained in:
		@@ -1,25 +1,49 @@
 | 
			
		||||
from flask import (
 | 
			
		||||
    abort,
 | 
			
		||||
    current_app,
 | 
			
		||||
    Flask,
 | 
			
		||||
    jsonify,
 | 
			
		||||
    redirect,
 | 
			
		||||
    render_template,
 | 
			
		||||
    send_from_directory,
 | 
			
		||||
    url_for
 | 
			
		||||
)
 | 
			
		||||
from flask_login import current_user
 | 
			
		||||
from app.models import Job, JobInput, JobResult
 | 
			
		||||
from threading import Thread
 | 
			
		||||
from app import db
 | 
			
		||||
from app.models import Job, JobInput, JobResult, JobStatus
 | 
			
		||||
from . import bp
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _delete_job(app: Flask, job_id: int):
 | 
			
		||||
    with app.app_context():
 | 
			
		||||
        job = Job.query.get(job_id)
 | 
			
		||||
        job.delete()
 | 
			
		||||
        db.session.commit()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _restart_job(app: Flask, job_id: int):
 | 
			
		||||
    with app.app_context():
 | 
			
		||||
        job = Job.query.get(job_id)
 | 
			
		||||
        job.restart()
 | 
			
		||||
        db.session.commit()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@bp.route('')
 | 
			
		||||
def jobs():
 | 
			
		||||
    return redirect(url_for('main.dashboard', _anchor='jobs'))
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@bp.route('/<hashid:job_id>')
 | 
			
		||||
def job(job_id):
 | 
			
		||||
def job(job_id: int):
 | 
			
		||||
    job = Job.query.get_or_404(job_id)
 | 
			
		||||
    if not (job.user == current_user or current_user.is_administrator):
 | 
			
		||||
 | 
			
		||||
    if not (
 | 
			
		||||
        job.user == current_user
 | 
			
		||||
        or current_user.is_administrator
 | 
			
		||||
    ):
 | 
			
		||||
        abort(403)
 | 
			
		||||
 | 
			
		||||
    return render_template(
 | 
			
		||||
        'jobs/job.html.j2',
 | 
			
		||||
        title='Job',
 | 
			
		||||
@@ -27,11 +51,77 @@ def job(job_id):
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@bp.route('/<hashid:job_id>/inputs/<hashid:job_input_id>/download')
 | 
			
		||||
def download_job_input(job_id, job_input_id):
 | 
			
		||||
    job_input = JobInput.query.filter_by(job_id=job_id, id=job_input_id).first_or_404()
 | 
			
		||||
    if not (job_input.job.user == current_user or current_user.is_administrator):
 | 
			
		||||
@bp.route('/<hashid:job_id>', methods=['DELETE'])
 | 
			
		||||
def delete_job(job_id: int):
 | 
			
		||||
    job = Job.query.get_or_404(job_id)
 | 
			
		||||
 | 
			
		||||
    if not (
 | 
			
		||||
        job.user == current_user
 | 
			
		||||
        or current_user.is_administrator
 | 
			
		||||
    ):
 | 
			
		||||
        abort(403)
 | 
			
		||||
 | 
			
		||||
    thread = Thread(
 | 
			
		||||
        target=_delete_job,
 | 
			
		||||
        args=(current_app._get_current_object(), job.id)
 | 
			
		||||
    )
 | 
			
		||||
    thread.start()
 | 
			
		||||
 | 
			
		||||
    return jsonify(f'Job "{job.title}" marked for deletion.'), 202
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@bp.route('/<hashid:job_id>/log')
 | 
			
		||||
def get_job_log(job_id: int):
 | 
			
		||||
    job = Job.query.get_or_404(job_id)
 | 
			
		||||
 | 
			
		||||
    if not current_user.is_administrator:
 | 
			
		||||
        abort(403)
 | 
			
		||||
 | 
			
		||||
    if job.status not in [JobStatus.COMPLETED, JobStatus.FAILED]:
 | 
			
		||||
        abort(409)
 | 
			
		||||
 | 
			
		||||
    log_file_path = job.path / 'pipeline_data' / 'logs' / 'pyflow_log.txt'
 | 
			
		||||
    with log_file_path.open() as log_file:
 | 
			
		||||
        log = log_file.read()
 | 
			
		||||
 | 
			
		||||
    return jsonify(log)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@bp.route('/<hashid:job_id>/restart')
 | 
			
		||||
def restart_job(job_id: int):
 | 
			
		||||
    job = Job.query.get_or_404(job_id)
 | 
			
		||||
 | 
			
		||||
    if not (
 | 
			
		||||
        job.user == current_user
 | 
			
		||||
        or current_user.is_administrator
 | 
			
		||||
    ):
 | 
			
		||||
        abort(403)
 | 
			
		||||
 | 
			
		||||
    if job.status != JobStatus.FAILED:
 | 
			
		||||
        abort(409)
 | 
			
		||||
 | 
			
		||||
    thread = Thread(
 | 
			
		||||
        target=_restart_job,
 | 
			
		||||
        args=(current_app._get_current_object(), job.id)
 | 
			
		||||
    )
 | 
			
		||||
    thread.start()
 | 
			
		||||
 | 
			
		||||
    return jsonify(f'Job "{job.title}" marked for restarting.'), 202
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@bp.route('/<hashid:job_id>/inputs/<hashid:job_input_id>/download')
 | 
			
		||||
def download_job_input(job_id: int, job_input_id: int):
 | 
			
		||||
    job_input = JobInput.query.filter_by(
 | 
			
		||||
        job_id=job_id,
 | 
			
		||||
        id=job_input_id
 | 
			
		||||
    ).first_or_404()
 | 
			
		||||
 | 
			
		||||
    if not (
 | 
			
		||||
        job_input.job.user == current_user
 | 
			
		||||
        or current_user.is_administrator
 | 
			
		||||
    ):
 | 
			
		||||
        abort(403)
 | 
			
		||||
 | 
			
		||||
    return send_from_directory(
 | 
			
		||||
        job_input.path.parent,
 | 
			
		||||
        job_input.path.name,
 | 
			
		||||
@@ -42,10 +132,18 @@ def download_job_input(job_id, job_input_id):
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@bp.route('/<hashid:job_id>/results/<hashid:job_result_id>/download')
 | 
			
		||||
def download_job_result(job_id, job_result_id):
 | 
			
		||||
    job_result = JobResult.query.filter_by(job_id=job_id, id=job_result_id).first_or_404()
 | 
			
		||||
    if not (job_result.job.user == current_user or current_user.is_administrator):
 | 
			
		||||
def download_job_result(job_id: int, job_result_id: int):
 | 
			
		||||
    job_result = JobResult.query.filter_by(
 | 
			
		||||
        job_id=job_id,
 | 
			
		||||
        id=job_result_id
 | 
			
		||||
    ).first_or_404()
 | 
			
		||||
 | 
			
		||||
    if not (
 | 
			
		||||
        job_result.job.user == current_user
 | 
			
		||||
        or current_user.is_administrator
 | 
			
		||||
    ):
 | 
			
		||||
        abort(403)
 | 
			
		||||
 | 
			
		||||
    return send_from_directory(
 | 
			
		||||
        job_result.path.parent,
 | 
			
		||||
        job_result.path.name,
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user