from flask import Blueprint, request

from Esg.EsgModel import EsgModel
from common.scripts import verify_token

esg_route = Blueprint('esg', __name__)


@esg_route.route('/esg_model', methods=['POST'])
@verify_token
def data_fill():
    input_data = request.json['input_data']
    try:
        model = EsgModel()
        model.prepare_params(input_data)
        model.save_result()
        return {"info": "模型执行成功", "result": model.rating_result['评价等级']}, 200
    except Exception:
        return {"info": "模型执行失败"}, 210