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


class QuestionObject(SpecObject):
    """问题对象"""
    qid = ValidateAttr(field='qid', type=str, length=8)
    types = ValidateAttr(field='types', type=str, in_list=['单选题', '多选题', '填空题'])
    stem = ValidateAttr(field='stem', type=str)
    option = ValidateAttr(field='option', type=[dict, list])

    fields_map = {
        "qid": "问题ID",
        "types": "类型",
        "stem": "题干",
        "option": "选项"
    }


class QuestionnaireObject(SpecObject):
    """问卷对象"""
    qid = ValidateAttr(field='qid', type=str, length=8)
    industry = ValidateAttr(field='industry', type=str)
    content = ValidateAttr(field='content', type=list)

    fields_map = {
        "qid": "问卷ID",
        "industry": "行业",
        "content": "问卷内容"
    }


class CreditQuestionnarie(SpecObject):
    """综信评价问卷"""
    industry = ValidateAttr(field='industry', type=str)
    qid = ValidateAttr(field='qid', type=str, length=8)
    content = ValidateAttr(field='content', type=list)

    fields_map = {
        "industry": "行业",
        "qid": "问卷ID",
        "content": "问卷内容"
    }