EmojiCaptchaBot/plugins/callbacks/emoji_button.py

import logging

from pyrogram import filters
from pyrogram.types import (
    CallbackQuery,
    ChatPermissions,
    InlineKeyboardButton,
    InlineKeyboardMarkup,
    Message,
)

from classes.callbacks import CallbackEmoji
from classes.pyroclient import PyroClient
from classes.pyrogroup import PyroGroup

logger = logging.getLogger(__name__)


@PyroClient.on_callback_query(filters.regex(r"emoji:[\s\S]*"))  # type: ignore
async def callback_emoji_button(app: PyroClient, callback: CallbackQuery):
    parsed = CallbackEmoji.from_callback(callback)
    group = await PyroGroup.create_if_not_exists(callback.message.chat.id, None, True)
    locale = group.select_locale(app, callback.message.from_user)

    if callback.from_user.id != parsed.user_id:
        await callback.answer(
            app._("wrong_user", "callbacks", locale=locale), show_alert=True
        )
        return

    user = await app.find_user(callback.from_user, group.id)

    logger.debug(
        "User %s has pressed the %s emoji '%s'",
        user.id,
        "correct" if parsed.emoji in user.emojis else "wrong",
        parsed.emoji,
    )

    if parsed.emoji in user.selected:
        await callback.answer()
        return

    await user.update_selected(parsed.emoji)

    if parsed.emoji in user.emojis:
        await user.update_score(1)

        if user.score >= 5:
            logger.info("User %s has passed the captcha", user.id)

            await callback.message.delete()
            await callback.answer(
                app._("captcha_passed", "callbacks", locale=locale), show_alert=True
            )
            await app.send_message(
                group.id,
                app._("welcome", "messages", locale=locale).format(
                    mention=callback.from_user.mention
                ),
            )
            await app.restrict_chat_member(
                chat_id=group.id,
                user_id=callback.from_user.id,
                permissions=callback.message.chat.permissions,
            )
            return

        logger.info(
            "User %s has chosen correctly and needs to select %s more emoji(s)",
            user.id,
            5 - user.score,
        )

        emoji_placeholder = "✅"
    else:
        if user.mistakes >= 2:
            logger.info("User %s has failed the captcha", user.id)

            await user.set_failed(True)
            await callback.answer(
                app._("captcha_failed", "callbacks", locale=locale), show_alert=True
            )

            banned = await app.ban_chat_member(group.id, callback.from_user.id)

            if isinstance(banned, Message):
                await banned.delete()

            await callback.message.delete()
            return

        await user.update_mistakes(1)
        logger.info(
            "User %s has made a mistake and has %s attempt(s) left",
            user.id,
            2 - user.mistakes,
        )

        await callback.answer(
            app._("captcha_mistake", "callbacks", locale=locale).format(
                remaining=2 - user.mistakes
            ),
            show_alert=True,
        )

        emoji_placeholder = "🛑"

    button_replace = (0, 0)

    for row_index, row in enumerate(callback.message.reply_markup.inline_keyboard):
        for button_index, button in enumerate(row):
            if button.text == parsed.emoji:
                button_replace = (row_index, button_index)

    new_keyboard = callback.message.reply_markup.inline_keyboard
    new_keyboard[button_replace[0]][button_replace[1]] = InlineKeyboardButton(
        emoji_placeholder, "nothing"
    )

    await callback.edit_message_reply_markup(
        reply_markup=InlineKeyboardMarkup(new_keyboard)
    )