You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.1 KiB
83 lines
2.1 KiB
import re
|
|
from pydantic import BaseModel, ConfigDict, model_validator
|
|
from pydantic.alias_generators import to_camel
|
|
from typing import Optional, List, Union
|
|
from exceptions.exception import ModelValidatorException
|
|
from module_admin.entity.vo.menu_vo import MenuModel
|
|
|
|
|
|
class UserLogin(BaseModel):
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
user_name: str
|
|
password: str
|
|
code: Optional[str] = None
|
|
uuid: Optional[str] = None
|
|
login_info: Optional[dict] = None
|
|
captcha_enabled: Optional[bool] = None
|
|
|
|
|
|
class UserRegister(BaseModel):
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
username: str
|
|
password: str
|
|
confirm_password: str
|
|
code: Optional[str] = None
|
|
uuid: Optional[str] = None
|
|
|
|
@model_validator(mode='after')
|
|
def check_password(self) -> 'UserRegister':
|
|
pattern = r'''^[^<>"'|\\]+$'''
|
|
if self.password is None or re.match(pattern, self.password):
|
|
return self
|
|
else:
|
|
raise ModelValidatorException(message="密码不能包含非法字符：< > \" ' \\ |")
|
|
|
|
|
|
class Token(BaseModel):
|
|
access_token: str
|
|
token_type: str
|
|
|
|
|
|
class CaptchaCode(BaseModel):
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
captcha_enabled: bool
|
|
register_enabled: bool
|
|
img: str
|
|
uuid: str
|
|
|
|
|
|
class SmsCode(BaseModel):
|
|
is_success: Optional[bool] = None
|
|
sms_code: str
|
|
session_id: str
|
|
message: Optional[str] = None
|
|
|
|
|
|
class MenuTreeModel(MenuModel):
|
|
children: Optional[Union[List['MenuTreeModel'], None]] = None
|
|
|
|
|
|
class MetaModel(BaseModel):
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
title: Optional[str] = None
|
|
icon: Optional[str] = None
|
|
no_cache: Optional[bool] = None
|
|
link: Optional[str] = None
|
|
|
|
|
|
class RouterModel(BaseModel):
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
name: Optional[str] = None
|
|
path: Optional[str] = None
|
|
hidden: Optional[bool] = None
|
|
redirect: Optional[str] = None
|
|
component: Optional[str] = None
|
|
query: Optional[str] = None
|
|
always_show: Optional[bool] = None
|
|
meta: Optional[MetaModel] = None
|
|
children: Optional[Union[List['RouterModel'], None]] = None
|
|
|