mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-04 02:26:11 +08:00
69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
from dataclasses import field, dataclass
|
|
|
|
from nonebot.log import logger
|
|
from nonebot.plugin import require
|
|
from nonebot.adapters.onebot.v11 import MessageSegment
|
|
from nonebot_plugin_saa import Text, Image, MessageSegmentFactory
|
|
|
|
from .abstract_post import BasePost, AbstractPost
|
|
|
|
|
|
@dataclass
|
|
class _CustomPost(BasePost):
|
|
ms_factories: list[MessageSegmentFactory] = field(default_factory=list)
|
|
css_path: str = "" # 模板文件所用css路径
|
|
|
|
async def generate_text_messages(self) -> list[MessageSegmentFactory]:
|
|
return self.ms_factories
|
|
|
|
async def generate_pic_messages(self) -> list[MessageSegmentFactory]:
|
|
require("nonebot_plugin_htmlrender")
|
|
from nonebot_plugin_htmlrender import md_to_pic
|
|
|
|
pic_bytes = await md_to_pic(md=self._generate_md(), css_path=self.css_path)
|
|
return [Image(pic_bytes)]
|
|
|
|
def _generate_md(self) -> str:
|
|
md = ""
|
|
|
|
for message_segment in self.ms_factories:
|
|
match message_segment:
|
|
case Text(data={"text": text}):
|
|
md += f"{text}<br>"
|
|
case Image(data={"image": image}):
|
|
# use onebot v11 to convert image into url
|
|
ob11_image = MessageSegment.image(image)
|
|
md += "\n".format(ob11_image.data["file"])
|
|
case _:
|
|
logger.warning(f"custom_post不支持处理类型:{type(message_segment)}")
|
|
continue
|
|
|
|
return md
|
|
|
|
|
|
@dataclass
|
|
class CustomPost(_CustomPost, AbstractPost):
|
|
"""基于 markdown 语法的，自由度较高的推送内容格式
|
|
|
|
简介:
|
|
支持处理text/image两种MessageSegmentFactory,
|
|
通过将text/image转换成对应的markdown语法以生成markdown文本。
|
|
理论上text类型中可以直接使用markdown语法,例如`##第一章`。
|
|
但会导致不启用`override_use_pic`时, 发送不会被渲染的纯文本消息。
|
|
图片渲染最终由htmlrender执行。
|
|
|
|
注意:
|
|
每一个MessageSegmentFactory元素都会被解释为单独的一行
|
|
|
|
可选参数:
|
|
`override_use_pic`:是否覆盖`bison_use_pic`全局配置
|
|
`compress`:将所有消息压缩为一条进行发送
|
|
`extra_msg`:需要附带发送的额外消息
|
|
|
|
成员函数:
|
|
`generate_text_messages()`:负责生成文本消息
|
|
`generate_pic_messages()`:负责生成图片消息
|
|
"""
|
|
|
|
pass
|