""" Module to automate message deletion. """ from asyncio import sleep from os import path, remove from os.path import exists from PIL import Image from pagermaid import redis, log, redis_status from pagermaid.listener import listener @listener(outgoing=True, command="dme", description="编辑并删除当前对话您发送的特定数量的消息。限制:基于消息 ID 的 1000 条消息,大于 1000 条可能会触发删除消息过快限制。入群消息非管理员无法删除。(倒序)当数字足够大时即可实现删除所有消息。", parameters="<数量> [文本]") async def selfprune(context): """ Deletes specific amount of messages you sent. """ reply = await context.get_reply_message() if reply and reply.photo: if exists('plugins/dme.jpg'): remove('plugins/dme.jpg') target_file = reply.photo await context.client.download_media( await context.get_reply_message(), file = "plugins/dme.jpg" ) await context.edit("替换图片设置完成。") elif reply and reply.sticker: if exists('plugins/dme.jpg'): remove('plugins/dme.jpg') await context.client.download_media(reply.media.document, file = "plugins/dme.webp") im = Image.open("plugins/dme.webp") im.save('plugins/dme.png', "png") remove('plugins/dme.webp') target_file = await context.client.upload_file('plugins/dme.png') await context.edit("替换图片设置完成。") elif path.isfile("plugins/dme.jpg"): target_file = await context.client.upload_file('plugins/dme.jpg') elif path.isfile("plugins/dme.png"): target_file = await context.client.upload_file('plugins/dme.png') else: target_file = False await context.edit("注意:没有图片进行替换。") try: count = int(context.parameter[0]) + 1 except ValueError: await context.edit("出错了呜呜呜 ~ 无效的参数。") return except IndexError: await context.edit("出错了呜呜呜 ~ 无效的参数。") return dme_msg = "别搁这防撤回了。。。" if len(context.parameter) == 1: if not redis_status: pass else: try: dme_msg = redis.get("dme_msg").decode() except: pass elif len(context.parameter) == 2: dme_msg = context.parameter[1] if not redis_status(): pass elif not dme_msg == str(count): try: redis.set("dme_msg", dme_msg) except: pass count_buffer = 0 async for message in context.client.iter_messages(context.chat_id, from_user="me"): if count_buffer == count: break if message.forward or message.via_bot or message.sticker or message.contact or message.poll or message.game or message.geo: pass elif message.text or message.voice: if not message.text == dme_msg: try: await message.edit(dme_msg) except: pass elif message.document or message.photo or message.file or message.audio or message.video or message.gif: if target_file: if not message.text == dme_msg: try: await message.edit(dme_msg, file=target_file) except: pass else: if not message.text == dme_msg: try: await message.edit(dme_msg) except: pass else: pass await message.delete() count_buffer += 1 count -= 1 count_buffer -= 1 await log(f"批量删除了自行发送的 {str(count_buffer)} / {str(count)} 条消息。") notification = await send_prune_notify(context, count_buffer, count) await sleep(.5) await notification.delete() async def send_prune_notify(context, count_buffer, count): return await context.client.send_message( context.chat_id, "删除了 " + str(count_buffer) + " / " + str(count) + " 条消息。" )