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


class RatingRecord(SpecObject):
    """评价记录 """
    rid = ValidateAttr(field='rid', type=str, length=8)
    cid = ValidateAttr(field='cid', type=str, length=8)
    company = ValidateAttr(field='company', type=str)
    project = ValidateAttr(field='project', type=str, in_list=["综合信用评价", "ESG评价"])
    methods = ValidateAttr(field='methods', type=str, in_list=["企业申报"])
    status = ValidateAttr(field='status', type=str, in_list=["进行", "完成"])
    result = ValidateAttr(field='result', type=str,
                          in_list=["AAA", "AA+", "AA", "AA-", "A+", "A", "A-", "BBB+", "BBB", "BBB-", "BB+", "BB",
                                   "BB-", "B+", "B", "B-", "CCC", "CC", None], default=None)
    report_fid = ValidateAttr(field='report_fid', type=str, default=None)
    certificate_fid = ValidateAttr(field='certificate_fid', type=str, default=None)
    begin_time = ValidateAttr(field='begin_time', type=str)
    submit_time = ValidateAttr(field='submit_time', type=str, default=None)
    finish_time = ValidateAttr(field='finish_time', type=str, default=None)
    finish_date = ValidateAttr(field='finish_date', type=str, default=None)

    fields_map = {
        "rid": "评价ID",
        "cid": "企业ID",
        "company": "企业名称",
        "project": "评价项目",
        "methods": "评价方式",
        "status": "进行状态",
        "result": "评价结果",
        "report_fid": "报告fid",
        "certificate_fid": "证书fid",
        "begin_time": "开始填报时间",
        "submit_time": "提交填报时间",
        "finish_time": "评价完成时间",
        "finish_date": "评价完成日期"
    }