from flask import Blueprint, request

from common.auth import verify_token
from company.CompanyIndex import CompanyIndex

company_route = Blueprint('company', __name__)


@company_route.route('/company_index', methods=['POST'])
@verify_token
def company_page(**kwargs):
    """

    Parameters:
        param desc
    Returns:
        res: desc
    """
    company_index = CompanyIndex()
    company_index.cid = kwargs['cid']
    company_index.table = request.json['table']
    if company_index.table != '指标明细':
        info, result = company_index.get_company_data_obj()
    elif company_index.table == '综合评价分析':
        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('/head_info', methods=['GET'])
@verify_token
def head_info(**kwargs):
    company_index = CompanyIndex()
    company_index.cid = kwargs['cid']
    info, result = company_index.head_info_obj()
    return {"info": info, "result": result}, 200