from flask import Blueprint, request, Response

from Rating.RatingModel import RatingModel
from common.scripts import verify_token

rating_route = Blueprint('rating', __name__)


@rating_route.route('/general_model', methods=['POST'])
@verify_token
def general_model():
    input_data = request.json['input_data']
    try:
        model = RatingModel(input_data)
        model.save_indicators()
        model.save_result()
        return {"info": "模型执行成功", "result": model.rating_result['信用等级']}, 200
    except Exception:
        return {"info": "模型执行失败"}, 210