from flask import Blueprint, request

from user.user_auth import verify_token
from company.company_obj import CompanyIndex

company_route = Blueprint('company', __name__)


@company_route.route('/head_info', methods=['GET'])
@verify_token
def head_info_route(**kwargs):
    company = CompanyIndex()
    company.cid = kwargs['cid']
    info, result = company.get_head_info()
    return {"info": info, "result": result}, 200


@company_route.route('/company_index', methods=['POST'])
@verify_token
def company_index_route(**kwargs):
    """
    企业主页接口
    Parameters:
        param desc
    Returns:
        res: desc
    """
    company_index = CompanyIndex()
    company_index.cid = kwargs['cid']
    company_index.tab = request.json['table']

    if company_index.tab != '指标明细':
        info, result = company_index.get_company_data_obj()
    elif company_index.tab == '综合评价分析':
        info, result = company_index.get_index_detail_obj()
        result.pop('企业名称')
        result.pop('综合信用等级')
    else:
        info, result = company_index.get_index_detail_obj()
    return {"info": info, "result": result}, 200


@company_route.route('/industry_analysis', methods=['GET'])
@verify_token
def industry_analysis_route(**kwargs):
    """
    宏观行业分析
    """
    ci = CompanyIndex()
    ci.cid = kwargs['cid']
    result = ci.industry_analysis()
    return {"info": "查询成功", "result": result}


@company_route.route('/esg_analysis', methods=['GET'])
@verify_token
def esg_analysis_route(**kwargs):
    """
    ESG分析数据
    Parameters:
        param desc
    Returns:
        res: desc
    """
    ci = CompanyIndex()
    ci.cid = kwargs['cid']
    result = ci.esg_analysis()
    return {"info": "查询成功", "result": result}