from Utils.ObjUtil import SpecObject
from Utils.ValidateUtil import ValidateAttr


class BasicInfo(SpecObject):
    """基本工商信息"""

    status = ValidateAttr(field="status", type=str, default=None)
    legal_person = ValidateAttr(field="legal_person", type=str, default=None)
    company_type = ValidateAttr(field="company_type", type=str, default=None)
    taxpayer_id = ValidateAttr(field="taxpayer_id", type=str, default=None)
    business_scope = ValidateAttr(field="business_scope", type=str, default=None)
    registered_capital = ValidateAttr(field="registered_capital", type=str, default=None)
    paid_capital = ValidateAttr(field="paid_capital", type=str, default=None)
    registered_address = ValidateAttr(field="registered_address", type=str, default=None)
    registration_authority = ValidateAttr(field="registration_authority", type=str, default=None)
    industry = ValidateAttr(field="industry", type=str, default=None)
    staff_size = ValidateAttr(field="staff_size", type=str, default=None)
    people_insured_num = ValidateAttr(field="people_insured_num", type=int, default=None)
    micro_company = ValidateAttr(field="micro_company", type=str, default=None)

    fields_map = {
        "status": "企业状态",
        "legal_person": "法定代表人",
        "company_type": "企业类型",
        "taxpayer_id": "纳税人识别号",
        "business_scope": "经营范围",
        "registered_capital": "注册资本",
        "paid_capital": "实缴资本",
        "registered_address": "注册地址",
        "registration_authority": "登记机关",
        "industry": "行业",
        "staff_size": "人员规模",
        "people_insured_num": "参保人数",
        "micro_company": "小微企业"
    }


class ShareHolder(SpecObject):
    """股东信息"""

    name = ValidateAttr(field='name', type=str)
    share_holder_type = ValidateAttr(field='share_holder_type', type=str, default=None)
    share_holding_ratio = ValidateAttr(field="share_holding_ratio", type=str, default=None)
    subscription_amount = ValidateAttr(field="subscription_amount", type=str, default=None)
    subscription_date = ValidateAttr(field="subscription_date", type=str, default=None)
    paid_amount = ValidateAttr(field="paid_amount", type=list, default=[])
    payment_method = ValidateAttr(field="payment_method", type=list, default=[])
    payment_time = ValidateAttr(field="payment_time", type=list, default=[])

    fields_map = {
        "name": "股东",
        "share_holder_type": "股东类型",
        "share_holding_ratio": "持股比例",
        "subscription_amount": "认缴金额",
        "subscription_date": "认缴日期",
        "paid_amount": "实缴金额",
        "payment_method": "实缴方式",
        "payment_time": "实缴时间"
    }


class MainMember(SpecObject):
    """主要成员"""

    name = ValidateAttr(field="name", type=str)
    job_title = ValidateAttr(field="job_title", type=list, default=None)

    fields_map = {
        "name": "姓名",
        "job_title": "职务"
    }


class BalanceSheet(SpecObject):
    """资产负债表"""
    accounts_receivable = ValidateAttr(field='accounts_receivable', type=float)
    stock = ValidateAttr(field='stock', type=float)
    total_current_assets = ValidateAttr(field='total_current_assets', type=float)
    total_assets = ValidateAttr(field='total_assets', type=float)
    short_loan = ValidateAttr(field='short_loan', type=float)
    one_year_liabilities = ValidateAttr(field='one_year_liabilities', type=float)
    total_current_liabilities = ValidateAttr(field='total_current_liabilities', type=float)
    long_term_loan = ValidateAttr(field='long_term_loan', type=float)
    total_liabilities = ValidateAttr(field='total_liabilities', type=float)
    total_owners_equity = ValidateAttr(field='total_owners_equity', type=float)

    fields_map = {
        'accounts_receivable': '应收账款',
        'stock': '存货',
        'total_current_assets': '流动资产合计',
        'total_assets': '资产总计',
        'short_loan': '短期借款',
        'one_year_liabilities': '一年内到期非流动负债',
        'total_current_liabilities': '流动负债合计',
        'long_term_loan': '长期借款',
        'total_liabilities': '负债合计',
        'total_owners_equity': '所有者权益合计'
    }


class ProfitSheet(SpecObject):
    """利润表"""
    operating_income = ValidateAttr(field='operating_income', type=float)
    operating_cost = ValidateAttr(field='operating_cost', type=float)
    total_profit = ValidateAttr(field='total_profit', type=float)
    net_profit = ValidateAttr(field='net_profit', type=float)

    fields_map = {
        'operating_income': '营业收入',
        'operating_cost': '营业成本',
        'total_profit': '利润总额',
        'net_profit': '净利润'
    }


class AppendixDataSheet(SpecObject):
    """补充数据表"""
    rd_expenses = ValidateAttr(field='rd_expenses', type=float)
    interest_disbursement = ValidateAttr(field='interest_disbursement', type=float)
    interest_expense = ValidateAttr(field='interest_expense', type=float)

    fields_map = {
        'rd_expenses': '研发费用异常',
        'interest_disbursement': '计入财务费的利息支出异常',
        'interest_expense': '资本化利息支出异常'
    }


class FinancialIndex(SpecObject):
    """财务指标"""
    roe = ValidateAttr(field="roe", type=float)
    inventory_turnover = ValidateAttr(field="inventory_turnover", type=float)
    interest_multiple = ValidateAttr(field="interest_multiple", type=float)
    accounts_receivable_turnover = ValidateAttr(field="accounts_receivable_turnover", type=float)
    total_asset_turnover = ValidateAttr(field="total_asset_turnover", type=float)
    total_asset_growth_rate = ValidateAttr(field="total_asset_growth_rate", type=float)
    roa = ValidateAttr(field="roa", type=float)
    technology_investment_ratio = ValidateAttr(field="technology_investment_ratio", type=float)
    operating_growth_rate = ValidateAttr(field="operating_growth_rate", type=float)
    assets_and_liabilities = ValidateAttr(field="assets_and_liabilities", type=float)
    quick_ratio = ValidateAttr(field="quick_ratio", type=float)

    fields_map = {
        "roe": "净资产收益率",
        "inventory_turnover": "存货周转率",
        "interest_multiple": "已获利息倍数",
        "accounts_receivable_turnover": "应收账款周转率",
        "total_asset_turnover": "总资产周转率",
        "total_asset_growth_rate": "总资产增长率",
        "roa": "总资产报酬率",
        "technology_investment_ratio": "技术投入比率",
        "operating_growth_rate": "营业增长率",
        "assets_and_liabilities": "资产负债率",
        "quick_ratio": "速动比率"
    }


class CompanyData(SpecObject):
    """企业数据"""

    cid = ValidateAttr(field='cid', type=str, length=8)
    name = ValidateAttr(field='name', type=str)
    industry_l1 = ValidateAttr(field='industry_l1', type=str)
    industry_l2 = ValidateAttr(field='industry_l2', type=str)
    basic_info = ValidateAttr(field='basic_info', type=BasicInfo)
    share_holder = ValidateAttr(field='share_holders', instance_list=ShareHolder)
    main_members = ValidateAttr(field="main_members", instance_list=MainMember)
    balance_sheet = ValidateAttr(field='balance_sheet', type=BalanceSheet)
    profit_sheet = ValidateAttr(field='profit_sheet', type=ProfitSheet)
    # income_sheet = {}
    appendix_sheet = ValidateAttr(field='appendix_sheet', type=AppendixDataSheet)
    fin_index = ValidateAttr(field='fin_index', type=FinancialIndex)
    cc_rating_result = ValidateAttr(field='cc_rating_result', type=dict)
    esg_rating_result = ValidateAttr(field='esg_rating_result', type=dict)
    update_time = ValidateAttr(field='update_time', type=dict)

    fields_map = {
        "cid": "企业ID",
        "name": "企业名称",
        "industry_l1": "一级行业",
        "industry_l2": "二级行业",
        "basic_info": "工商信息",
        "share_holders": "股东信息",
        "main_members": "主要成员",
        "balance_sheet": "资产负债表",
        "profit_sheet": "利润表",
        "income_sheet": "现金流量表",
        "appendix_sheet": "财务补充数据",
        "fin_index": "财务指标",
        "cc_rating_result": "综信评价结果",
        "esg_rating_result": "ESG评价结果",
        "update_time": "更新时间"
    }

    def drag_data_from_tyc_api(self):
        """拉取天眼查接口数据"""

    def prepare_company_init_data(self):
        """准备企业初始化数据"""


class IndexScore(SpecObject):
    """指标得分"""
    environment = ValidateAttr(field='environment', type=(int, float))
    social_responsibility = ValidateAttr(field='social_responsibility', type=(int, float))
    corporate_governance = ValidateAttr(field='corporate_governance', type=(int, float))
    profitability = ValidateAttr(field='profitability', type=(int, float))
    asset_quality = ValidateAttr(field='asset_quality', type=(int, float))
    debt_risk = ValidateAttr(field='debt_risk', type=(int, float))
    operational_growth = ValidateAttr(field='operational_growth', type=(int, float))
    compliance_risk = ValidateAttr(field='compliance_risk', type=(int, float))
    cperational_risk = ValidateAttr(field='cperational_risk', type=(int, float))
    associated_risk = ValidateAttr(field='associated_risk', type=(int, float))

    fields_map = {
        "environment": "环境",
        "social_responsibility": "社会责任",
        "corporate_governance": "公司治理",
        "profitability": "盈利能力",
        "asset_quality": "资产质量",
        "debt_risk": "债务风险",
        "operational_growth": "经营增长",
        "compliance_risk": "合规风险",
        "cperational_risk": "经营风险",
        "associated_risk": "关联风险"
    }


class MaxScore(SpecObject):
    """最大分数"""
    environment = ValidateAttr(field='environment', type=int)
    social_responsibility = ValidateAttr(field='social_responsibility', type=int)
    corporate_governance = ValidateAttr(field='corporate_governance', type=int)
    profitability = ValidateAttr(field='profitability', type=int)
    asset_quality = ValidateAttr(field='asset_quality', type=int)
    debt_risk = ValidateAttr(field='debt_risk', type=int)
    operational_growth = ValidateAttr(field='operational_growth', type=int)
    compliance_risk = ValidateAttr(field='compliance_risk', type=int)
    cperational_risk = ValidateAttr(field='cperational_risk', type=int)
    associated_risk = ValidateAttr(field='associated_risk', type=int)

    fields_map = {
        "environment": "环境",
        "social_responsibility": "社会责任",
        "corporate_governance": "公司治理",
        "profitability": "盈利能力",
        "asset_quality": "资产质量",
        "debt_risk": "债务风险",
        "operational_growth": "经营增长",
        "compliance_risk": "合规风险",
        "cperational_risk": "经营风险",
        "associated_risk": "关联风险"
    }


class IndicatorRadar(SpecObject):
    """指标雷达"""
    index_score = ValidateAttr(field='index_score', type=IndexScore)
    max_score = ValidateAttr(field='max_score', type=MaxScore)

    fields_map = {
        "index_score": "指标得分",
        "max_score": "最大分数"
    }


class CreditPerformance(SpecObject):
    """综合信用表现"""
    indicator_radar = ValidateAttr(field='indicator_radar', type=IndicatorRadar)

    fields_map = {
        "indicator_radar": "指标雷达"
    }


class CreditAnalysis(SpecObject):
    """信用分析"""
    management_analysis = ValidateAttr(field='management_analysis', type=str)
    finance_analysis = ValidateAttr(field='finance_analysis', type=str)
    risk_analysis = ValidateAttr(field='risk_analysis', type=str)
    evaluation_opinion = ValidateAttr(field='evaluation_opinion', type=str)
    view_report = ValidateAttr(field='view_report', type=str)

    fields_map = {
        "management_analysis": "经营分析",
        "finance_analysis": "财务分析",
        "risk_analysis": "风险分析",
        "evaluation_opinion": "评价意见",
        "view_report": "查看报告"
    }


class FinancialIndicators(SpecObject):
    """财务指标"""
    profitability = ValidateAttr(field='financial_indicators', in_list=['高', '中', '低', '差'])
    asset_quality = ValidateAttr(field='asset_quality', in_list=['高', '中', '低', '差'])
    debt_risk = ValidateAttr(field='debt_risk', in_list=['高', '中', '低', '差'])
    operating_growth = ValidateAttr(field='operating_growth', in_list=['高', '中', '低', '差'])

    fields_map = {
        "profitability": "盈利能力",
        "asset_quality": "资产质量",
        "debt_risk": "债务风险",
        "operating_growth": "经营增长"
    }


class RiskIndicators(SpecObject):
    """风险指标"""
    compliance_risk = ValidateAttr(field='compliance_risk', in_list=['高', '中', '低', '差'])
    operational_risk = ValidateAttr(field='operational_risk', in_list=['高', '中', '低', '差'])
    associated_risk = ValidateAttr(field='associated_risk', in_list=['高', '中', '低', '差'])

    fields_map = {
        "compliance_risk": "合规风险",
        "operational_risk": "经营风险",
        "associated_risk": "关联风险"
    }


class OperatingIndicators(SpecObject):
    """经营指标"""
    environmental = ValidateAttr(field='environmental', in_list=['高', '中', '低', '差'])
    social_responsibility = ValidateAttr(field='social_responsibility', in_list=['高', '中', '低', '差'])
    corporate_governance = ValidateAttr(field='corporate_governance', in_list=['高', '中', '低', '差'])

    fields_map = {
        "environmental": "环境",
        "social_responsibility": "社会责任",
        "corporate_governance": "公司治理"
    }


class IndexTable(SpecObject):
    """指标表格"""
    financial_indicators = ValidateAttr(field='financial_indicators', type=FinancialIndicators)
    risk_indicators = ValidateAttr(field='risk_indicators', type=RiskIndicators)
    operating_indicators = ValidateAttr(field='operating_indicators', type=OperatingIndicators)

    fields_map = {
        "financial_indicators": "财务指标",
        "risk_indicators": "风险指标",
        "operating_indicators": "经营指标"
    }


class HistoryLevel(SpecObject):
    """历史级别"""
    credit_rating = ValidateAttr(field='credit_rating', type=str)
    credit_score = ValidateAttr(field='credit_score', type=(float, int))
    evaluation_time = ValidateAttr(field='evaluation_time', type=str)
    evaluation_method = ValidateAttr(field='evaluation_method', type=str, in_list=['企业申报'])
    change_trend = ValidateAttr(field='change_trend', type=str)

    fields_map = {
        "credit_rating": "信用等级",
        "credit_score": "信用评分",
        "evaluation_time": "评价时间",
        "evaluation_method": "评价方式",
        "change_trend": "变化趋势"
    }


class ComprehensiveCreditAnalysis(SpecObject):
    """综合信用分析"""
    cid = ValidateAttr(field='cid', type=str, length=8)
    rid = ValidateAttr(field='rid', type=str, length=8)
    name = ValidateAttr(field='name', type=str)
    update_time = ValidateAttr(field='update_time', type=str)
    performance = ValidateAttr(field='performance', type=CreditPerformance)
    credit_analysis = ValidateAttr(field='credit_analysis', type=CreditAnalysis)
    index_table = ValidateAttr(field='index_table', type=IndexTable)
    history_level = ValidateAttr(field='history_level', instance_list=HistoryLevel)

    fields_map = {
        "cid": "企业ID",
        "rid": "评价ID",
        "name": "企业名称",
        "update_time": "更新时间",
        "performance": "综合信用表现",
        "analysis": "信用分析",
        "index_table": "指标表格",
        "history_level": "历史级别"
    }