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()
        model.prepare_params(input_data)
        model.save_indicators()
        model.save_result()
        return {"info": "模型执行成功", "result": model.rating_result['信用等级']}, 200
    except Exception:
        return {"info": "模型执行失败"}, 210


@rating_route.route('/financial_score', methods=['POST'])
@verify_token
def financial_score():
    input_data = request.json
    try:
        model = RatingModel()
        model.industry = input_data['所属行业']
        model.financial_indicators = input_data['财务指标']
        result = model.calculate_financial_elements_score()
        return {"info": "模型执行成功", "result": result}, 200
    except Exception:
        return {"info": "模型执行失败"}, 210