Bot_TG/main.py at main · Lemetist/Bot_TG · 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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
import telebot
from telebot import types
from datetime import date
from config import TOKEN
from logics import reforming_list, data, save_to_excel

# Токен бота
bot = telebot.TeleBot(TOKEN)

# Динамический список отсутствующих
absentees = [
    "Алаторцев П.В.",
    "Багиров Э.Э.",
    "Башкиров И.В.",
    "Борхоев Б.Б.",
    "Евстафьев А.А.",
    "Журавлева М.Р.",
    "Киреев А.А.",
    "Ламажап Н.А.",
    "Ленючев Д.А.",
    "Львов М.Д.",
    "Макаров Е.С.",
    "Овчинникова И.А.",
    "Подлинный М.В.",
    "Сатин Т.С.",
    "Секункова П.А.",
    "Семечаевский П.В.",
    "Сержантов И.А.",
    "Токарев А.Д.",
    "Чичёв А.Б.",
    "Шахкиримов К.И.",
    "Шишалов С.Д.",
    "Максимов М.М.",
    "Маковский А.С."
]

# Создаем клавиатуру для выбора отсутствующих
def generate_keyboard():
    keyboard = types.ReplyKeyboardMarkup(one_time_keyboard=True, resize_keyboard=True)
    for name in absentees:
        button = types.KeyboardButton(name)
        keyboard.add(button)
    keyboard.add(types.KeyboardButton('Вернуться на /start'))  # Добавляем кнопку "Вернуться на /start"
    return keyboard

# Команда /start для начала общения с ботом
@bot.message_handler(commands=['start'])
def start(message):
    markup = types.ReplyKeyboardMarkup(row_width=2)
    itembtn1 = types.KeyboardButton('Посмотреть список')
    itembtn2 = types.KeyboardButton('Заполнить список заново')
    itembtn3 = types.KeyboardButton('Выбрать присутствующих')
    markup.add(itembtn1, itembtn2, itembtn3)

    if len(absentees) > 0:
        bot.send_message(message.chat.id,
                         "Привет! Я твой бот для учета отсутствующих. Чтобы выбрать действие, нажми на соответствующую кнопку.",
                         reply_markup=markup)
    else:
        bot.send_message(message.chat.id,
                         "Привет! Я твой бот для учета отсутствующих. На данный момент отсутствующих нет.",
                         reply_markup=markup)

# Команда /absentees для вывода списка отсутствующих и заполнения таблицы
@bot.message_handler(commands=['absentees'])
def show_absentees(message):
    if len(absentees) > 0:
        bot.send_message(message.chat.id, f"{date.today()}")
        absentees_list = "\n".join(absentees)
        bot.send_message(message.chat.id, f"Список отсутствующих:\n{absentees_list}")
        print(absentees)
        reforming_list(absentees)  # Передаем список, а не строку
        save_to_excel(data)
        bot.send_message(message.chat.id, f"Таблица отсутствующих сохранена в файле ИСП11-322АП.xlsx.")
    else:
        bot.send_message(message.chat.id, "На данный момент отсутствующих нет.")

# Команда /fill для заполнения списка заново
@bot.message_handler(commands=['fill'])
def refill_absentees(message):
    global absentees

    absentees = [
        "Алаторцев П.В.",
        "Багиров Э.Э.",
        "Башкиров И.В.",
        "Борхоев Б.Б.",
        "Евстафьев А.А.",
        "Журавлева М.Р.",
        "Киреев А.А.",
        "Ламажап Н.А.",
        "Ленючев Д.А.",
        "Львов М.Д.",
        "Макаров Е.С.",
        "Овчинникова И.А.",
        "Подлинный М.В.",
        "Сатин Т.С.",
        "Секункова П.А.",
        "Семечаевский П.В.",
        "Сержантов И.А.",
        "Токарев А.Д.",
        "Чичёв А.Б.",
        "Шахкиримов К.И.",
        "Шишалов С.Д.",
        "Максимов М.М.",
        "Маковский А.С."
    ]
    bot.send_message(message.chat.id, "Список отсутствующих успешно заполнен заново.")

# Обработчик нажатий на кнопки
@bot.message_handler(func=lambda message: True)
def handle_buttons(message):
    if message.text == 'Посмотреть список':
        show_absentees(message)
    elif message.text == 'Заполнить список заново':
        refill_absentees(message)
    elif message.text == 'Выбрать присутствующих':
        bot.send_message(message.chat.id, "Выбрать присутствующих:", reply_markup=generate_keyboard())
    elif message.text in absentees:
        remove_absentee(message)
    elif message.text == 'Вернуться на /start':
        start(message)
    else:
        bot.send_message(message.chat.id, "Неизвестная команда. Используйте кнопки для взаимодействия.")

# Обработчик выбора отсутствующего
def remove_absentee(message):
    global absentees
    removed_name = message.text
    absentees.remove(removed_name)
    bot.send_message(message.chat.id, "Выбрать присутствующих:", reply_markup=generate_keyboard())

# Запускаем бота
bot.polling()