EmojiCaptchaBot/plugins/callbacks/ban.py

import logging

from pyrogram import filters
from pyrogram.enums.chat_member_status import ChatMemberStatus
from pyrogram.types import CallbackQuery, Message

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

logger = logging.getLogger(__name__)


@PyroClient.on_callback_query(filters.regex(r"ban;[\s\S]*"))
async def callback_ban(app: PyroClient, callback: CallbackQuery):
    group = await PyroGroup.create_if_not_exists(callback.message.chat.id, None, True)
    locale = group.select_locale(app, callback.message.from_user)

    if (await app.get_chat_member(group.id, callback.from_user.id)).status not in [
        ChatMemberStatus.ADMINISTRATOR,
        ChatMemberStatus.OWNER,
    ]:
        await callback.answer(
            app._("wrong_user", "callbacks", locale=locale), show_alert=True
        )
        return

    parsed = CallbackBan.from_callback(callback)

    user = await app.find_user(parsed.user_id, group.id)

    logger.info(
        "User %s has been marked as failed the captcha by %s",
        user.id,
        callback.from_user.id,
    )

    await user.set_mistakes(3)
    await user.set_failed(True)
    await callback.answer(
        app._("captcha_failed_force", "callbacks", locale=locale).format(
            user_id=user.id
        ),
        show_alert=True,
    )

    banned = await app.ban_chat_member(group.id, user.id)

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

    await callback.message.delete()
    return