blankie
/
sukuinote
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Refactor yeetpurge

This commit is contained in:
blank X 2020-12-15 20:16:15 +07:00
parent a2fe972793
commit 0f33b4b114
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
sukuinote/plugins/delete.py
View File

@ -1,4 +1,5 @@
import html import html
import asyncio
from pyrogram import Client, filters from pyrogram import Client, filters
from .. import config, help_dict, log_errors, public_log_errors from .. import config, help_dict, log_errors, public_log_errors
@ -49,6 +50,7 @@ async def purge(client, message):
await client.delete_messages(message.chat.id, ids) await client.delete_messages(message.chat.id, ids)
yeetpurge_info = {True: dict(), False: dict()} yeetpurge_info = {True: dict(), False: dict()}
yeetpurge_lock = asyncio.Lock()
@Client.on_message(~filters.sticker & ~filters.via_bot & ~filters.edited & filters.me & filters.command(['yp', 'yeetpurge', 'syp', 'selfyeetpurge'], prefixes=config['config']['prefixes'])) @Client.on_message(~filters.sticker & ~filters.via_bot & ~filters.edited & filters.me & filters.command(['yp', 'yeetpurge', 'syp', 'selfyeetpurge'], prefixes=config['config']['prefixes']))
@log_errors @log_errors
@public_log_errors @public_log_errors
@ -58,21 +60,25 @@ async def yeetpurge(client, message):
await message.delete() await message.delete()
return return
info = yeetpurge_info['s' in message.command[0]] info = yeetpurge_info['s' in message.command[0]]
if message.chat.id not in info: async with yeetpurge_lock:
resp = await message.reply_text('Reply to end destination') if message.from_user.id not in info:
info[message.chat.id] = (message.message_id, resp.message_id, reply.message_id, reply.outgoing) info[message.from_user.id] = dict()
return info = info[message.from_user.id]
og_message, og_resp, og_reply, og_reply_outgoing = info.pop(message.chat.id) if message.chat.id not in info:
messages = set((og_message, og_resp, message.message_id)) resp = await message.reply_text('Reply to end destination')
thing = [og_reply, reply.message_id] info[message.chat.id] = (message, reply, resp)
thing.sort() return
thing0, thing1 = thing og_message, og_reply, og_resp = info.pop(message.chat.id)
if not ('s' in message.command[0] and not og_reply_outgoing): messages = set((og_message.message_id, message.message_id, og_resp.message_id))
messages.add(og_reply) if not ('s' in message.command[0] and not og_reply.outgoing):
async for i in client.iter_history(message.chat.id, offset_id=thing1 + 1): messages.add(og_reply.message_id)
if not ('s' in message.command[0] and not reply.outgoing):
messages.add(reply.message_id)
from_id, to_id = sorted((og_reply.message_id, reply.message_id))
async for i in client.iter_history(message.chat.id, offset_id=to_id):
if not ('s' in message.command[0] and not i.outgoing): if not ('s' in message.command[0] and not i.outgoing):
messages.add(i.message_id) messages.add(i.message_id)
if thing0 + 1 >= i.message_id: if from_id >= i.message_id:
break break
await client.delete_messages(message.chat.id, messages) await client.delete_messages(message.chat.id, messages)