from flask import Blueprint, request
from common.scripts import verify_token
from rating.ReportEtl import CleanModel
from rating.scripts.common import get_tfse_data
from rating.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('/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