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

from typing import Optional, List

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


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

    parent_id: int = 0
    item_name: str
    item_code: str
    level: int = 1
    sort_order: int = 0
    status: int = 1
    remark: Optional[str] = None


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

    parent_id: Optional[int] = None
    item_name: Optional[str] = None
    item_code: Optional[str] = None
    level: Optional[int] = None
    sort_order: Optional[int] = None
    status: Optional[int] = None
    remark: Optional[str] = None


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

    id: int
    parent_id: int
    item_name: str
    item_code: str
    level: int
    sort_order: int
    status: int
    remark: Optional[str] = None
    child_count: int = 0
    children: Optional[List["ExpenseItemOut"]] = None


class ExpenseItemTreeOut(ExpenseItemOut):
    pass


ExpenseItemOut.model_rebuild()