add new func in voice bot

database/db.py

@@ -719,7 +719,7 @@ class BotDB:
             None: Если обновление прошло успешно
             sqlite3. Error: Если произошла ошибка при выполнении запроса
         """
-        self.logger.info(f"Запуск функции update_date_for_user: user_id={user_id}, emoji={emoji}")
+        self.logger.info(f"Запуск функции update_emoji_for_user: user_id={user_id}, emoji={emoji}")
         try:
             self.connect()
             self.cursor.execute("UPDATE our_users SET emoji = ? WHERE user_id = ?",
@@ -750,8 +750,12 @@ class BotDB:
         try:
             self.connect()
             self.cursor.execute("SELECT emoji FROM our_users WHERE user_id = ?", (user_id,))
-            result = self.cursor.fetchone()[0]
-            return result
+            pre_result = self.cursor.fetchone()
+            if pre_result[0] is not None:
+                print(f'Результат функции check_emoji_for_user - {str(pre_result[0])}')
+                return str(pre_result[0])
+            else:
+                return None
         except sqlite3.Error as error:
             self.logger.error(f"Ошибка проверки эмодзи в базе: {error}")
             return None
@@ -1046,6 +1050,48 @@ class BotDB:
         except Exception as e:
             self.logger.error(f"Ошибка в функции get_author_id_by_helper_message_id {str(e)}")
 
+    def get_user_id_by_message_id_for_voice_bot(self, message_id: int):
+        self.logger.info(f"Запуск функции get_user_id_by_message_id_for_voice_bot, идентификатор поста "
+                         f"{message_id}")
+        try:
+            self.connect()
+            result = self.cursor.execute("SELECT user_id "
+                                         "FROM audio_moderate WHERE message_id = ?",
+                                         (message_id,))
+            user_id = result.fetchone()[0]
+            self.logger.info(f"Функция get_user_id_by_message_id_for_voice_bot получила author_id {user_id}")
+            return user_id
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции get_user_id_by_message_id_for_voice_bot {str(e)}")
+
+    def set_user_id_and_message_id_for_voice_bot(self, message_id: int, user_id: int):
+        self.logger.info(f"Запуск функции set_user_id_and_message_id_for_voice_bot, идентификатор поста "
+                         f"{message_id}, user_id {user_id}")
+        try:
+            self.connect()
+            result = self.cursor.execute(
+                "INSERT INTO audio_moderate (message_id, user_id)"
+                "VALUES (?, ?)", (message_id, user_id))
+            self.conn.commit()
+            self.logger.info(f"Функция set_user_id_and_message_id_for_voice_bot отработала успешно")
+            return True
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции set_user_id_and_message_id_for_voice_bot {str(e)}")
+
+    def get_user_id_by_file_name(self, file_name: str):
+        self.logger.info(f"Запуск функции get_user_id_by_file_name, идентификатор поста "
+                         f"{file_name}")
+        try:
+            self.connect()
+            result = self.cursor.execute("SELECT author_id "
+                                         "FROM audio_message_reference WHERE file_name = ?",
+                                         (file_name,))
+            user_id = result.fetchone()[0]
+            self.logger.info(f"Функция get_user_id_by_file_name получила author_id {user_id}")
+            return user_id
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции get_user_id_by_file_name {str(e)}")
+
     def add_post_content_in_db(self, post_id: int, message_id: int, content_name: str, type_content: str):
         self.logger.info(
             f"Запуск функции add_post_content_in_db: post_id={post_id}, message_id={message_id}, "
@@ -1156,6 +1202,24 @@ class BotDB:
         finally:
             self.close()
 
+    def delete_listen_count_for_user(self, user_id):
+        """Удаляет данные о прослушанных пользователем аудио"""
+        self.logger.info(
+            f"Запуск функции delete_listen_count_for_user. user_id={user_id}")
+        try:
+            self.connect()
+            self.cursor.execute("DELETE FROM `listen_audio_users` WHERE `user_id` = ?",
+                                (user_id,))
+            self.conn.commit()
+            self.logger.info(
+                f"Функция delete_listen_count_for_user успешно отработала")
+            return None
+        except sqlite3.Error as error:
+            self.logger.error(f"Ошибка удаления записей прослушивания по пользователю: {error}")
+            raise
+        finally:
+            self.close()
+
     def get_id_for_audio_record(self, user_id):
         """Получает ID аудио сообщения пользователя"""
         self.logger.info(