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


class ShareholderList(SpecObject):
    """股东列表"""
    sid = ValidateAttr(field='sid', type=str, default=None)
    name = ValidateAttr(field='name', type=str)
    stype = ValidateAttr(field='stype', type=str)
    quality = ValidateAttr(field='quality', type=str, default=None)
    proportion = ValidateAttr(field='proportion', type=str)
    amount = ValidateAttr(field='amount', type=[str, float, int])
    date = ValidateAttr(field='date', fun=Validate.date_format)

    fields_map = {
        "sid": "股东ID",
        "name": "股东名称",
        "stype": "股东类型",
        "quality": "股东性质",
        "proportion": "持股比例",
        "amount": "认缴出资额",
        "date": "认缴出资日期"
    }


class ShareholderObj(SpecObject):
    """股东信息"""
    cid = ValidateAttr(field='cid', type=str, length=8)
    shareholder_list = ValidateAttr(field='shareholder_list', instance_list=ShareholderList)
    update_time = ValidateAttr(field='update_time', func=Validate.time_format)

    fields_map = {
        "cid": "企业ID",
        "shareholder_list": "股东列表",
        "update_time": "更新时间"
    }