refactor code

plugins/poke/__init__.py

@@ -3,21 +3,34 @@ from nonebot.adapters.cqhttp import Bot, PokeNotifyEvent
 from nonebot.typing import T_State
 from configs.path_config import VOICE_PATH, IMAGE_PATH
 import os
-from utils.init_result import record, image, poke
+from utils.message_builder import record, image, poke
 from services.log import logger
 import random
 from utils.utils import CountLimiter
 from models.ban_user import BanUser
 
-__plugin_name__ = '戳一戳 [Hidden]'
+__plugin_name__ = "戳一戳 [Hidden]"
 
-__plugin_usage__ = '用法：无'
+__plugin_usage__ = "用法：无"
 
 poke__reply = [
-    "lsp你再戳？", "连个可爱美少女都要戳的肥宅真恶心啊。",
-    "你再戳！", "？再戳试试？", "别戳了别戳了再戳就坏了555", "我爪巴爪巴，球球别再戳了", "你戳你🐎呢？！",
-    "那...那里...那里不能戳...绝对...", "(。´・ω・)ん?", "有事恁叫我，别天天一个劲戳戳戳！", "欸很烦欸！你戳🔨呢",
-    "?", "再戳一下试试？", "???", "正在关闭对您的所有服务...关闭成功", "啊呜，太舒服刚刚竟然睡着了。什么事？", "正在定位您的真实地址。。。\r\n定位成功。轰炸机已起飞"
+    "lsp你再戳？",
+    "连个可爱美少女都要戳的肥宅真恶心啊。",
+    "你再戳！",
+    "？再戳试试？",
+    "别戳了别戳了再戳就坏了555",
+    "我爪巴爪巴，球球别再戳了",
+    "你戳你🐎呢？！",
+    "那...那里...那里不能戳...绝对...",
+    "(。´・ω・)ん?",
+    "有事恁叫我，别天天一个劲戳戳戳！",
+    "欸很烦欸！你戳🔨呢",
+    "?",
+    "再戳一下试试？",
+    "???",
+    "正在关闭对您的所有服务...关闭成功",
+    "啊呜，太舒服刚刚竟然睡着了。什么事？",
+    "正在定位您的真实地址...定位成功。轰炸机已起飞",
 ]
 
 
@@ -31,24 +44,25 @@ async def _poke_(bot: Bot, event: PokeNotifyEvent, state: T_State):
     if event.self_id == event.target_id:
         _clmt.add(event.user_id)
         if _clmt.check(event.user_id) or random.random() < 0.3:
-            rst = ''
+            rst = ""
             if random.random() < 0.15:
                 await BanUser.ban(event.user_id, 1, 60)
-                rst = '气死我了！'
+                rst = "气死我了！"
             await poke_.finish(rst + random.choice(poke__reply), at_sender=True)
         rand = random.random()
         if rand <= 0.3:
-            path = random.choice(['luoli/', 'meitu/'])
+            path = random.choice(["luoli/", "meitu/"])
             index = random.randint(0, len(os.listdir(IMAGE_PATH + path)))
-            result = f'id：{index}' + image(f'{index}.jpg', path)
+            result = f"id：{index}" + image(f"{index}.jpg", path)
             await poke_.send(result)
-            logger.info(f'USER {event.user_id} 戳了戳我 回复: {result} \n {result}')
+            logger.info(f"USER {event.user_id} 戳了戳我 回复: {result} \n {result}")
         elif 0.3 < rand < 0.6:
-            voice = random.choice(os.listdir(VOICE_PATH + 'dinggong/'))
+            voice = random.choice(os.listdir(VOICE_PATH + "dinggong/"))
             result = record(voice, "dinggong")
             await poke_.send(result)
-            await poke_.send(voice.split('_')[1])
-            logger.info(f'USER {event.user_id} 戳了戳我 回复: {result} \n {voice.split("_")[1]}')
+            await poke_.send(voice.split("_")[1])
+            logger.info(
+                f'USER {event.user_id} 戳了戳我 回复: {result} \n {voice.split("_")[1]}'
+            )
         else:
             await poke_.send(poke(event.user_id))
-