TelegramBotAdmin/wbb/modules/quotly.py

from io import BytesIO

from pyrogram import filters
from pyrogram.types import Message

from wbb import app, app2, arq
from wbb.core.decorators.errors import capture_err

__MODULE__ = "Quotly"
__HELP__ = """
/q - To quote a message.
/q [INTEGER] - To quote more than 1 messages.
"""


async def quotify(messages: list):
    response = await arq.quotly(messages)
    sticker = response.result
    sticker = BytesIO(sticker)
    sticker.name = "sticker.webp"
    return sticker


def getArg(message: Message) -> str:
    arg = message.text.strip().split(None, 1)[1].strip()
    return arg


def isArgInt(message: Message) -> bool:
    count = getArg(message)
    try:
        count = int(count)
        return [True, count]
    except ValueError:
        return [False, 0]


@app.on_message(filters.command("q"))
async def quotly_func(_, message: Message):
    if not message.reply_to_message:
        await message.reply_text("Reply to a message to quote it.")
        return
    if not message.reply_to_message.text:
        await message.reply_text("Replied message has no text, can't quote it.")
        return
    m = await message.reply_text("Quoting Messages")
    if len(message.command) < 2:
        messages = [message.reply_to_message]

    elif len(message.command) == 2:
        arg = isArgInt(message)
        if arg[0]:
            if arg[1] < 2 or arg[1] > 6:
                await m.edit("Argument must be between 2-6.")
                return
            count = arg[1]
            messages = await app.get_messages(
                message.chat.id,
                [
                    i
                    for i in range(
                        message.reply_to_message.message_id,
                        message.reply_to_message.message_id + count,
                    )
                ],
                replies=0,
            )
        else:
            if getArg(message) != "r":
                await m.edit(
                    "Incorrect Argument, Pass **'r'** or **'INT'**, **EX:** __/q 2__"
                )
                return
            reply_message = await app.get_messages(
                message.chat.id, message.reply_to_message.message_id, replies=1
            )
            reply_message = reply_message.reply_to_message
            messages = [reply_message, message.reply_to_message]
    else:
        await m.edit("Incorrect argument, check quotly module in help section.")
        return
    sticker = await quotify(messages)
    await message.reply_sticker(sticker)
    await m.delete()
    sticker.close()