refactor code

plugins/nonebot_plugin_withdraw/__init__.py

@@ -5,6 +5,7 @@ from nonebot.rule import to_me
 from nonebot.typing import T_State, T_CalledAPIHook
 
 from .config import Config
+
 global_config = get_driver().config
 withdraw_config = Config(**global_config.dict())
 
@@ -12,33 +13,34 @@ msg_ids = {}
 max_size = withdraw_config.withdraw_max_size
 
 
-__plugin_name__ = '撤回'
+__plugin_name__ = "撤回"
 
-__plugin_usage__ = '用法：撤回 [消息位置](默认0)\n' \
-            '示例：\n' \
-            '\t撤回0  -> 撤回倒数第一条消息(即最新发送的消息)' \
-            '\t撤回1  -> 撤回倒数第2条消息'
+__plugin_usage__ = (
+    "用法：撤回 [消息位置](默认0)\n" "示例：\n" "\t撤回0  -> 撤回倒数第一条消息(即最新发送的消息)" "\t撤回1  -> 撤回倒数第2条消息"
+)
 
 
 def get_key(msg_type, id):
-    return f'{msg_type}_{id}'
+    return f"{msg_type}_{id}"
 
 
-async def save_msg_id(bot: Bot, e: Exception, api: str, data: Dict[str, Any], result: Any) -> T_CalledAPIHook:
+async def save_msg_id(
+    bot: Bot, e: Exception, api: str, data: Dict[str, Any], result: Any
+) -> T_CalledAPIHook:
     try:
-        if api == 'send_msg':
-            msg_type = data['message_type']
-            id = data['group_id'] if msg_type == 'group' else data['user_id']
-        elif api == 'send_private_msg':
-            msg_type = 'private'
-            id = data['user_id']
-        elif api == 'send_group_msg':
-            msg_type = 'group'
-            id = data['group_id']
+        if api == "send_msg":
+            msg_type = data["message_type"]
+            id = data["group_id"] if msg_type == "group" else data["user_id"]
+        elif api == "send_private_msg":
+            msg_type = "private"
+            id = data["user_id"]
+        elif api == "send_group_msg":
+            msg_type = "group"
+            id = data["group_id"]
         else:
             return
         key = get_key(msg_type, id)
-        msg_id = result['message_id']
+        msg_id = result["message_id"]
 
         if key not in msg_ids:
             msg_ids[key] = []
@@ -52,16 +54,16 @@ async def save_msg_id(bot: Bot, e: Exception, api: str, data: Dict[str, Any], re
 Bot._called_api_hook.add(save_msg_id)
 
 
-withdraw = on_command('withdraw', aliases={'撤回'}, rule=to_me(), priority=1, block=True)
+withdraw = on_command("withdraw", aliases={"撤回"}, rule=to_me(), priority=1, block=True)
 
 
 @withdraw.handle()
 async def _(bot: Bot, event: Event, state: T_State):
     if isinstance(event, GroupMessageEvent):
-        msg_type = 'group'
+        msg_type = "group"
         id = event.group_id
     elif isinstance(event, PrivateMessageEvent):
-        msg_type = 'private'
+        msg_type = "private"
         id = event.user_id
     else:
         return
@@ -80,4 +82,4 @@ async def _(bot: Bot, event: Event, state: T_State):
         await bot.delete_msg(message_id=msg_ids[key][idx])
         msg_ids[key].pop(idx)
     except:
-        await withdraw.finish('撤回失败，可能已超时')
+        await withdraw.finish("撤回失败，可能已超时")