mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
|
from utils.utils import get_message_text
|
||
|
|
from utils.message_builder import image, at
|
||
|
|
from .rule import check
|
||
|
|
from .model import WordBank
|
||
|
|
from configs.path_config import DATA_PATH
|
||
|
|
from pathlib import Path
|
||
|
|
from nonebot.adapters.cqhttp import (
|
||
|
|
Bot,
|
||
|
|
GroupMessageEvent,
|
||
|
|
)
|
||
|
|
from nonebot.typing import T_State
|
||
|
|
from nonebot import on_message
|
||
|
|
|
||
|
|
|
||
|
|
__zx_plugin_name__ = "词库问答回复操作 [Hidden]"
|
||
|
|
|
||
|
|
|
||
|
|
data_dir = Path(DATA_PATH) / "word_bank"
|
||
|
|
data_dir.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
|
||
|
|
message_handle = on_message(priority=7, block=True, rule=check)
|
||
|
|
|
||
|
|
|
||
|
|
@message_handle.handle()
|
||
|
|
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||
|
|
path = data_dir / f"{event.group_id}"
|
||
|
|
q = await WordBank.check(event.group_id, get_message_text(event.json()))
|
||
|
|
placeholder_list = [
|
||
|
|
(x.split("<_s>")[0], x.split("<_s>")[1])
|
||
|
|
for x in q.format.split("<format>")[:-1]
|
||
|
|
] if q.format else []
|
||
|
|
answer = ""
|
||
|
|
_a = q.answer
|
||
|
|
if not placeholder_list:
|
||
|
|
answer = _a
|
||
|
|
else:
|
||
|
|
for idx, placeholder in placeholder_list:
|
||
|
|
if placeholder.endswith("jpg"):
|
||
|
|
answer += _a[:_a.find(f"[__placeholder_{idx}]")] + image(path / placeholder)
|
||
|
|
else:
|
||
|
|
answer += _a[:_a.find(f"[__placeholder_{idx}]")] + at(placeholder)
|
||
|
|
_a = _a[_a.find(f"[__placeholder_{idx}]") + len(f"[__placeholder_{idx}]"):]
|
||
|
|
await message_handle.send(answer)
|