From c3a6bcbb8a2bba43103dc9bb6a7570b598b65419 Mon Sep 17 00:00:00 2001
From: KatykhinAA <andrejj.katykhin@vitech.team>
Date: Sat, 6 Jul 2024 00:33:37 +0300
Subject: [PATCH] some fix

---
 db.py   | 11 ++++++++---
 main.py |  8 +++++++-
 2 files changed, 15 insertions(+), 4 deletions(-)

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'<b>Ты заблокирован\nПричина блокировки:</b> {attribute[2]}\n<b>Дата разблокировки:</b> {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':