blacklist
This commit is contained in:
KatykhinAA
158
main.py
158
main.py
@@ -8,7 +8,7 @@ import db
|
||||
from db import BotDB
|
||||
import telebot
|
||||
import random
|
||||
from datetime import datetime
|
||||
from datetime import datetime, timedelta
|
||||
import time
|
||||
from telebot import types
|
||||
from telebot.apihelper import ApiTelegramException
|
||||
@@ -56,36 +56,29 @@ class TelegramHelperBot:
|
||||
def handle_message(message):
|
||||
if self.state == State.START:
|
||||
if message.text == '/start':
|
||||
print(f'Внутри функции handle_message // Команда /start // state - {self.state.value}')
|
||||
self.start_message(message)
|
||||
elif message.text == '📢Предложить свой пост':
|
||||
print(f'Внутри функции handle_message // Команда /suggest // state - {self.state.value}')
|
||||
self.suggest_post(message)
|
||||
self.state = State.SUGGEST
|
||||
elif message.text == '🤪Хочу стикеры':
|
||||
print(f'Внутри функции handle_message // Команда /stickers // state - {self.state.value}')
|
||||
self.stickers(message)
|
||||
self.state = State.START
|
||||
elif message.text == '📩Связаться с админами':
|
||||
print(f'Внутри функции handle_message // Команда /connect // state - {self.state.value}')
|
||||
self.connect_with_admin(message)
|
||||
self.state = State.PRE_CHAT
|
||||
print(f'В state.START - {self.state.value}')
|
||||
elif message.text == '👋🏼Сказать пока!':
|
||||
print(f'Внутри функции handle_message // Команда /end // state - {self.state.value}')
|
||||
self.end_message(message)
|
||||
self.state = State.START
|
||||
elif message.text == 'Выйти из чата':
|
||||
print(f'Внутри функции handle_message // Команда /end // state - {self.state.value}')
|
||||
self.end_message(message)
|
||||
elif message.text == '/admin':
|
||||
self.state = State.ADMIN
|
||||
#TODO: Админку сделать!
|
||||
self.bot.send_message(message.chat.id,
|
||||
"Ты в админке, Ура! Делай что хочешь")
|
||||
self.admin_panel(message)
|
||||
access = self.check_access(message.from_user.id)
|
||||
if access:
|
||||
self.admin_panel(message)
|
||||
self.state = State.ADMIN
|
||||
else:
|
||||
self.bot.send_message(message.chat.id, 'Доступ запрещен, досвидания!')
|
||||
elif message.text == '/state':
|
||||
print(f'Внутри функции handle_message // Команда /state // state - {self.state.value}')
|
||||
self.bot.send_message(message.chat.id,
|
||||
f'Твой state == {self.state.value}')
|
||||
else:
|
||||
@@ -101,15 +94,20 @@ class TelegramHelperBot:
|
||||
self.state = State.START
|
||||
|
||||
if self.state == State.CHAT:
|
||||
print(f'В state.CHAT - {self.state.value}')
|
||||
if message.text == 'Выйти из чата':
|
||||
self.state = State.START
|
||||
self.end_message(message)
|
||||
print(f'Обработчик в CHAT - {self.state.value}')
|
||||
else:
|
||||
print(f'Обработчик чата в чате CHAT - {self.state.value}')
|
||||
self.resend_message_in_group_for_message(message)
|
||||
|
||||
if self.state == State.ADMIN:
|
||||
if message == '/admin' or message == '/restart':
|
||||
access = self.check_access(message.from_user.id)
|
||||
if access:
|
||||
self.admin_panel(message)
|
||||
else:
|
||||
self.bot.send_message(message.chat.id, 'Доступ запрещен, досвидания!')
|
||||
|
||||
@self.bot.message_handler(func=lambda message: True, chat_types=['group'])
|
||||
def handle_message(message):
|
||||
"""Функция ответа админа пользователю через закрытый чат"""
|
||||
@@ -132,7 +130,7 @@ class TelegramHelperBot:
|
||||
except Exception as e:
|
||||
if LOGS:
|
||||
self.bot.send_message(chat_id=IMPORTANT_LOGS,
|
||||
text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
|
||||
text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
|
||||
elif call.data == 'publish' and call.message.content_type == 'photo':
|
||||
try:
|
||||
self.bot.send_photo(
|
||||
@@ -144,14 +142,17 @@ class TelegramHelperBot:
|
||||
except Exception as e:
|
||||
if LOGS:
|
||||
self.bot.send_message(chat_id=IMPORTANT_LOGS,
|
||||
text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
|
||||
text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
|
||||
elif call.data == 'decline':
|
||||
try:
|
||||
self.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
|
||||
except Exception as e:
|
||||
if LOGS:
|
||||
self.bot.send_message(IMPORTANT_LOGS,
|
||||
f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
|
||||
f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
|
||||
elif call.data[:3] == 'ban':
|
||||
user_id = call.data[4:]
|
||||
self.ban_user(call.message, user_id)
|
||||
|
||||
def register_chat_handler(self, message):
|
||||
self.bot.register_next_step_handler(message, self.resend_message_in_group_for_message)
|
||||
@@ -166,29 +167,96 @@ class TelegramHelperBot:
|
||||
# Черный список
|
||||
def admin_panel(self, message):
|
||||
try:
|
||||
#Добавить админа: 842766148
|
||||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||||
item1 = types.KeyboardButton("Забанить пользователя из списка")
|
||||
item2 = types.KeyboardButton("Забанить пользователя по ID")
|
||||
markup.add(item1, item2)
|
||||
self.bot.send_message(message.chat.id, "Добро пожаловать в админку. Выбери что хочешь:", reply_markup=markup)
|
||||
self.bot.register_next_step_handler(message, self.ban_user)
|
||||
item1 = types.KeyboardButton("Бан (Список)")
|
||||
item2 = types.KeyboardButton("Добавить админа")
|
||||
item3 = types.KeyboardButton("Удалить админа")
|
||||
markup.add(item1, item2, item3)
|
||||
self.bot.send_message(message.chat.id, "Добро пожаловать в админку. Выбери что хочешь:",
|
||||
reply_markup=markup)
|
||||
self.bot.register_next_step_handler(message, self.handle_admin_message)
|
||||
except Exception as e:
|
||||
self.bot.register_next_step_handler(message, self.admin_panel)
|
||||
|
||||
def ban_user(self, message):
|
||||
# проверяем, что ID передан правильно
|
||||
#TODO: остановился где-то тут, функция не дописана. Хотел сделать админку + бан пользователей
|
||||
def handle_admin_message(self, message):
|
||||
try:
|
||||
if message.text == "Забанить пользователя из списка":
|
||||
pass
|
||||
elif message.text == "Забанить пользователя по ID":
|
||||
self.bot.send_message(message.chat.id, "Пришли ID юзера")
|
||||
#BotDB.set_user_blacklist(ban_user_id)
|
||||
#bot.reply_to(message, f"Пользователь {ban_user_id} заблокирован.")
|
||||
self.get_last_users(message)
|
||||
except Exception as e:
|
||||
self.bot.reply_to(message, f"Укажи ID пользователя. Ошибка\n\n {e}")
|
||||
self.bot.register_next_step_handler(message, self.ban_user)
|
||||
self.bot.reply_to(message, f"Ошибка\n\n {e}")
|
||||
self.admin_panel(message)
|
||||
|
||||
def ban_user(self, message, user_id: int):
|
||||
user_name = BotDB.get_username(user_id=user_id)
|
||||
ban_object = {'user_id': user_id, 'user_name': user_name, 'message_for_user': None, 'date_to_unban': None}
|
||||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||||
item1 = types.KeyboardButton("Спам")
|
||||
item2 = types.KeyboardButton("Заебал стикерами")
|
||||
markup.add(item1, item2)
|
||||
self.bot.send_message(message.chat.id,
|
||||
f"Выбран пользователь: {user_id}. Выбери причину бана из списка или напиши ее в чат",
|
||||
reply_markup=markup)
|
||||
self.bot.register_next_step_handler(message, self.ban_user_step_2, ban_object)
|
||||
|
||||
def ban_user_step_2(self, message, ban_object: dict):
|
||||
ban_object['message_for_user'] = message.text
|
||||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||||
item1 = types.KeyboardButton("1")
|
||||
item2 = types.KeyboardButton("7")
|
||||
item3 = types.KeyboardButton("30")
|
||||
item4 = types.KeyboardButton("Навсегда")
|
||||
markup.add(item1, item2, item3, item4)
|
||||
self.bot.send_message(message.chat.id, f"Выбрана причина: {message.text}. Выбери срок бана в днях или напиши "
|
||||
f"его в чат",
|
||||
reply_markup=markup)
|
||||
self.bot.register_next_step_handler(message, self.ban_user_step_3, ban_object)
|
||||
|
||||
def ban_user_step_3(self, message, ban_object: dict):
|
||||
date_to_unban = None
|
||||
if message.text != 'Навсегда':
|
||||
date_to_unban = self.add_days_to_date(message.text)
|
||||
else:
|
||||
pass
|
||||
ban_object['date_to_unban'] = date_to_unban
|
||||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||||
item1 = types.KeyboardButton("Подтвердить")
|
||||
item2 = types.KeyboardButton("Отменить")
|
||||
markup.add(item1, item2)
|
||||
self.bot.send_message(message.chat.id,
|
||||
f"Необходимо подтверждение:\nПользователь:{ban_object['user_id']}\nПричина бана:{ban_object['message_for_user']}.\nСрок бана:{ban_object['date_to_unban']}",
|
||||
parse_mode='html',
|
||||
reply_markup=markup)
|
||||
self.bot.register_next_step_handler(message, self.ban_user_final_step, ban_object)
|
||||
|
||||
def ban_user_final_step(self, message, ban_object: dict):
|
||||
if message.text == 'Подтвердить':
|
||||
exists = BotDB.check_user_in_blacklist(ban_object['user_id'])
|
||||
if exists:
|
||||
self.bot.reply_to(message, f"Пользователь уже был заблокирован ранее.")
|
||||
self.admin_panel(message)
|
||||
else:
|
||||
result = BotDB.set_user_blacklist(ban_object['user_id'],
|
||||
ban_object['user_name'],
|
||||
ban_object['message_for_user'],
|
||||
ban_object['date_to_unban'])
|
||||
if result:
|
||||
self.bot.reply_to(message, f"Пользователь {ban_object['user_name']} успешно заблокирован.")
|
||||
self.admin_panel(message)
|
||||
else:
|
||||
self.bot.reply_to(message, f"Произошла ошибка при блокировке пользователя.")
|
||||
self.admin_panel(message)
|
||||
|
||||
def get_last_users(self, message):
|
||||
list_users = BotDB.get_last_users_from_bot()
|
||||
markup = types.InlineKeyboardMarkup(row_width=1)
|
||||
for user_id, full_name in list_users.items():
|
||||
button = types.InlineKeyboardButton(
|
||||
f"{full_name}", callback_data=f'ban_{user_id}'
|
||||
)
|
||||
# Добавляем кнопки на клавиатуру
|
||||
markup.add(button)
|
||||
self.bot.send_message(chat_id=message.chat.id, text="Список пользователей которые последними обращались к боту",
|
||||
reply_markup=markup)
|
||||
|
||||
def start_message(self, message):
|
||||
try:
|
||||
@@ -230,7 +298,6 @@ class TelegramHelperBot:
|
||||
self.bot.send_message(IMPORTANT_LOGS,
|
||||
f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
|
||||
|
||||
#TODO: При отправке более одного сообщения, не пересылает сообщение в чат, БАГ!!!!
|
||||
def resend_message_in_group_for_message(self, message):
|
||||
self.bot.forward_message(chat_id=GROUP_FOR_MESSAGE,
|
||||
from_chat_id=message.chat.id,
|
||||
@@ -401,11 +468,22 @@ class TelegramHelperBot:
|
||||
def __get_first_name(message):
|
||||
return message.from_user.first_name
|
||||
|
||||
@staticmethod
|
||||
def check_access(user_id: int):
|
||||
"""Проверка прав на совершение действий"""
|
||||
return BotDB.is_admin(user_id)
|
||||
|
||||
@staticmethod
|
||||
def add_days_to_date(days):
|
||||
"""Прибавляет указанное количество дней к текущей дате и возвращает дату в формате DD-MM-YYYY."""
|
||||
current_date = datetime.now()
|
||||
future_date = current_date + timedelta(days=int(days))
|
||||
formatted_date = future_date.strftime("%d-%m-%Y")
|
||||
return formatted_date
|
||||
|
||||
|
||||
bot = TelegramHelperBot(BOT_TOKEN)
|
||||
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# Запускаем бота
|
||||
bot.start()
|
||||
bot.start()
|
||||
|
||||
Reference in New Issue
Block a user