100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
import html
|
|
import asyncio
|
|
from pyrogram import Client, filters
|
|
from .. import config, help_dict, log_errors, public_log_errors
|
|
|
|
@Client.on_message(~filters.sticker & ~filters.via_bot & ~filters.edited & filters.me & filters.command(['d', 'del', 'delete'], prefixes=config['config']['prefixes']))
|
|
@log_errors
|
|
@public_log_errors
|
|
async def delete(client, message):
|
|
messages = set((message.message_id,))
|
|
reply = message.reply_to_message
|
|
if not getattr(reply, 'empty', True):
|
|
messages.add(reply.message_id)
|
|
else:
|
|
async for i in client.iter_history(message.chat.id, offset=1):
|
|
if i.outgoing:
|
|
messages.add(i.message_id)
|
|
break
|
|
await client.delete_messages(message.chat.id, messages)
|
|
|
|
@Client.on_message(~filters.sticker & ~filters.via_bot & ~filters.edited & filters.me & filters.command(['p', 'purge', 'sp', 'selfpurge'], prefixes=config['config']['prefixes']))
|
|
@log_errors
|
|
@public_log_errors
|
|
async def purge(client, message):
|
|
command = message.command
|
|
selfpurge = 's' in command.pop(0)
|
|
command = ' '.join(command)
|
|
ids = set((message.message_id,))
|
|
reply = message.reply_to_message
|
|
if command.isnumeric():
|
|
command = int(command)
|
|
if selfpurge:
|
|
async for i in client.iter_history(message.chat.id, offset=1):
|
|
if not (selfpurge and not i.outgoing):
|
|
ids.add(i.message_id)
|
|
command -= 1
|
|
if command <= 0:
|
|
break
|
|
else:
|
|
async for i in client.iter_history(message.chat.id, offset=1, limit=command):
|
|
ids.add(i.message_id)
|
|
elif not getattr(reply, 'empty', True):
|
|
if not (selfpurge and not reply.outgoing):
|
|
ids.add(reply.message_id)
|
|
async for i in client.iter_history(message.chat.id, offset=1):
|
|
if not (selfpurge and not i.outgoing):
|
|
ids.add(i.message_id)
|
|
if reply.message_id + 1 >= i.message_id:
|
|
break
|
|
await client.delete_messages(message.chat.id, ids)
|
|
|
|
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']))
|
|
@log_errors
|
|
@public_log_errors
|
|
async def yeetpurge(client, message):
|
|
reply = message.reply_to_message
|
|
if getattr(reply, 'empty', True):
|
|
await message.delete()
|
|
return
|
|
info = yeetpurge_info['s' in message.command[0]]
|
|
async with yeetpurge_lock:
|
|
if message.from_user.id not in info:
|
|
info[message.from_user.id] = dict()
|
|
info = info[message.from_user.id]
|
|
if message.chat.id not in info:
|
|
resp = await message.reply_text('Reply to end destination')
|
|
info[message.chat.id] = (message, reply, resp)
|
|
return
|
|
og_message, og_reply, og_resp = info.pop(message.chat.id)
|
|
messages = set((og_message.message_id, message.message_id, og_resp.message_id))
|
|
if not ('s' in message.command[0] and not og_reply.outgoing):
|
|
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):
|
|
messages.add(i.message_id)
|
|
if from_id >= i.message_id:
|
|
break
|
|
await client.delete_messages(message.chat.id, messages)
|
|
|
|
help_dict['delete'] = ('Delete',
|
|
'''{prefix}delete <i>(maybe reply to a message)</i> - Deletes the replied to message, or your latest message
|
|
Aliases: {prefix}d, {prefix}del
|
|
|
|
{prefix}purge <i>(as reply to a message)</i> - Purges the messages between the one you replied (and including the one you replied)
|
|
Aliases: {prefix}p
|
|
|
|
{prefix}selfpurge <i>(as reply to a message)</i> - {prefix}p but only your messages
|
|
Aliases: {prefix}sp
|
|
|
|
{prefix}yeetpurge <i>(as reply to a message)</i> - Purges messages in between
|
|
Aliases: {prefix}yp
|
|
|
|
{prefix}selfyeetpurge <i>(as reply to a message)</i> - {prefix}yp but only your messages
|
|
Aliases: {prefix}syp''')
|