diff --git a/db.py b/db.py
index 6c6d4d5..75dab11 100644
--- a/db.py
+++ b/db.py
@@ -233,9 +233,14 @@ class BotDB:
except sqlite3.Error as error:
print(error)
- def get_blacklist_users_by_filters(self):
- """Возвращает список пользователей в черном списке по фильтру"""
- return None
+ def get_blacklist_users_by_id(self, user_id):
+ """Возвращает список пользователей в черном списке по user_id"""
+ try:
+ result = self.cursor.execute("SELECT user_id, user_name, message_for_user, date_to_unban FROM `blacklist` WHERE user_id = ?", (user_id, ))
+ return self.cursor.fetchone()
+ except sqlite3.Error as error:
+ print(error)
+
def check_user_in_blacklist(self, user_id):
"""Проверяет, существует ли запись с данным user_id в blacklist."""
diff --git a/main.py b/main.py
index a023ec7..6cb05a3 100644
--- a/main.py
+++ b/main.py
@@ -57,6 +57,11 @@ class TelegramHelperBot:
# Router for user
@self.bot.message_handler(func=lambda message: True, chat_types=['private'])
def handle_message(message):
+ if BotDB.check_user_in_blacklist(message.from_user.id):
+ attribute = BotDB.get_blacklist_users_by_id(message.from_user.id)
+ self.bot.send_message(message.chat.id,
+ f'Ты заблокирован\nПричина блокировки: {attribute[2]}\nДата разблокировки: {attribute[3]}', parse_mode='HTML')
+ return
if self.state == State.START:
if message.text == '/start':
self.start_message(message)
@@ -88,13 +93,14 @@ class TelegramHelperBot:
self.bot.send_message(message.chat.id,
"Не понимаю где ты находишься. Нажми /state, и я расскажу что ты можешь "
"сделать")
+
if self.state == State.SUGGEST:
self.bot.register_next_step_handler(message, self.send_to_suggest)
self.state = State.START
if message.text == '/start':
self.state = State.START
self.start_message(message)
- elif self.state == State.PRE_CHAT:
+ if self.state == State.PRE_CHAT:
self.bot.register_next_step_handler(message, self.resend_message_in_group_for_message)
self.state = State.START
if message.text == '/start':