aboutsummaryrefslogtreecommitdiff
path: root/handlers/member.py
blob: b228f6cb02b75ecc7ba6b70e795a80e030b2eacb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from aiogram import types as t

from shared.instances import bot, chats, dp
from utils import filters as f

polls: dict[int, int] = {}


@dp.chat_join_request_handler()
async def new_member(cjr: t.ChatJoinRequest) -> None:
    if cjr.from_user.id in polls:
        return

    reply = await bot.send_message(
        cjr.chat.id,
        f'<a href="tg://user?id={cjr.from_user.id}">{cjr.from_user.mention}</a> хочет в чат',
        parse_mode=t.ParseMode.HTML,
    )
    poll = await reply.reply_poll(
        "Пускаем ?",
        [
            "Да",
            "Нет",
        ],
        False,
        reply_markup=t.InlineKeyboardMarkup().add(
            t.InlineKeyboardButton(
                "Проверить опрос",
                callback_data=f"check_request_poll:{cjr.from_user.id}",
            )
        ),
    )
    await bot.send_message(
        cjr.from_user.id, "Заявка на вступление в группу будет вскоре рассмотрена"
    )
    polls[cjr.from_user.id] = poll.message_id


@dp.chat_member_handler(f.user.new_user)
async def admin_accept_member(cmu: t.ChatMemberUpdated) -> None:
    user_id = cmu.new_chat_member.user.id
    if user_id in polls:
        polls.pop(user_id)
        await bot.send_message(
            user_id,
            "Ваша заявка на вступление принята, добро пожаловать в группу",
        )


@dp.callback_query_handler(
    f.message.is_chat, lambda clb: clb.data.split(":")[0] == "check_request_poll"
)
async def check_poll(clb: t.CallbackQuery) -> None:
    poll = clb.message.poll
    msg = clb.message
    data = clb.data.split(":")
    user_id = int(data[1])
    min_answers = chats.get(msg.chat.id).members.answer_count

    if poll.total_voter_count < min_answers:
        await clb.answer(
            f"Нужно хотябы {min_answers} голоса, сейчас {poll.total_voter_count}"
        )
    else:
        if not poll.is_closed:
            await bot.stop_poll(msg.chat.id, msg.message_id)

        if poll.options[0].voter_count > poll.options[1].voter_count:
            await bot.approve_chat_join_request(msg.chat.id, user_id)
            await bot.send_message(
                user_id, "Ваша заявка на вступление принята, добро пожаловать в группу"
            )
        else:
            await bot.decline_chat_join_request(msg.chat.id, user_id)
            await bot.send_message(user_id, "Ваша заявка на вступление НЕ принята")
            if not msg.chat.has_protected_content:
                await msg.forward(user_id)