PoseCorrection/backend_refactor/route/analyze.py

import uuid, json

from service.video_analyzer import BaseVideoAnalyzer, CrunchVideoAnalyzer, HighKneesVideoAnalyzer, PlankVideoAnalyzer, SquatVideoAnalyzer, StandingLongJumpVideoAnalyzer
# from service.video.Video3DAnalyzer import Video3DAnalyzer

from db import db
from flask import request
from db.task import Task
from worker.thread_pool import pool
from route.util import make_response
from flask import Blueprint
from service.photo_analyzer import StandingPhotoAnalyzer

import logging

logging.basicConfig(format='%(asctime)s - %(name)s[line:%(lineno)d] - %(levelname)s: %(message)s',
                    level=logging.DEBUG)


bp = Blueprint("analyze", __name__, url_prefix='/')

def callback_generator(task_uuid):
    def callback(status, **kwargs):
        try:
            from app import app
            with app.app_context():
                task = Task.query.get(task_uuid)
                task.status = status
                if status == 'FINISHED':
                    task.result = json.dumps(kwargs['result'])
                if status == 'ERROR':
                    logging.error("error analyzing: ", exc_info=kwargs['error'])
                
                db.session.commit()
                logging.debug("calling to " + status)
        except Exception as e:
            logging.debug("nmsl", exc_info=e)
    return callback


@bp.route("analysis", methods=['POST'])
def analysis():
    req = request.get_json()

    types = {
        "PHOTO_STANDING": StandingPhotoAnalyzer,
        "VIDEO_HIGHKNEES": HighKneesVideoAnalyzer,
        "VIDEO_CRUNCH": CrunchVideoAnalyzer,
        "VIDEO_STANDINGLONGJUMP": StandingLongJumpVideoAnalyzer,
        "VIDEO_PLANK": PlankVideoAnalyzer,
        "VIDEO_SQUAT": SquatVideoAnalyzer
    }


    ana_type = types.get(req.get('type'))

    if ana_type is None:
        return make_response(400, "invalid type", {})

    try:
        ana = ana_type(req.get('data'))
    except Exception as e:
        return make_response(400, str(e), {})

    

    with __import__("app").app.app_context():
        task_uuid = str(uuid.uuid4())
        new_task = Task(task_uuid, req.get('type'))
        db.session.add(new_task)
        db.session.commit()

    logging.debug("task created.")

    pool.submit(ana.analyze, callback_generator(task_uuid))

    logging.debug("task submited")

    return make_response(201, "success", {'task_uuid': task_uuid})