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


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


class VecsetModel(BaseModel):
    """
    智能语句配置表对应pydantic模型
    """
    onum: Optional[int]
    dasset_id: Optional[int]
    stab_name: Optional[str]
    squery: Optional[str]
    sanal_plan: Optional[str]
    sintnt_term: Optional[str]
    ssql: Optional[str]  
    sim_thrsh: Optional[str]  
    status: Optional[str]
    del_flag: Optional[str]
    create_by: Optional[str]
    create_time: Optional[str]
    update_by: Optional[str]
    update_time: Optional[str]
    remark: Optional[str]

    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 CurrentVecsetInfo(BaseModel):
    """
    数据库返回当前智能语句配置信息
    """
    vecset_basic_info: Union[VecsetModel, None]
    vecset_dasset_info: Union[DassetModel, None]

class VecsetDetailModel(BaseModel):
    """
    获取智能语句配置详情信息响应模型
    """
    vecset: Union[VecsetModel, None]
    dasset: Union[DassetModel, None]


class CurrentVecsetInfoServiceResponse(VecsetDetailModel):
    """
    获取当前智能语句配置信息响应模型
    """
    menu: Union[List, None]


class VecsetQueryModel(VecsetModel):
    """
    智能语句配置管理不分页查询模型
    """
    create_time_start: Optional[str]
    create_time_end: Optional[str]


class VecsetPageObject(VecsetQueryModel):
    """
    智能语句配置管理分页查询模型
    """
    page_num: int
    page_size: int


class VecsetInfoJoinDasset(VecsetModel):
    """
    数据库查询智能语句配置列表返回模型
    """
    dasset_name: Optional[str]


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


class AddVecsetModel(VecsetModel):
    """
    新增智能语句配置模型
    """
    type: Optional[str]




class DeleteVecsetModel(BaseModel):
    """
    删除智能语句配置模型
    """
    onums: str
    #onums: Optional[str]
    update_by: Optional[str]
    update_time: Optional[str]

class ImportVecsetModel(BaseModel):
    """
    批量导入智能语句配置模型
    """
    url: str
    is_update: bool


class CrudVecsetResponse(BaseModel):
    """
    操作智能语句配置响应模型
    """
    is_success: bool
    message: str


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

class ImportVecsetModel(BaseModel):
    """
    批量导入智能语句配置模型
    """
    url: str
    is_update: bool