RC-CHN
b31b520c7c
* feat:添加t2i模板管理后端api,移除config.py中重复功能 * feat: 添加T2I模板管理功能前端，支持模板的创建、应用和重置 * refactor: 修复错误的保存逻辑，将t2i注册时打印路由信息部分移到基类实现 * remove:移除了路由注册时的打印 * chore: format code * fix: update input variant from solo to outlined for better UI consistency --------- Co-authored-by: Soulter <905617992@qq.com>
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
from .star import StarMetadata, star_map, star_registry
|
|
from .star_manager import PluginManager
|
|
from .context import Context
|
|
from astrbot.core.provider import Provider
|
|
from astrbot.core.utils.command_parser import CommandParserMixin
|
|
from astrbot.core import html_renderer
|
|
from astrbot.core.star.star_tools import StarTools
|
|
|
|
|
|
class Star(CommandParserMixin):
|
|
"""所有插件（Star）的父类，所有插件都应该继承于这个类"""
|
|
|
|
def __init__(self, context: Context, config: dict | None = None):
|
|
StarTools.initialize(context)
|
|
self.context = context
|
|
|
|
def __init_subclass__(cls, **kwargs):
|
|
super().__init_subclass__(**kwargs)
|
|
if not star_map.get(cls.__module__):
|
|
metadata = StarMetadata(
|
|
star_cls_type=cls,
|
|
module_path=cls.__module__,
|
|
)
|
|
star_map[cls.__module__] = metadata
|
|
star_registry.append(metadata)
|
|
else:
|
|
star_map[cls.__module__].star_cls_type = cls
|
|
star_map[cls.__module__].module_path = cls.__module__
|
|
|
|
async def text_to_image(self, text: str, return_url=True) -> str:
|
|
"""将文本转换为图片"""
|
|
return await html_renderer.render_t2i(
|
|
text,
|
|
return_url=return_url,
|
|
template_name=self.context._config.get("t2i_active_template"),
|
|
)
|
|
|
|
async def html_render(
|
|
self, tmpl: str, data: dict, return_url=True, options: dict | None = None
|
|
) -> str:
|
|
"""渲染 HTML"""
|
|
return await html_renderer.render_custom_template(
|
|
tmpl, data, return_url=return_url, options=options
|
|
)
|
|
|
|
async def initialize(self):
|
|
"""当插件被激活时会调用这个方法"""
|
|
pass
|
|
|
|
async def terminate(self):
|
|
"""当插件被禁用、重载插件时会调用这个方法"""
|
|
pass
|
|
|
|
def __del__(self):
|
|
"""[Deprecated] 当插件被禁用、重载插件时会调用这个方法"""
|
|
pass
|
|
|
|
|
|
__all__ = ["Star", "StarMetadata", "PluginManager", "Context", "Provider", "StarTools"]
|