update 0.0.6.0

utils/manager/withdraw_message_manager.py

@@ -1,8 +1,8 @@
 from typing import Tuple, Union, Dict
+from nonebot.adapters.cqhttp import MessageEvent, PrivateMessageEvent, GroupMessageEvent
 
 
 class WithdrawMessageManager:
-
     def __init__(self):
         self.data = []
 
@@ -14,7 +14,7 @@ class WithdrawMessageManager:
             :param message_data: 撤回消息id和时间
         """
         if isinstance(message_data[0], dict):
-            message_data = (message_data[0]['message_id'], message_data[1])
+            message_data = (message_data[0]["message_id"], message_data[1])
         self.data.append(message_data)
 
     def remove(self, message_data: Tuple[int, int]):
@@ -26,3 +26,19 @@ class WithdrawMessageManager:
         """
         self.data.remove(message_data)
 
+    def withdraw_message(
+        self, event: MessageEvent, id_: Union[int, Dict[str, int]], conditions: Tuple[int, int]
+    ):
+        """
+        便捷判断消息撤回
+        :param event: event
+        :param id_: 消息id 或 send 返回的字典
+        :param conditions: 判断条件
+        """
+        if conditions[0]:
+            if (
+                (conditions[1] == 0 and isinstance(event, PrivateMessageEvent))
+                or (conditions[1] == 1 and isinstance(event, GroupMessageEvent))
+                or conditions[1] == 2
+            ):
+                self.append((id_, conditions[0]))