from flask import Blueprint, request

from common.scripts import verify_token
from company.scripts import *

company_route = Blueprint('company', __name__)


@company_route.route('/init_company_data', methods=['POST'])
@verify_token
def basic_info():
    """
    初始化企业数据
    包括 天眼查数据库、企业基本信息
    Parameters:
        cid 企业ID
        company_name 企业名称
    Returns:
        res: desc
    """
    cid = json.loads(request.data)['cid']
    company_name = json.loads(request.data)['company_name']

    # 拉取企业数据
    info1 = drag_company_data_request(company_name)
    if info1 != "企业数据拉取成功":
        return {"info": "数据拉取失败"}, 210

    # 清洗基本信息数据
    basic_info_etl(cid, company_name)

    return {"info": "企业基本信息初始化成功"}


@company_route.route('/company_index_etl', methods=['POST'])
@verify_token
def general_rating():
    """
    1.准备综合评价信息数据
    2.准备财务要素信息
    3.准备风险要素信息
    4.准备指标明细数据
    Parameters:
        rid str 评价ID
        cid str 企业ID
    Returns:
        -
    """
    rid = request.json['rid']

    general_rating_etl(rid)
    financial_analysis(rid)
    risk_analysis_etl(rid)

    return {"info": "数据准备完成"}