rr/app/util/job.py

"""Scheduler class"""

import os
import subprocess
import shutil
from datetime import datetime

from app import scheduler


def add():
    """Schedule job"""
    job = scheduler.get_job("set")
    if job is None:
        job = scheduler.add_job(
            id="set",
            func=run,
            trigger="date",
        )


def run():
    """Run the job"""
    print("Running")
    working_dir = os.getcwd()
    if os.path.basename(working_dir) != 'rr_tutorial':
        os.chdir('rr_tutorial')

    book_toc_size = os.path.getsize('master_book.toc') \
        if os.path.isfile('master_book.toc') else 0
    article_toc_size = os.path.getsize('master_article.toc') \
        if os.path.isfile('master_article.toc') else 0
    dist_toc_size = os.path.getsize('master_dist.toc') \
        if os.path.isfile('master_dist.toc') else 0

    subprocess.check_output(['xelatex', '-halt-on-error', 'master_book.tex'])
    subprocess.check_output(['xelatex', '-halt-on-error', 'master_article.tex'])
    subprocess.check_output(['./set.sh'])

    if book_toc_size != os.path.getsize('master_book.toc'):
        subprocess.check_output(['xelatex', '-halt-on-error', 'master_book.tex'])

    if article_toc_size != os.path.getsize('master_article.toc'):
        subprocess.check_output(['xelatex', '-halt-on-error', 'master_article.tex'])

    if dist_toc_size != os.path.getsize('master_dist.toc'):
        subprocess.check_output(['./set.sh'])

    shutil.copyfile('master_book.pdf', '../app/static/tutorial/book.pdf')
    shutil.copyfile('master_article.pdf', '../app/static/tutorial/article.pdf')
    shutil.copyfile('master_dist.pdf', '../app/static/tutorial/dist.pdf')

    subprocess.check_output(['./convert.sh'])
    shutil.copyfile('master_dist.png', '../app/static/tutorial/dist.png')


if __name__ == "__main__":
    run.__module__ = "scheduler"