SQLBot/backend/apps/system/schemas/system_schema.py

from typing import Optional
from pydantic import BaseModel

from common.core.schemas import BaseCreatorDTO

class model_status(BaseModel):
    status: bool
    ids: list[int]
    

class UserLanguage(BaseModel):
    language: str
    

class BaseUser(BaseModel):
    account: str
    oid: int

class BaseUserDTO(BaseUser, BaseCreatorDTO):
    language: str
    password: str
    def to_dict(self):
        return {
            "id": self.id,
            "account": self.account,
            "oid": self.oid
        }

class UserCreator(BaseUser):
    name: str
    email: str
    status: int = 1

class UserEditor(UserCreator, BaseCreatorDTO):
   pass

class UserGrid(UserEditor):
    create_time: int
    language: str = "zh-CN"
    
class PwdEditor(BaseModel):
    pwd: str
    new_pwd: str
    
class UserWsBase(BaseModel):
    uid_list: list[int]
    oid: int
class UserWsDTO(UserWsBase):
    weight: int = 0
    

class UserInfoDTO(UserEditor):
    language: str = "zh-CN"
    weight: int = 0
    isAdmin: bool = False
    

class AssistantBase(BaseModel):
    name: str
    domain: str
    type: int = 0
    configuration: Optional[str] = None
class AssistantDTO(AssistantBase, BaseCreatorDTO):
    pass

class AssistantValidator(BaseModel):
    valid: bool = False
    id_match: bool = False
    domain_match: bool = False
    token: Optional[str] = None
    
    def __init__(
        self,
        valid: bool = False,
        id_match: bool = False,
        domain_match: bool = False,
        token: Optional[str] = None,
        **kwargs
    ):
        super().__init__(
            valid=valid,
            id_match=id_match,
            domain_match=domain_match,
            token=token,
            **kwargs
        )
        
class WorkspaceUser(UserEditor):
    weight: int
    create_time: int
    
class UserWs(BaseCreatorDTO):
    name: str