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


class CreditProcess(SpecObject):
    """综合信用评价流程数据"""
    fill_in_questionnaire = ValidateAttr(field='fill_in_questionnaire', type=int, in_list=[0, 1, -1])
    fill_in_financial_data = ValidateAttr(field='fill_in_financial_data', type=int, in_list=[0, 1, -1])
    clean_risk_data = ValidateAttr(field='clean_risk_data', type=int, in_list=[0, 1, -1])
    model_scoring = ValidateAttr(field='model_scoring', type=int, in_list=[0, 1, -1])
    report_generation = ValidateAttr(field='report_generation', type=int, in_list=[0, 1, -1])
    certificate_generation = ValidateAttr(field='certificate_generation', type=int, in_list=[0, 1, -1])
    cc_evaluation_homepage = ValidateAttr(field='cc_evaluation_homepage', type=int, in_list=[0, 1, -1])
    fe_evaluation_homepage = ValidateAttr(field='fe_evaluation_homepage', type=int, in_list=[0, 1, -1])
    cr_analysis_homepage = ValidateAttr(field='cr_analysis_homepage', type=int, in_list=[0, 1, -1])
    am_details_homepage = ValidateAttr(field='am_details_homepage', type=int, in_list=[0, 1, -1])

    fields_map = {
        "fill_in_questionnaire": "问卷填报",
        "fill_in_financial_data": "财务数据填报",
        "clean_risk_data": "清洗风险数据",
        "model_scoring": "模型打分",
        "report_generation": "报告生成",
        "certificate_generation": "证书生成",
        "cc_evaluation_homepage": "综合信用评价主页",
        "fe_evaluation_homepage": "财务要素评价主页",
        "cr_analysis_homepage": "公司风险分析主页",
        "am_details_homepage": "分析指标明细主页"
    }


class EsgProcess(SpecObject):
    """ESG评价流程"""
    data_filling = ValidateAttr(field='data_filling', type=int, in_list=[0, 1, -1])
    clean_risk_data = ValidateAttr(field='clean_risk_data', type=int, in_list=[0, 1, -1])
    model_scoring = ValidateAttr(field='model_scoring', type=int, in_list=[0, 1, -1])
    report_generation = ValidateAttr(field='report_generation', type=int, in_list=[0, 1, -1])
    certificate_generation = ValidateAttr(field='certificate_generation', type=int, in_list=[0, 1, -1])
    esg_data_homepage = ValidateAttr(field='esg_data_homepage', type=int, in_list=[0, 1, -1])

    fields_map = {
        "data_filling": "数据填报",
        "clean_risk_data": "清洗风险数据",
        "model_scoring": "模型打分",
        "report_generation": "报告生成",
        "certificate_generation": "证书生成",
        "esg_data_homepage": "主页ESG数据生成"
    }