Soulter
b8e4068c75
* feat: support key-value storage for plugins * fix: remove unnecessary initialization method from Main class
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from astrbot.core import html_renderer
|
|
from astrbot.core.provider import Provider
|
|
from astrbot.core.star.star_tools import StarTools
|
|
from astrbot.core.utils.command_parser import CommandParserMixin
|
|
from astrbot.core.utils.plugin_kv_store import PluginKVStoreMixin
|
|
|
|
from .context import Context
|
|
from .star import StarMetadata, star_map, star_registry
|
|
from .star_manager import PluginManager
|
|
|
|
|
|
class Star(CommandParserMixin, PluginKVStoreMixin):
|
|
"""所有插件（Star）的父类，所有插件都应该继承于这个类"""
|
|
|
|
author: str
|
|
name: str
|
|
|
|
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):
|
|
"""当插件被激活时会调用这个方法"""
|
|
|
|
async def terminate(self):
|
|
"""当插件被禁用、重载插件时会调用这个方法"""
|
|
|
|
def __del__(self):
|
|
"""[Deprecated] 当插件被禁用、重载插件时会调用这个方法"""
|
|
|
|
|
|
__all__ = ["Context", "PluginManager", "Provider", "Star", "StarMetadata", "StarTools"]
|