PagerMaid_Plugins/dme.py

""" Module to automate message deletion. """
from asyncio import sleep
from os import path, remove
from os.path import exists
from PIL import Image, UnidentifiedImageError
from pagermaid import redis, log, redis_status
from pagermaid.listener import listener
from pagermaid.utils import alias_command


@listener(is_plugin=True, outgoing=True, command=alias_command("dme"),
          description="编辑并删除当前对话您发送的特定数量的消息。限制：基于消息 ID 的 1000 条消息，大于 1000 "
                      "条可能会触发删除消息过快限制。入群消息非管理员无法删除。（倒序）当数字足够大时即可实现删除所有消息。",
          parameters="<数量> [文本]")
async def dme(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")
        try:
            im = Image.open("plugins/dme.webp")
        except UnidentifiedImageError:
            await context.edit("替换图片设置发生错误。")
            return
        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)
        + " 条消息。"
    )