zhenxun_bot/zhenxun/ui/models/core/table.py

from typing import Literal

from pydantic import BaseModel, Field

from ...models.components.progress_bar import ProgressBar
from .base import RenderableComponent
from .text import TextSpan

__all__ = [
    "BaseCell",
    "ComponentCell",
    "ImageCell",
    "ProgressBarCell",
    "RichTextCell",
    "StatusBadgeCell",
    "TableCell",
    "TableData",
    "TextCell",
]


class BaseCell(BaseModel):
    """单元格基础模型"""

    type: str


class TextCell(BaseCell):
    """文本单元格"""

    type: Literal["text"] = "text"  # type: ignore
    content: str
    bold: bool = False
    color: str | None = None


class ImageCell(BaseCell):
    """图片单元格"""

    type: Literal["image"] = "image"  # type: ignore
    src: str
    width: int = 40
    height: int = 40
    shape: Literal["square", "circle"] = "square"
    alt: str = "image"


class StatusBadgeCell(BaseCell):
    """状态徽章单元格"""

    type: Literal["badge"] = "badge"  # type: ignore
    text: str
    status_type: Literal["ok", "error", "warning", "info"] = "info"


class ProgressBarCell(BaseCell, ProgressBar):
    """进度条单元格，继承ProgressBar模型以复用其字段"""

    type: Literal["progress_bar"] = "progress_bar"  # type: ignore


class RichTextCell(BaseCell):
    """富文本单元格，支持多个带样式的文本片段"""

    type: Literal["rich_text"] = "rich_text"  # type: ignore
    spans: list[TextSpan] = Field(default_factory=list, description="文本片段列表")
    """文本片段列表"""
    direction: Literal["column", "row"] = Field("column", description="片段排列方向")
    """片段排列方向"""
    gap: str = Field("4px", description="片段之间的间距")
    """片段之间的间距"""


class ComponentCell(BaseCell):
    """一个通用的单元格，可以容纳任何可渲染的组件。"""

    type: str = "component"
    component: RenderableComponent


TableCell = (
    TextCell
    | ImageCell
    | StatusBadgeCell
    | ProgressBarCell
    | RichTextCell
    | ComponentCell
    | str
    | int
    | float
    | None
)


class TableData(RenderableComponent):
    """通用表格的数据模型"""

    style_name: str | None = None
    """应用于表格容器的样式名称"""
    title: str = Field(..., description="表格主标题")
    """表格主标题"""
    tip: str | None = Field(None, description="表格下方的提示信息")
    """表格下方的提示信息"""
    headers: list[str] = Field(default_factory=list, description="表头列表")
    """表头列表"""
    rows: list[list[TableCell]] = Field(default_factory=list, description="数据行列表")
    """数据行列表"""
    column_alignments: list[Literal["left", "center", "right"]] | None = Field(
        default=None, description="每列的对齐方式"
    )
    """每列的对齐方式"""
    column_widths: list[str | int] | None = Field(
        default=None, description="每列的宽度 (e.g., ['50px', 'auto', 100])"
    )
    """每列的宽度 (e.g., ['50px', 'auto', 100])"""

    @property
    def template_name(self) -> str:
        return "components/core/table"