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


class TaxViolationObj(SpecObject):
    """行政处罚"""
    cid = ValidateAttr(field='cid', type=str, length=8)
    violation_id = ValidateAttr(field='violation_id', type=str, length=8)
    date = ValidateAttr(field='date', func=Validate.date_format)
    department = ValidateAttr(field='department', type=str, default=None)
    quality = ValidateAttr(field='quality', type=str, default=None)
    illegal_facts = ValidateAttr(field='illegal_facts', type=str, default=None)
    penalties = ValidateAttr(field='penalties', type=str, default=None)
    update_time = ValidateAttr(field='update_time', func=Validate.time_format)

    fields_map = {
        "cid": "企业ID",
        "violation_id": "违法ID",
        "date": "发布日期",
        "department": "税务机关",
        "quality": "案件性质",
        "illegal_facts": "违法事实",
        "penalties": "处罚情况",
        "update_time": "更新时间"
    }