update word_bank

plugins/word_bank/message_handle.py

@@ -1,9 +1,10 @@
-from utils.message_builder import image, at
+from utils.message_builder import image, at, face
+from typing import Tuple
 from ._rule import check
 from .model import WordBank
 from configs.path_config import DATA_PATH
 from nonebot.adapters.onebot.v11 import GroupMessageEvent
-from utils.utils import get_message_at, get_message_img
+from utils.utils import get_message_at, get_message_img, change_img_md5
 from nonebot import on_message
 from models.group_member_info import GroupInfoUser
 from utils.utils import get_message_img_file, is_number
@@ -25,14 +26,14 @@ async def _(event: GroupMessageEvent):
         for img_file in list_img:
             strinfo = re.compile(f"{img_file},.*?]")
             msg = strinfo.sub(f'{img_file}]', msg)
-    q = await WordBank.check(
-        event.group_id, msg, event.is_tome()
-    )
+    strinfo_face = re.compile(f",type=sticker]")
+    msg = strinfo_face.sub(f']', msg)
+    q = await WordBank.check(event.group_id, msg, )
     await message_handle.send(await get_one_answer(event, q.format, q.answer))
 
 
 # 处理单条回答
-async def get_one_answer(event, format, _answer, all=1):
+async def get_one_answer(event, format: str, _answer: str, all: bool = True) -> str:
     path = data_dir / f"{event.group_id}"
     placeholder_list = (
         [
@@ -50,12 +51,13 @@ async def get_one_answer(event, format, _answer, all=1):
     else:
         for idx, placeholder in placeholder_list:
             if placeholder.endswith("jpg"):
+                change_img_md5(path / placeholder)
                 answer += _a[: _a.find(f"[__placeholder_{idx}]")] + image(
                     path / placeholder
                 )
             else:
-                if all == 1:
-                    answer += _a[: _a.find(f"[__placeholder_{idx}]")] + at(placeholder)
+                if all:
+                    answer += _a[: _a.find(f"[__placeholder_{idx}]")] + at(int(placeholder))
                 else:
                     q = await GroupInfoUser.get_member_info(
                         int(placeholder), event.group_id)
@@ -65,7 +67,7 @@ async def get_one_answer(event, format, _answer, all=1):
 
 
 # 处理单条问题
-async def get_one_problem(event, problem):
+async def get_one_problem(event, problem: str, ) -> Tuple[str, str]:
     strinfo = re.compile(f",subType=\d")
     problem = strinfo.sub('', problem)
     _problem = problem
@@ -93,7 +95,7 @@ async def get_one_problem(event, problem):
 
 
 # 显示单条数据库问题
-async def get_one_image_problem(event, problem):
+async def get_one_image_problem(event, problem: str) -> str:
     path = data_dir / f"{event.group_id}" / "problem"
     placeholder_list = []
     idx = 0
@@ -127,3 +129,19 @@ async def get_one_image_problem(event, problem):
             _p = _p[_p.find(f"[__placeholder_{idx}]") + len(f"[__placeholder_{idx}]"):]
 
     return problem + _p
+
+
+# 替换cq码
+async def replace_cq(group_id, msg: str, is_face: bool = True) -> str:
+    strinfo_img = re.compile(f"\[CQ:image.*?]")
+    msg = strinfo_img.sub('[图片]', msg)
+    at_list = re.findall(rf"\[CQ:at,qq=(.*?)]", msg)
+    if at_list:
+        for ats in at_list:
+            q = await GroupInfoUser.get_member_info(
+                int(ats), group_id)
+            msg = msg.replace(f'[CQ:at,qq={ats}]', "@" + q.user_name)
+    if is_face:
+        strinfo_face = re.compile(f"\[CQ:face,id=.*?]")
+        msg = strinfo_face.sub('[表情]', msg)
+    return msg