from flask import Blueprint, request

from esg.esg_obj import EsgForm
from user.user_auth import verify_token
esg_route = Blueprint('esg', __name__)


@esg_route.route('/rating_records', methods=['GET'])
@verify_token
def rating_records(**kwargs):
    """
    获取企业评价记录(esg)
    Parameters:
        cid 企业ID
    Returns:
        info 信息
        result 记录
    """
    inp = EsgForm()
    inp.cid = kwargs['cid']
    result = inp.rating_records_result()
    status = 200 if result else 210
    return {"info": "评价记录", "result": result}, status


@esg_route.route('/start_esg_rating', methods=['GET'])
@verify_token
def start_esg_rating_input(**kwargs):
    """
    开始esg评价
    Parameters:
        cid 企业ID
    Returns:
        info 信息
        result 记录
    """
    input_form = EsgForm()
    input_form.cid = kwargs['cid']
    info, result = input_form.start_esg_rating_obj()
    return {"info": info, "result": result}, 200


@esg_route.route('/save_esg_input_form', methods=['POST'])
@verify_token
def save_esg_input_form(**kwargs):
    """
    保存综合填报数据
    Parameters:
        cid 企业ID
    Returns:
        info: 信息
    """
    esg_form = EsgForm()
    esg_form.cid = kwargs['cid']
    info = esg_form.save_input_general(request.json['input_form'])
    return {"info": info}, 200


@esg_route.route('/submit_esg_rating_input', methods=['GET'])
@verify_token
def submit_general_rating_input(**kwargs):
    """
    提交填报数据，开始esg评价
    Parameters:
        cid 企业ID
    Returns:
        res: desc
    """
    esg_form = EsgForm()
    esg_form.cid = kwargs['cid']
    info, status = esg_form.exec_esg_rating()
    return {"info": info}, status