refactor code

plugins/send_dinggong_voice/__init__.py

@@ -1,21 +1,21 @@
 from nonebot import on_keyword
-from utils.init_result import record
+from utils.message_builder import record
 from configs.path_config import VOICE_PATH
 import random
 from services.log import logger
 from utils.utils import FreqLimiter
 from nonebot.typing import T_State
-from nonebot.adapters.cqhttp import Bot, MessageEvent
+from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
 from nonebot.rule import to_me
 import os
 
-__plugin_name__ = '骂我'
+__plugin_name__ = "骂我"
 __plugin_usage__ = '对我说 "骂我"，我真的会骂你哦'
 
 _flmt = FreqLimiter(3)
 
 
-dg_voice = on_keyword({'骂'}, rule=to_me(), priority=5, block=True)
+dg_voice = on_keyword({"骂"}, rule=to_me(), priority=5, block=True)
 
 
 @dg_voice.handle()
@@ -23,11 +23,13 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
     if len(str((event.get_message()))) == 1:
         return
     if not _flmt.check(event.user_id):
-        await dg_voice.finish('就...就算求我骂你也得慢慢来...', at_sender=True)
+        await dg_voice.finish("就...就算求我骂你也得慢慢来...", at_sender=True)
     _flmt.start_cd(event.user_id)
-    voice = random.choice(os.listdir(VOICE_PATH + 'dinggong/'))
+    voice = random.choice(os.listdir(VOICE_PATH + "dinggong/"))
     result = record(voice, "dinggong")
     await dg_voice.send(result)
-    await dg_voice.send(voice.split('_')[1])
+    await dg_voice.send(voice.split("_")[1])
     logger.info(
-        f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'}) 发送钉宫骂人:" + result)
+        f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'}) 发送钉宫骂人:"
+        + result
+    )