for_IT_MGKB_bot/alerts_bot.py at main · DinoWithPython/for_IT_MGKB_bot · GitHub

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
78
79
80
81
82
83
84
85
86
87
88
import time
from telebot import types
import config


def alert_func(bot):
    @bot.message_handler(func=lambda message: message.text == config.BTN_ALRT)
    def alert(message):
        kb_alert = types.ReplyKeyboardMarkup(
            resize_keyboard=True, one_time_keyboard=True
        )
        button = types.KeyboardButton
        kb_alert.add(
            button(text=config.BTN_ALRT_TEXT_1),
            button(text=config.BTN_ALRT_TEXT_2),
        )
        bot.send_message(
            message.chat.id, config.TEXT_ALRT_TEXT_1, reply_markup=kb_alert
        )


    @bot.message_handler(func=lambda message: message.text == config.BTN_ALRT_TEXT_1)
    def alert_policl(message):
        alert_policl = types.ReplyKeyboardMarkup(
            resize_keyboard=True, one_time_keyboard=True
        )
        button = types.KeyboardButton
        alert_policl.add(
            button(text=config.MAIN_MENU),
        )

        bot.send_message(
            message.chat.id,
            config.TEXT_ALRT_TEXT_2
        )

        bot.register_next_step_handler(message, alert_text, config.TEXT_ALRT_TEXT_5)

        # bot.send_message(
        #     message.chat.id, config.TEXT_ALRT_TEXT_3, reply_markup=alert_policl
        # )

    @bot.message_handler(func=lambda message: message.text == config.BTN_ALRT_TEXT_2)
    def alert_stac(message):
        alert_stac = types.ReplyKeyboardMarkup(
            resize_keyboard=True, one_time_keyboard=True
        )
        button = types.KeyboardButton
        alert_stac.add(
            button(text=config.MAIN_MENU),
        )

        bot.send_message(
            message.chat.id,
            config.TEXT_ALRT_TEXT_2
        )

        bot.register_next_step_handler(message, alert_text, config.TEXT_ALRT_TEXT_4)


    def alert_text(message, system):
        """Принимает текст ошибки и отправляет его."""
        alert_mes = types.ReplyKeyboardMarkup(
            resize_keyboard=True, one_time_keyboard=True
        )
        button = types.KeyboardButton
        alert_mes.add(
            button(text=config.MAIN_MENU),
        )
        text_from_user = message.text

        id_users = set(config.USERS)
        # id_users.remove(message.from_user.id)
        for id_user in id_users:
            try:
                bot.send_message(
                    id_user,
                    f"❗❗ Оповещение о {system}:\n\n{text_from_user}",
                    reply_markup=alert_mes
                )
                time.sleep(0.5)
            except Exception:
                print(f'{id_user} не активен.')
        time.sleep(1)

        bot.send_message(
            message.chat.id, config.TEXT_ALRT_TEXT_3, reply_markup=alert_mes
        )