112 lines
4.2 KiB
Python
112 lines
4.2 KiB
Python
""" Module to automate message deletion. """
|
|
|
|
from asyncio import sleep
|
|
from telethon.errors.rpcbaseerrors import BadRequestError
|
|
from pagermaid import log
|
|
from pagermaid.listener import listener
|
|
|
|
|
|
@listener(outgoing=True, command="prune",
|
|
description="以此命令回复某条消息，将删除最新一条消息至该条消息之间的所有消息。限制：基于消息 ID 的 1000 条消息，大于 1000 条将不作出应答。（非群组管理员只删除自己的消息）")
|
|
async def prune(context):
|
|
""" Purge every single message after the message you replied to. """
|
|
if not context.reply_to_msg_id:
|
|
await context.edit("出错了呜呜呜 ~ 没有回复的消息。")
|
|
return
|
|
input_chat = await context.get_input_chat()
|
|
messages = []
|
|
count = 0
|
|
async for message in context.client.iter_messages(input_chat, min_id=context.reply_to_msg_id):
|
|
messages.append(message)
|
|
count += 1
|
|
messages.append(context.reply_to_msg_id)
|
|
if len(messages) == 100:
|
|
await context.client.delete_messages(input_chat, messages)
|
|
messages = []
|
|
|
|
if messages:
|
|
await context.client.delete_messages(input_chat, messages)
|
|
await log(f"批量删除了 {str(count)} 条消息。")
|
|
notification = await send_prune_notify(context, count)
|
|
await sleep(.5)
|
|
await notification.delete()
|
|
|
|
|
|
@listener(outgoing=True, command="selfprune",
|
|
description="删除当前对话您发送的特定数量的消息。（倒序）",
|
|
parameters="<数量>")
|
|
async def selfprune(context):
|
|
""" Deletes specific amount of messages you sent. """
|
|
if not len(context.parameter) == 1:
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
return
|
|
try:
|
|
count = int(context.parameter[0])
|
|
except ValueError:
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
return
|
|
count_buffer = 0
|
|
async for message in context.client.iter_messages(context.chat_id, from_user="me"):
|
|
if count_buffer == count:
|
|
break
|
|
await message.delete()
|
|
count_buffer += 1
|
|
await log(f"批量删除了自行发送的 {str(count)} 条消息。")
|
|
notification = await send_prune_notify(context, count)
|
|
await sleep(.5)
|
|
await notification.delete()
|
|
|
|
|
|
@listener(outgoing=True, command="yourprune",
|
|
description="删除当前对话您回复用户所发送的特定数量的消息。（倒序、需要删除消息权限）",
|
|
parameters="<数量>")
|
|
async def yourprune(context):
|
|
""" Deletes specific amount of messages someone sent. """
|
|
if not context.reply_to_msg_id:
|
|
await context.edit("出错了呜呜呜 ~ 没有回复的消息。")
|
|
return
|
|
target = await context.get_reply_message()
|
|
if not len(context.parameter) == 1:
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
return
|
|
try:
|
|
count = int(context.parameter[0])
|
|
except ValueError:
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
return
|
|
count_buffer = 0
|
|
async for message in context.client.iter_messages(context.chat_id, from_user=target.from_id):
|
|
if count_buffer == count:
|
|
break
|
|
await message.delete()
|
|
count_buffer += 1
|
|
await log(f"批量删除了回复用户所发送的 {str(count)} 条消息。")
|
|
notification = await send_prune_notify(context, count)
|
|
await sleep(.5)
|
|
await notification.delete()
|
|
|
|
|
|
@listener(outgoing=True, command="del",
|
|
description="删除当前对话您回复的那条消息。（需要回复一条消息）")
|
|
async def delete(context):
|
|
""" Deletes the message you replied to. """
|
|
target = await context.get_reply_message()
|
|
if context.reply_to_msg_id:
|
|
try:
|
|
await target.delete()
|
|
await context.delete()
|
|
await log("删除了一条消息。")
|
|
except BadRequestError:
|
|
await context.edit("出错了呜呜呜 ~ 缺少删除此消息的权限。")
|
|
else:
|
|
await context.delete()
|
|
|
|
|
|
async def send_prune_notify(context, count):
|
|
return await context.client.send_message(
|
|
context.chat_id,
|
|
"删除了 "
|
|
+ str(count)
|
|
+ " 条消息。"
|
|
)
|