from pydantic import BaseModel
from typing import Union, Optional, List


class TokenData(BaseModel):
    """
    token解析结果
    """
    onum: Union[int, None] = None


from pydantic import BaseModel, Field
from typing import Optional
from datetime import datetime

class SscfModel(BaseModel):
    """
    短句配置表对应pydantic模型
    """
    onum: Optional[int]
    dasset_id: Optional[str]
    keywords: Optional[str]
    keyword: Optional[str]
    algorithm: Optional[str]
    order: Optional[str]
    whole_sentence: Optional[str]
    type: Optional[str]
    supp_expl: Optional[str]
    bak1: Optional[str]
    bak2: Optional[str]
    bak3: Optional[str]
    bak4: Optional[str]
    bak5: Optional[str]
    status: Optional[str]
    del_flag: Optional[str]
    update_time: Optional[datetime]
    update_by: Optional[str]
    create_time: Optional[datetime]
    create_by: Optional[str]

    # 配置类，用于开启ORM模式
    class Config:
        orm_mode = True





class DassetModel(BaseModel):
    """
    数据资产表对应pydantic模型
    """
    dasset_id: Optional[int]
    dasset_parent_id: Optional[int]
    dasset_ancestors: Optional[str]
    dasset_name: Optional[str]
    dasset_remark: Optional[str]
    dasset_order_num: Optional[int]
    status: Optional[str]
    del_flag: Optional[str]
    dasset_area: Optional[str]
    create_by: Optional[str]
    create_time: Optional[str]
    update_by: Optional[str]
    update_time: Optional[str]

    class Config:
        orm_mode = True



class CurrentSscfInfo(BaseModel):
    """
    数据库返回当前短句配置信息
    """
    sscf_basic_info: Union[SscfModel, None]
    sscf_dasset_info: Union[DassetModel, None]

class SscfDetailModel(BaseModel):
    """
    获取短句配置详情信息响应模型
    """
    sscf: Union[SscfModel, None]
    dasset: Union[DassetModel, None]


class CurrentSscfInfoServiceResponse(SscfDetailModel):
    """
    获取当前短句配置信息响应模型
    """
    menu: Union[List, None]


class SscfQueryModel(SscfModel):
    """
    短句配置管理不分页查询模型
    """
    create_time_start: Optional[str]
    create_time_end: Optional[str]


class SscfPageObject(SscfQueryModel):
    """
    短句配置管理分页查询模型
    """
    page_num: int
    page_size: int


class SscfInfoJoinDasset(SscfModel):
    """
    数据库查询短句配置列表返回模型
    """
    dasset_name: Optional[str]


class SscfPageObjectResponse(BaseModel):
    """
    短句配置管理列表分页查询返回模型
    """
    rows: List[Union[SscfInfoJoinDasset, None]] = []
    page_num: int
    page_size: int
    total: int
    has_next: bool


class AddSscfModel(SscfModel):
    """
    新增短句配置模型
    """
    type: Optional[str]




class DeleteSscfModel(BaseModel):
    """
    删除短句配置模型
    """
    onums: str
    #onums: Optional[str]
    update_by: Optional[str]
    update_time: Optional[str]

class ImportSscfModel(BaseModel):
    """
    批量导入短句配置模型
    """
    url: str
    is_update: bool


class CrudSscfResponse(BaseModel):
    """
    操作短句配置响应模型
    """
    is_success: bool
    message: str


class DassetInfo(BaseModel):
    """
    查询数据资产树
    """
    dasset_id: int
    dasset_name: str
    dasset_ancestors: str

class ImportSscfModel(BaseModel):
    """
    批量导入短句配置模型
    """
    url: str
    is_update: bool