from flask import render_template
from . import scheduler
from flask_login import current_user, login_required
from .. import background_scheduler
import json
import subprocess


@scheduler.route('/')
@login_required
def scheduler():
    job = {
        'creator': current_user.id,
        'files': ['file_1', 'file_2', 'file_3'],
        'service': 'ocr',
        'service_args': {
            'lang': 'deu'
        },
        'status': 'queued'
    }
    job = background_scheduler.add_job(process_job, trigger='date', args=[job])
    print('### New job added for scheduling ###')
    print(job)
    return render_template('scheduler/index.html.j2', title='Scheduler')


def process_job(job):
<<<<<<< HEAD
    input_dir = '/home/pjentsch/%s/input' % job['service']
    output_dir = '/home/pjentsch/%s/output' % job['service']
=======
    input_dir = '/home/pjentsch/ocr/input'
    output_dir = '/home/pjentsch/ocr/output'
>>>>>>> e6446834b4c4251154497a7104cd84d1188be981

    job['status'] = 'running'
    print('### Job status changed ###')
    print(job)

    subprocess.run([job['service'], '-i', input_dir, '-l', job['service_args']['lang'], '-o', output_dir, '--keep-intermediates'])

    job['status'] = 'finished'
    print('### Job status changed ###')
    print(job)