185 lines
4.7 KiB
Python
185 lines
4.7 KiB
Python
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
|
|
|
|
|
|
|
|
|