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

from pathlib import Path
from typing import Any

from pydantic import Field

from .base import RenderableComponent

__all__ = ["TemplateComponent"]


class TemplateComponent(RenderableComponent):
    """基于独立模板文件的UI组件"""

    _is_standalone_template: bool = True
    """标记此组件为独立模板"""
    template_path: str | Path = Field(..., description="指向HTML模板文件的路径")
    """指向HTML模板文件的路径"""
    data: dict[str, Any] = Field(..., description="传递给模板的上下文数据字典")
    """传递给模板的上下文数据字典"""

    @property
    def template_name(self) -> str:
        """返回模板路径"""
        if isinstance(self.template_path, Path):
            return self.template_path.as_posix()
        return str(self.template_path)

    def get_render_data(self) -> dict[str, Any]:
        """返回传递给模板的数据"""
        return self.data

    def __getattr__(self, name: str) -> Any:
        """允许直接访问 `data` 字典中的属性。"""
        try:
            return self.data[name]
        except KeyError:
            raise AttributeError(
                f"'{type(self).__name__}' 对象没有属性 '{name}'"
            ) from None