from flask import Blueprint, request
from CompanyUser.CompanyUserAuth import verify_token, verify_real_company
from Rating.Esg.EsgImpl import EsgRating
from Utils.ErrorUtil import ReturnConditionCheckFailed

esg_route = Blueprint('esg', __name__)


@esg_route.route('/start_esg_rating', methods=['GET'])
@verify_token
def start_esg_rating_input(**kwargs):
    """
    开始esg评价
    Parameters:
        cid 企业ID
    Returns:
        info 信息
        result 记录
    """
    try:
        esg = EsgRating()
        esg.cid = kwargs['cid']
        info, result = esg.start_esg_rating_obj()
        return {"info": info, "result": result}, 200
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.__str__()}, e.status_code
    except KeyError:
        return {"info": "参数错误"}, 400


@esg_route.route('/save_esg_input_form', methods=['POST'])
@verify_token
def save_esg_input_form(**kwargs):
    """
    保存综合填报数据
    Parameters:
        cid 企业ID
    Returns:
        info: 信息
    """
    try:
        esg = EsgRating()
        esg.cid = kwargs['cid']
        esg.save_input_general(request.json['input_form'])
        return {"info": '保存成功'}, 200
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.__str__()}, e.status_code
    except KeyError:
        return {"info": "参数错误"}, 400


@esg_route.route('/submit_esg_rating_input', methods=['GET'])
@verify_token
def submit_general_rating_input(**kwargs):
    """
    提交填报数据，开始esg评价
    Parameters:
        cid 企业ID
    Returns:
        res: desc
    """
    try:
        esg = EsgRating()
        esg.cid = kwargs['cid']
        info, status = esg.exec_esg_rating()
        return {"info": info}, status
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.__str__()}, e.status_code
    except KeyError:
        return {"info": "参数错误"}, 400