clear_private_chat 通过指定关键词清除私聊

clear_private_chat/main.py

@@ -0,0 +1,51 @@
+from asyncio import sleep
+
+from pyrogram.enums import ChatType
+from pyrogram.errors import FloodWait
+from pyrogram.raw.functions.messages import DeleteHistory
+
+from pagermaid.listener import listener
+from pagermaid.enums import Message
+from pagermaid.services import bot
+
+
+async def delete_private_chat(cid: int):
+    try:
+        await bot.invoke(DeleteHistory(
+            just_clear=False,
+            revoke=False,
+            peer=await bot.resolve_peer(cid),
+            max_id=0, )
+        )
+    except FloodWait as e:
+        await sleep(e.value)
+        await delete_private_chat(cid)
+    except Exception:
+        pass
+
+
+@listener(command="clear_private_chat",
+          need_admin=True,
+          description="通过指定关键词清除私聊",
+          parameters="<关键词>")
+async def clear_private_chat(message: Message):
+    """ 通过指定关键词清除私聊记录 """
+    if not message.arguments:
+        await message.edit("请输入关键词来清除指定的私聊对话，为了保证速度，仅匹配每个对话的最后一条消息")
+        return
+    count = 0
+    keywords = message.arguments.split(" ")
+    message: Message = await message.edit("清除私聊对话中。。。")
+    async for dialog in bot.get_dialogs():
+        if dialog.chat.type != ChatType.PRIVATE:
+            continue
+        if dialog.top_message is None:
+            continue
+        if dialog.top_message.text is None:
+            continue
+        for i in keywords:
+            if i in dialog.top_message.text:
+                await delete_private_chat(dialog.chat.id)
+                count += 1
+                break
+    await message.edit(f"成功清除了 {count} 个私聊对话")