67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
|
import asyncio
|
||
|
|
import time
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from core import command
|
||
|
|
from loguru import logger
|
||
|
|
from pyrogram import Client
|
||
|
|
from pyrogram.errors import FloodWait, RPCError
|
||
|
|
from pyrogram.types import Message
|
||
|
|
from tools.helpers import Parameters, get_iterlimit, is_deleted_id
|
||
|
|
|
||
|
|
|
||
|
|
@Client.on_message(command('dme'))
|
||
|
|
async def dme(client: Client, message: Message):
|
||
|
|
"""删除指令数量的消息"""
|
||
|
|
cmd, limit = Parameters.get_int(message, max_num=1500)
|
||
|
|
counter, ids_deleted = 0, []
|
||
|
|
await message.edit_text("🧹`Clearing history...`")
|
||
|
|
start = time.time()
|
||
|
|
|
||
|
|
async def delete_messages(cli: Client, ids_deleted: List[int]):
|
||
|
|
if len(ids_deleted) == 100:
|
||
|
|
try:
|
||
|
|
await cli.delete_messages(message.chat.id, ids_deleted)
|
||
|
|
except FloodWait as e:
|
||
|
|
await asyncio.sleep(e.x + 0.5)
|
||
|
|
except RPCError as e:
|
||
|
|
logger.error(e)
|
||
|
|
else:
|
||
|
|
ids_deleted.clear()
|
||
|
|
|
||
|
|
# 第一阶段，暴力扫描最近的消息，这些消息有可能无法搜索到
|
||
|
|
async for msg in client.iter_history(message.chat.id, limit=get_iterlimit(limit)):
|
||
|
|
if is_deleted_id(msg):
|
||
|
|
logger.info(f'{cmd} | scanning | {msg.message_id}')
|
||
|
|
ids_deleted.append(msg.message_id)
|
||
|
|
counter = counter + 1
|
||
|
|
await delete_messages(client, ids_deleted)
|
||
|
|
if counter == limit:
|
||
|
|
break
|
||
|
|
|
||
|
|
# 第二阶段，对于老的消息直接扫描性能不好，还会触发限制，使用搜索功能来提速
|
||
|
|
if counter < limit:
|
||
|
|
async for msg in client.search_messages(
|
||
|
|
chat_id=message.chat.id,
|
||
|
|
offset=counter,
|
||
|
|
limit=limit - counter,
|
||
|
|
from_user='me',
|
||
|
|
):
|
||
|
|
if is_deleted_id(msg) and msg.message_id not in ids_deleted:
|
||
|
|
logger.info(f'{cmd} | searching | {msg.message_id}')
|
||
|
|
ids_deleted.append(msg.message_id)
|
||
|
|
counter = counter + 1
|
||
|
|
await delete_messages(client, ids_deleted)
|
||
|
|
if counter == limit:
|
||
|
|
break
|
||
|
|
|
||
|
|
if len(ids_deleted) != 0:
|
||
|
|
await client.delete_messages(message.chat.id, ids_deleted)
|
||
|
|
text = f"🧹Deleted {counter} messages in {time.time() - start:.3f} seconds."
|
||
|
|
res = await message.reply(text)
|
||
|
|
await asyncio.sleep(3)
|
||
|
|
await res.delete()
|
||
|
|
# log
|
||
|
|
logger.success(f"{cmd} | {text}")
|
||
|
|
await logger.complete()
|