from flask import Blueprint, request, g
from handlers.base import fail, success

from models.context import Context


inner_bp = Blueprint('inner', __name__)


@inner_bp.get('/data_summary')
def dump_data_summary():
    data_summary_id = request.args.get('data_summary_id', type=int)

    context = Context.get_by_id(g.session, data_summary_id)
    if not context:
        return fail(404, "Context not found")

    result = context.load_data()

    return success(200, "", result)


@inner_bp.put("/data_summary")
def update_data_summary():
    data_summary_id = request.args.get('data_summary_id', type=int)
    data = request.get_json()

    if not data:
        return fail(400, "No data")

    context = Context.get_by_id(g.session, data_summary_id)
    if not context:
        return fail(404, "Context not found")

    context.update_data(data)

    return success(200, "Update success")