nopaque/nopaque.py

# First things first: apply monkey patch, so that no code gets executed without
# patched libraries!
import eventlet
eventlet.monkey_patch()

from app import db, cli, create_app  # noqa
from app.models import (
    Corpus,
    CorpusFile,
    Job,
    JobInput,
    JobResult,
    Permission,
    Role,
    TesseractOCRModel,
    TranskribusHTRModel,
    User
)  # noqa
from flask import Flask  # noqa
from typing import Any, Dict  # noqa


app: Flask = create_app()
cli.register(app)


@app.context_processor
def make_context() -> Dict[str, Any]:
    ''' Adds variables to the template context. '''
    return {'Permission': Permission}


@app.shell_context_processor
def make_shell_context() -> Dict[str, Any]:
    ''' Adds variables to the shell context. '''
    return {
        'Corpus': Corpus,
        'CorpusFile': CorpusFile,
        'db': db,
        'Job': Job,
        'JobInput': JobInput,
        'JobResult': JobResult,
        'Permission': Permission,
        'Role': Role,
        'TesseractOCRModel': TesseractOCRModel,
        'TranskribusHTRModel': TranskribusHTRModel,
        'User': User
    }