mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 21:52:56 +08:00
60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
|
from typing import Literal
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"BaseCell",
|
||
|
|
"ImageCell",
|
||
|
|
"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"
|
||
|
|
|
||
|
|
|
||
|
|
TableCell = TextCell | ImageCell | StatusBadgeCell | str | int | float | None
|
||
|
|
|
||
|
|
|
||
|
|
class TableData(BaseModel):
|
||
|
|
"""通用表格的数据模型"""
|
||
|
|
|
||
|
|
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="数据行列表")
|