ax-backend-v1/app/schemas/block.py

"""block.py — Block 相关 Schema"""

from typing import Optional

from pydantic import BaseModel, Field


class BlockBase(BaseModel):
    """Block 基础信息"""
    id: str = Field(..., validation_alias="id", serialization_alias="id")
    block_order: int = Field(..., validation_alias="blockOrder", serialization_alias="blockOrder")
    type: str = Field(..., validation_alias="type", serialization_alias="type")
    level: int = Field(..., validation_alias="level", serialization_alias="level")
    index: int = Field(..., validation_alias="index", serialization_alias="index")
    content: str | dict | list = Field(..., validation_alias="content", serialization_alias="content")
    word_style: str = Field(..., validation_alias="wordStyle", serialization_alias="wordStyle")
    style: dict = Field(..., validation_alias="style", serialization_alias="style")
    metadata: dict = Field(..., validation_alias="metadata", serialization_alias="metadata")
    
    class Config:
        populate_by_name = True


class BlockUpdate(BaseModel):
    """Block 更新请求"""
    content: Optional[str | dict | list] = None
    style: Optional[dict] = None
    word_style: Optional[str] = None
    metadata: Optional[dict] = None


class BlockCreate(BaseModel):
    """Block 创建请求"""
    type: str
    level: int = 0
    index: int = 0
    content: str | dict | list
    word_style: str = ""
    style: dict = {}
    metadata: dict = {}
    after_block_id: Optional[str] = None  # 在哪个 block 后插入


class TOCItem(BaseModel):
    """目录树节点"""
    id: str
    level: int
    content: str
    children: list['TOCItem'] = []


class BlockSearchResult(BaseModel):
    """Block 搜索结果"""
    blocks: list[BlockBase]
    total: int


class BlockStats(BaseModel):
    """Block 统计信息"""
    total: int
    by_type: dict[str, int] = Field(..., alias="byType")
    
    class Config:
        populate_by_name = True