from flask import Blueprint, request
from common.security.APIAuth import api_secret
from rating.Rating import Rating

rating_route = Blueprint('rating', __name__)


@rating_route.route('/search', methods=['POST'])
@api_secret
def search_rating():
    """
    企业评价记录搜索
    Parameters:
        -
    Returns:
        info: 返回信息
        result: 返回数据
    """
    rating = Rating()
    rating.query = request.json['query']
    data = rating.search_rating()
    return {"info": '查询成功', "result": data}, 200


@rating_route.route('/rating_result', methods=['POST'])
@api_secret
def rating_result():
    """
    查询评价结果
    Parameters:
        -
    Returns:
        info: 返回信息
        result: 返回数据
    """
    try:
        rating = Rating()
        rating.rid = request.json['rid']
        data = rating.search_rating_result()
        return {"info": '查询成功', "result": data}, 200
    except KeyError:
        return {"info": "键值错误"}, 400
    except TypeError:
        return {"info": "参数错误"}, 400


@rating_route.route('/details', methods=['POST'])
@api_secret
def fill_details():
    """
    查看填报详情信息
    Parameters:
        -
    Returns:
        info: 返回信息
        result: 返回数据
    """
    try:
        rating = Rating()
        rating.rid = request.json['rid']
        data = rating.search_fill_details()
        return {"info": '查询成功', "result": data}, 200
    except KeyError:
        return {"info": "键值错误"}, 400
    except TypeError:
        return {"info": "参数错误"}, 400


@rating_route.route('/risk', methods=['POST'])
@api_secret
def risk_info():
    """
    查看风险数据信息
    Parameters:
        -
    Returns:
        info: 返回信息
        result: 返回数据
    """
    try:
        rating = Rating()
        rating.rid = request.json['rid']
        data = rating.search_risk_info()
        return {"info": '查询成功', "result": data}, 200
    except KeyError:
        return {"info": "键值错误"}, 400
    except TypeError:
        return {"info": "参数错误"}, 400


@rating_route.route('/report_edit', methods=['POST'])
@api_secret
def report_edit():
    """
    编辑报告数据
    Parameters:
        -
    Returns:
        info: 编辑结果
    """
    try:
        rating = Rating()
        rating.rid = request.json['rid']
        text_model = request.json['text_model']
        info = rating.report_edit(text_model)
        status = 200 if info else 210
        res = '编辑成功' if info else '编辑失败'
        return {"info": res}, status
    except KeyError:
        return {"info": "键值错误"}, 400
    except TypeError:
        return {"info": "参数错误"}, 400