from flask import Blueprint, request
from common.utils import verify_token
from RatingData.EsgReportEtl import EsgCleanModel
from RatingData.ReportEtl import CleanModel
from RatingData.scripts.common import get_tfse_data, get_esg_data
from RatingData.scripts.risk_count import *

rating_route = Blueprint('rating', __name__)


@rating_route.route('/risk_data', methods=['POST'])
@verify_token
def risk_data():
    """
    风险数据清洗
    Parameters:
        company_name 企业名称
        company_id 企业ID
        evaluation_id 评价ID
    Returns:
        res: desc
    """
    company_name = request.json['company_name']
    company_id = request.json['company_id']
    evaluation_id = request.json['evaluation_id']
    risk_data_etl(company_name, company_id, evaluation_id)
    return {"info": "执行成功"}, 200


@rating_route.route('/esg_risk_data', methods=['POST'])
@verify_token
def esg_risk_data():
    """
    风险数据清洗
    Parameters:
        company_name 企业名称
        company_id 企业ID
        evaluation_id 评价ID
    Returns:
        res: 清洗后的满足要求的风险数据
    """
    company_name = request.json['company_name']
    company_id = request.json['company_id']
    evaluation_id = request.json['evaluation_id']
    esg_risk_data_etl(company_name, company_id, evaluation_id)
    return {"info": "执行完成"}, 200


@rating_route.route('/report_data', methods=['POST'])
@verify_token
def report_data():
    """
    清洗报告数据
    Parameters:
        rid str 评价ID
        cid str 企业ID
        company str 企业名称
    Returns:
        -
    """
    info = request.json

    # 准备需要清洗的数据
    db_data = get_tfse_data(info)

    # 调用清洗数据模型
    data = CleanModel(info, db_data).storage_data()

    return {"data": data}, 200


@rating_route.route('/esg_report_data', methods=['POST'])
@verify_token
def esg_report_data():
    """
    清洗ESG报告数据
    Parameters:
        rid str 评价ID
        cid str 企业ID
        company str 企业名称
    Returns:
        -
    """
    info = request.json

    # 准备需要清洗的数据
    db_data = get_esg_data(info)

    # 调用清洗数据模型
    data = EsgCleanModel(db_data).storage_data()

    return {"data": data}, 200