71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
from enum import Enum
|
|
from typing import List, Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from .base import BaseResponse, SuccessExtra
|
|
|
|
|
|
class MenuType(str, Enum):
|
|
CATALOG = "catalog"
|
|
MENU = "menu"
|
|
|
|
|
|
class BaseMenu(BaseModel):
|
|
id: int
|
|
name: str
|
|
path: str
|
|
remark: Optional[dict]
|
|
menu_type: Optional[MenuType]
|
|
icon: Optional[str]
|
|
order: int
|
|
parent_id: int
|
|
is_hidden: bool
|
|
component: str
|
|
keepalive: bool
|
|
redirect: Optional[str]
|
|
children: Optional[list["BaseMenu"]]
|
|
|
|
|
|
class MenuCreate(BaseModel):
|
|
menu_type: MenuType = Field(default=MenuType.CATALOG.value)
|
|
name: str = Field(example="用户管理")
|
|
remark: Optional[dict] = Field(example={})
|
|
icon: Optional[str] = "ph:user-list-bold"
|
|
path: str = Field(example="/system/user")
|
|
order: Optional[int] = Field(example=1)
|
|
parent_id: Optional[int] = Field(example=0, default=0)
|
|
is_hidden: Optional[bool] = False
|
|
component: str = Field(default="Layout", example="/system/user")
|
|
keepalive: Optional[bool] = True
|
|
redirect: Optional[str] = ""
|
|
|
|
|
|
class MenuUpdate(BaseModel):
|
|
id: int
|
|
menu_type: Optional[MenuType] = Field(example=MenuType.CATALOG.value)
|
|
name: Optional[str] = Field(example="用户管理")
|
|
remark: Optional[dict] = Field(example={})
|
|
icon: Optional[str] = "ph:user-list-bold"
|
|
path: Optional[str] = Field(example="/system/user")
|
|
order: Optional[int] = Field(example=1)
|
|
parent_id: Optional[int] = Field(example=0)
|
|
is_hidden: Optional[bool] = False
|
|
component: str = Field(example="/system/user")
|
|
keepalive: Optional[bool] = False
|
|
redirect: Optional[str] = ""
|
|
|
|
def update_dict(self):
|
|
return self.dict(exclude_unset=True, exclude={"id"})
|
|
|
|
|
|
"""Response"""
|
|
|
|
|
|
class MenuOutList(SuccessExtra):
|
|
data: Optional[List[BaseMenu]]
|
|
|
|
|
|
class MenuOut(BaseResponse):
|
|
data: Optional[BaseMenu]
|