from company.company_db import FIND_SINGLE_DATA_BY_REQ_WITH_SORT, FIND_SINGLE_DATA_BY_REQ
from company.company_utils import judge_industry


def get_head_info_impl(cid):
    # company_info: 企业信息
    company_info = FIND_SINGLE_DATA_BY_REQ("应用端", "企业用户", {"企业ID": cid}, ["企业名称", "已认证"])

    # cc_rating: 最近一期综信评价结果
    cc_rating = FIND_SINGLE_DATA_BY_REQ_WITH_SORT("综信评价数据", "评价结果", {"企业ID": cid}, ["信用评分", "信用等级", "评价时间"], {"评价年度": -1})
    # esg_rating: 最近一期ESG评价结果
    esg_rating = FIND_SINGLE_DATA_BY_REQ_WITH_SORT("ESG评价数据", "评价结果", {"企业ID": cid}, ["ESG得分", "评价等级", "评价时间"], {"评价年度": -1})

    # head_info: 公司主要头部信息
    head_info = {
        "企业名称": company_info['企业名称'],
        "企业认证": "已认证" if company_info['已认证'] == "是" else "未认证",
        "综合信用评价": cc_rating if cc_rating else {"信用等级": "N/A", "信用评分": "N/A", "评价时间": "N/A"},
        "ESG评价": esg_rating if esg_rating else {"信用等级": "N/A", "信用评分": "N/A", "评价时间": "N/A"}
    }

    return head_info


def get_basic_info_impl(cid):
    # data: 企业基本工商信息
    data = FIND_SINGLE_DATA_BY_REQ_WITH_SORT("企业数据", "基本工商信息", {"企业ID": cid}, ["更新时间", "企业名称", "工商信息", "股东信息", "主要成员"], {"更新时间": -1})
    return data


def get_cc_rating_impl(cid):
    # data: 综合信用分析数据
    data = FIND_SINGLE_DATA_BY_REQ_WITH_SORT("企业数据", "综合信用分析", {"企业ID": cid}, ["更新时间", "综合信用表现", "信用分析", "指标表格", "历史级别"], {"更新时间": -1})
    return data


def get_industry_analysis_impl(cid):
    """
    行业宏观分析数据
    Parameters:
        cid: str 企业ID
    Returns:
        Case Success:
            data: dict 行业宏观分析数据
        Case Fail:
            data: dict 空字典
    """

    industry = judge_industry(cid)

    if industry is False:
        return {}

    data = FIND_SINGLE_DATA_BY_REQ("行业数据", "行业分析", {"行业": industry}, ["行业", "行业简介", "更新时间", "正面因素", "负面因素", "增长趋势", "规模分布"])

    if data is False:
        return {}

    return data


def get_esg_rating_impl(cid):
    """
    ESG要素分析数据
    cid: str 企业ID
    Returns:
    Case Success:
        data: dict ESG分析数据
    Case Fail:
        data: dict 空字典
    """
    data = FIND_SINGLE_DATA_BY_REQ_WITH_SORT('企业数据', 'ESG评价分析', {"企业ID": cid}, ["更新日期", "ESG评级", "维度得分情况", "实质性议题得分情况", "实质性议题"], {"更新日期": -1})

    if data is False:
        return {}

    return data


def get_financial_elements_impl(cid):
    """
    财务要素分析
    """
    data = FIND_SINGLE_DATA_BY_REQ_WITH_SORT('企业数据', '财务要素分析', {"企业ID": cid}, ["更新日期", "财报期", "同行对比图", "去年对比图", "去年对比表", "同行对比表", "指标详情"], {"更新日期": -1})
    return data


def get_operating_risk_impl(cid):
    """
    经营风险分析
    """
    data = FIND_SINGLE_DATA_BY_REQ_WITH_SORT('企业数据', '经营风险分析', {"企业ID": cid}, ["更新日期", "风险分数", "风险级别", "列入失信名单", "合规风险", "经营风险", "周边风险", "变更记录"], {"更新日期": -1})
    return data