TriM-Organization/LiteyukiBot-TriM
2
0
Fork 0
mirror of https://github.com/TriM-Organization/LiteyukiBot-TriM.git synced 2024-11-28 16:24:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 添加了对markdown的简单封装

Browse Source
This commit is contained in:
远野千束 2024-03-19 22:43:55 +08:00
parent 0bf56f79f1
commit ab5dc2200a
1 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/utils/message.py
View File

@ -2,11 +2,12 @@ import nonebot
from nonebot.adapters.onebot import v11, v12 from nonebot.adapters.onebot import v11, v12
from typing_extensions import Any from typing_extensions import Any
from .tools import de_escape
from .typing import T_Bot from .typing import T_Bot
async def send_markdown(markdown: str, bot: T_Bot, message_type: str, session_id: str) -> dict[str, Any]: async def send_markdown(markdown: str, bot: T_Bot, message_type: str, session_id: str) -> dict[str, Any]:
markdown = markdown.replace("\n", r"\n").replace("\"", r'\\\"') formatted_md = de_escape(markdown).replace("\n", r"\n").replace("\"", r'\\\"')
forward_data = await bot.call_api( forward_data = await bot.call_api(
api="send_private_forward_msg", api="send_private_forward_msg",
user_id=bot.self_id, user_id=bot.self_id,
@ -20,7 +21,7 @@ async def send_markdown(markdown: str, bot: T_Bot, message_type: str, session_id
{ {
"type": "markdown", "type": "markdown",
"data": { "data": {
"content": '{"content":"%s"}' % markdown "content": '{"content":"%s"}' % formatted_md
} }
} }
] ]
@ -29,18 +30,28 @@ async def send_markdown(markdown: str, bot: T_Bot, message_type: str, session_id
] ]
) )
data = await bot.send_msg( try:
message_type=message_type, data = await bot.send_msg(
message=[ message_type=message_type,
v11.MessageSegment( message=[
type="longmsg", v11.MessageSegment(
data={ type="longmsg",
"id": forward_data["forward_id"] data={
} "id": forward_data["forward_id"]
) }
], ),
user_id=session_id if message_type == "private" else None, ],
group_id=session_id if message_type == "group" else None
)
nonebot.logger.info("已发送md：%s" % forward_data["forward_id"]) )
except Exception as e:
nonebot.logger.warning("send_markdown error, send as plane text: %s", e)
data = await bot.send_msg(
message_type=message_type,
message=markdown,
user_id=session_id if message_type == "private" else None,
group_id=session_id if message_type == "group" else None
)
return data return data