shanxia/bx/bx-backend/app/schemas/menu.py

from typing import Optional, List

from pydantic import BaseModel, ConfigDict
from pydantic.alias_generators import to_camel


class MenuCreate(BaseModel):
    model_config = ConfigDict(populate_by_name=True, alias_generator=to_camel)

    parent_id: int = 0
    menu_name: str
    menu_code: Optional[str] = None
    menu_type: int
    path: Optional[str] = None
    sort_order: int = 0
    status: int = 1


class MenuUpdate(BaseModel):
    model_config = ConfigDict(populate_by_name=True, alias_generator=to_camel)

    parent_id: Optional[int] = None
    menu_name: Optional[str] = None
    menu_code: Optional[str] = None
    menu_type: Optional[int] = None
    path: Optional[str] = None
    sort_order: Optional[int] = None
    status: Optional[int] = None


class MenuOut(BaseModel):
    model_config = ConfigDict(from_attributes=True, populate_by_name=True, alias_generator=to_camel)

    id: int
    parent_id: int
    menu_name: str
    menu_code: Optional[str] = None
    menu_type: int
    path: Optional[str] = None
    sort_order: int
    status: int
    children: Optional[List["MenuOut"]] = None


class MenuTreeOut(MenuOut):
    pass


MenuOut.model_rebuild()