refactor voice_bot

database/db.py

@@ -1040,61 +1040,86 @@ class BotDB:
 
     def last_date_audio(self):
         """Получаем дату последнего войса"""
+        self.logger.info(
+            f"Запуск функции last_date_audio")
         try:
             self.connect()
             result = self.cursor.execute(
                 "SELECT `date_added` FROM `audio_message_reference` ORDER BY date_added DESC LIMIT 1")
-            return result.fetchone()[0]
+            last_date = result.fetchone()[0]
+            self.logger.info(f"Последняя дата сообщения {last_date}")
+            return last_date
         except sqlite3.Error as error:
-            print(error)
+            self.logger.error(f"Ошибка получения последней даты войса: {error}")
+            raise
         finally:
             self.close()
 
     def get_last_user_audio_record(self, user_id):
         """Получает данные о количестве записей пользователя"""
+        self.logger.info(
+            f"Запуск функции get_last_user_audio_record. user_id={user_id}")
         try:
             self.connect()
-            result = self.cursor.execute("SELECT `file_id` FROM `audio_message_reference` WHERE `author_id` = ?",
-                                         (user_id,))
-            return bool(len(result.fetchall()))
+            r = self.cursor.execute("SELECT `file_id` FROM `audio_message_reference` WHERE `author_id` = ?",
+                                    (user_id,))
+            result = bool(len(r.fetchall()))
+            self.logger.info(
+                f"Результат функции get_last_user_audio_record: {result}")
+            return result
         except sqlite3.Error as error:
-            print(error)
+            self.logger.error(f"Ошибка получения последней даты войса: {error}")
+            raise
         finally:
             self.close()
 
     def get_id_for_audio_record(self, user_id):
         """Получает ID аудио сообщения пользователя"""
+        self.logger.info(
+            f"Запуск функции get_id_for_audio_record. user_id={user_id}")
         try:
             self.connect()
-            result = self.cursor.execute(
+            r = self.cursor.execute(
                 "SELECT `file_id` FROM `audio_message_reference` WHERE `author_id` = ? "
                 "ORDER BY date_added DESC LIMIT 1",
                 (user_id,))
-            return result.fetchone()[0]
+            result = r.fetchone()[0]
+            self.logger.info(
+                f"Результат функции get_id_for_audio_record: {result}")
+            return result
         except sqlite3.Error as error:
-            print(error)
+            self.logger.error(f"Ошибка получения последней даты войса: {error}")
+            raise
         finally:
             self.close()
 
     def get_path_for_audio_record(self, user_id):
         """Получает данные о названии файла"""
+        self.logger.info(
+            f"Запуск функции get_path_for_audio_record. user_id={user_id}")
         try:
             self.connect()
-            result = self.cursor.execute(
+            r = self.cursor.execute(
                 "SELECT `file_name` "
                 "FROM `audio_message_reference` "
                 "WHERE `author_id` = ? "
                 "ORDER BY date_added "
                 "DESC LIMIT 1",
                 (user_id,))
-            return result.fetchone()[0]
+            result = r.fetchone()[0]
+            self.logger.info(
+                f"Результат функции get_path_for_audio_record: {result}")
+            return result
         except sqlite3.Error as error:
-            print(error)
+            self.logger.error(f"Ошибка получения последней даты войса: {error}")
+            raise
         finally:
             self.close()
 
     def check_listen_audio(self, user_id):
         """Проверяет прослушано ли аудио пользователем"""
+        self.logger.info(
+            f"Запуск функции check_listen_audio. user_id={user_id}")
         try:
             self.connect()
             query_listen_audio = self.cursor.execute(
@@ -1111,14 +1136,19 @@ class BotDB:
             new_sign = []
             for i in new_sign1:
                 new_sign.append(i[0])
+            self.logger.info(
+                f"Функция check_listen_audio успешно отработала.")
             return new_sign
         except sqlite3.Error as error:
-            print(error)
+            self.logger.error(f"Ошибка получения последней даты войса: {error}")
+            raise
         finally:
             self.close()
 
     def mark_listened_audio(self, file_name, user_id):
         """Отмечает аудио прослушанным для конкретного пользователя."""
+        self.logger.info(
+            f"Запуск функции mark_listened_audio. file_name={file_name}, user_id={user_id}")
         try:
             self.connect()
             result = self.cursor.execute(
@@ -1126,7 +1156,8 @@ class BotDB:
                 (file_name, user_id, 1))
             return self.conn.commit()
         except sqlite3.Error as error:
-            print(error)
+            self.logger.error(f"Ошибка получения последней даты войса: {error}")
+            raise
         finally:
             self.close()