from rating.scripts.report import *


class CleanModel:
    """
    清洗报告数据
    Parameters:
        param1: 企业基本信息（包括：企业名称、企业ID、评价ID）
        param2: 数据库数据
    Returns:
        report_data: 清洗后满足条件的报告数据
    """
    def __init__(self, param1, param2):
        # 企业名称
        self.company = param1['企业名称']
        # 企业ID
        self.cid = param1['企业ID']
        # 评价ID
        self.rid = param1['评价ID']
        # 数据库数据
        self.data = param2

    def abstract(self):
        # 摘要模块
        data = dict()
        data['评价结果'] = evaluation_results(self.data)
        data['主要财务数据'] = financial_data(self.data)
        data['评价观点'] = evaluation_view(self.company, self.data)
        data['关注'] = attention(self.data)
        return data

    def company_profile(self):
        # 公司概况
        data = dict()
        data['公司背景'] = company_background(self.company, self.data)
        data['股东实力'] = shareholder_strength(self.data)
        data['对外投资'] = outbound_investment(self.data)
        data['进出口权'] = import_export(self.data)
        data['知识产权'] = intellectual_property(self.data)
        data['资质荣誉'] = qualification_honor(self.data)
        data['融资信息'] = financing_information(self.data)
        return data

    def esg_evaluate(self):
        # ESG评价
        data = dict()
        data['环境治理'] = environmental_governance(self.data)
        data['社会责任'] = social_responsibility(self.data)
        data['公司治理'] = corporate_governance(self.data)
        return data

    def financial_analysis(self):
        # 财务分析
        data = dict()
        data['盈利能力'] = profitability(self.data)
        data['资产质量'] = asset_quality(self.data)
        data['债务风险'] = debt_risk(self.data)
        data['成长能力'] = growth_ability(self.data)
        return data

    def enterprise_risk(self):
        # 企业风险
        data = dict()
        data['合规风险'] = compliance_risk(self.data)
        data['经营风险'] = operational_ris(self.data)
        data['关联风险'] = associated_risk(self.data)
        return data

    def rating_conclusion(self):
        # 评级结论
        data = dict()
        data['结论'] = conclusion(self.data)
        return data

    def appendix(self):
        # 附录
        data = dict()
        data['附录一'] = appendix_one(self.data)
        return data

    def integrate_data(self):
        data = dict()
        data['摘要'] = None
        data['一、公司概况'] = self.abstract()
        data['二、ESG评价'] = self.esg_evaluate()
        data['三、财务分析'] = self.financial_analysis()
        data['四、企业风险'] = self.enterprise_risk()
        data['五、评级结论'] = self.rating_conclusion()
        data['附录'] = self.appendix()

        return data


if __name__ == '__main__':
    info = {
        "企业名称": "远东资信评估有限公司",
        "企业ID": "mKJreZRw",
        "评价ID": "kSVoCeJ1"
    }
    db_data = get_tfse_data(info)
    r = CleanModel(info, db_data).integrate_data()
    print(r)