Add File

backend/apps/system/schemas/system_schema.py

@@ -0,0 +1,185 @@
+from typing import Optional
+from pydantic import BaseModel, Field, field_validator
+
+from common.core.schemas import BaseCreatorDTO
+import re
+
+EMAIL_REGEX = re.compile(
+    r"^[a-zA-Z0-9]+([._-][a-zA-Z0-9]+)*@"
+    r"([a-zA-Z0-9]+(-[a-zA-Z0-9]+)*\.)+"
+    r"[a-zA-Z]{2,}$"
+)
+PWD_REGEX = re.compile(
+    r"^(?=.*[a-z])(?=.*[A-Z])(?=.*\d)"
+    r"(?=.*[~!@#$%^&*()_+\-={}|:\"<>?`\[\];',./])"
+    r"[A-Za-z\d~!@#$%^&*()_+\-={}|:\"<>?`\[\];',./]{8,20}$"
+)
+class UserStatus(BaseCreatorDTO):
+    status: int = 1
+    
+    
+
+class UserLanguage(BaseModel):
+    language: str
+    
+
+class BaseUser(BaseModel):
+    account: str = Field(min_length=1, max_length=100, description="用户账号")
+    oid: int
+    
+class BaseUserDTO(BaseUser, BaseCreatorDTO):
+    language: str = Field(pattern=r"^(zh-CN|en)$", default="zh-CN", description="用户语言")
+    password: str
+    status: int = 1
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "account": self.account,
+            "oid": self.oid
+        }
+    @field_validator("language")
+    def validate_language(cls, lang: str) -> str:
+        if not re.fullmatch(r"^(zh-CN|en)$", lang):
+            raise ValueError("Language must be 'zh-CN' or 'en'")
+        return lang
+
+class UserCreator(BaseUser):
+    name: str = Field(min_length=1, max_length=100, description="用户名")
+    email: str = Field(min_length=1, max_length=100, description="用户邮箱")
+    status: int = 1
+    oid_list: Optional[list[int]] = None 
+    
+    """ @field_validator("email")
+    def validate_email(cls, lang: str) -> str:
+        if not re.fullmatch(EMAIL_REGEX, lang):
+            raise ValueError("Email format is invalid!")
+        return lang """
+
+class UserEditor(UserCreator, BaseCreatorDTO):
+   pass
+
+class UserGrid(UserEditor):
+    create_time: int
+    language: str = "zh-CN"
+    #space_name: Optional[str] = None
+    origin: str = ''
+    
+    
+class PwdEditor(BaseModel):
+    pwd: str
+    new_pwd: str
+    
+class UserWsBase(BaseModel):
+    uid_list: list[int]
+    oid: Optional[int] = None
+class UserWsDTO(UserWsBase):
+    weight: Optional[int] = 0
+
+class UserWsEditor(BaseModel):
+    uid: int
+    oid: int    
+    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
+    description: Optional[str] = None
+class AssistantDTO(AssistantBase, BaseCreatorDTO):
+    pass
+
+class AssistantHeader(AssistantDTO):
+    unique: Optional[str] = None
+    certificate: Optional[str] = None
+    online: bool = False
+    
+
+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
+    
+class UserWsOption(UserWs):
+    account: str
+    
+    
+class AssistantFieldSchema(BaseModel):
+    id: Optional[int] = None
+    name: Optional[str] = None
+    type: Optional[str] = None
+    comment: Optional[str] = None
+class AssistantTableSchema(BaseModel):
+    id: Optional[int] = None
+    name: Optional[str] = None
+    comment: Optional[str] = None
+    rule: Optional[str] = None
+    sql: Optional[str] = None
+    fields: Optional[list[AssistantFieldSchema]] = None
+
+class AssistantOutDsBase(BaseModel):
+    id: Optional[int] = None
+    name: str
+    type: Optional[str] = None
+    type_name: Optional[str] = None
+    comment: Optional[str] = None
+    description: Optional[str] = None
+    
+        
+class AssistantOutDsSchema(AssistantOutDsBase):
+    host: Optional[str] = None
+    port: Optional[int] = None
+    dataBase: Optional[str] = None
+    user: Optional[str] = None
+    password: Optional[str] = None
+    db_schema: Optional[str] = None
+    extraParams: Optional[str] = None
+    tables: Optional[list[AssistantTableSchema]] = None
+    
+class AssistantUiSchema(BaseCreatorDTO):
+    theme: Optional[str] = None
+    header_font_color: Optional[str] = None
+    logo: Optional[str] = None
+    float_icon: Optional[str] = None
+    float_icon_drag: Optional[bool] = False
+    x_type: Optional[str] = 'right'
+    x_val: Optional[int] = 0
+    y_type: Optional[str] = 'bottom'
+    y_val: Optional[int] = 33
+    name: Optional[str] = None
+    welcome: Optional[str] = None
+    welcome_desc: Optional[str] = None
+    
+    
+    
+