from flask import Blueprint, request

from Modules.Models.SModels.SModelImpl import ScoreModelImpl
from Utils.ErrorUtil import APIReturnError
from Utils.RouteUtil import RouteParamsCheck

score_model_route = Blueprint('score_model', __name__)


@score_model_route.route('/search_score_model', methods=['GET'])
def search_score_model_route():
    """搜索打分模型"""
    try:
        RouteParamsCheck(request.args, ["model_name", "page_no", "page_size"]).required()
        model_name = request.args['model_name']
        page_size = request.args['page_size']
        page_no = request.args['page_no']
        ipml = ScoreModelImpl()
        result = ipml.search_score_model(model_name=model_name, page_size=page_size, page_no=page_no)
        return {"info": "打分模型搜索", "result": result}, 200
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@score_model_route.route('/new_score_model', methods=['POST'])
def new_score_model_route():
    """构建打分模型"""
    try:
        RouteParamsCheck(request.json, ["model_name", "dimension", "author", "tags"]).required()
        data = request.json
        ipml = ScoreModelImpl()
        result = ipml.new_score_model(data=data)
        return {"info": "构建打分模型", "result": result}, 200
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@score_model_route.route('check_score_model', methods=['POST'])
def check_score_model_route():
    """校验打分模型"""
    try:
        RouteParamsCheck(request.json, ["model_name", "dimension", "author", "tags"]).required()
        data = request.json
        ipml = ScoreModelImpl()
        result = ipml.check_score_model(data=data)
        return {"info": "校验打分模型", "result": result}, 200
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code