Merge pull request #3 from KerradKerridi/dev-3

Dev-3 Добавил новые функции MediaGroup, разные типы файлов, ответы пользователю

.gitignore

@@ -1 +1,4 @@
 /database/tg-bot-database
+/settings.ini
+/myenv/
+/venv/

database/db.py

@@ -1,14 +1,12 @@
-import sqlite3
 import os
+import sqlite3
 from datetime import datetime
-from logs.custom_logger import logger
 
-# Получение абсолютного пути к текущей директории
-current_dir = os.getcwd()
+from logs.custom_logger import logger
 
 
 class BotDB:
-    def __init__(self, name):
+    def __init__(self, current_dir, name):
         self.db_file = os.path.join(current_dir, name)
         self.conn = None
         self.cursor = None
@@ -59,7 +57,7 @@ class BotDB:
             self.logger.info(f'Получена текущая версия миграции: {version}')
             return version
         except Exception as e:
-            self.logger.error(f'Ошибка при получении текущей версии миграции: {e}')
+            self.logger.error(f'Ошибка при получении текущей версии миграции: {str(e)}')
             raise
         finally:
             self.close()
@@ -241,6 +239,68 @@ class BotDB:
         finally:
             self.close()
 
+    def get_user_id_by_username(self, username: str):
+        """
+        Возвращает user_id пользователя из базы данных по его user_name в Telegram.
+
+        Args:
+            username (str): Username пользователя.
+
+        Returns:
+            user_id (int): Идентификатор пользователя в Telegram.
+            None: Если пользователь не найден.
+
+        Raises:
+            sqlite3.Error: Если произошла ошибка при выполнении запроса.
+        """
+        try:
+            self.connect()
+            self.cursor.execute("SELECT user_id FROM our_users WHERE username = ?", (username,))
+            result = self.cursor.fetchone()
+            if result:
+                user_id = result[0]
+                self.logger.info(f"User_id пользователя найден: username={username}, user_id={user_id}")
+                return user_id
+            else:
+                self.logger.info(f"Пользователь с username={username} не найден в базе данных.")
+                return None
+        except sqlite3.Error as error:
+            self.logger.error(f"Ошибка при получении username из базы данных: {error}")
+            raise
+        finally:
+            self.close()
+
+    def get_full_name_by_id(self, user_id: str):
+        """
+        Возвращает full_name пользователя из базы данных по его username в Telegram.
+
+        Args:
+            user_id (int): Идентификатор пользователя в Telegram.
+
+        Returns:
+            full_name (str): Username пользователя.
+            None: Если пользователь не найден.
+
+        Raises:
+            sqlite3.Error: Если произошла ошибка при выполнении запроса.
+        """
+        try:
+            self.connect()
+            self.cursor.execute("SELECT full_name FROM our_users WHERE user_id = ?", (user_id,))
+            result = self.cursor.fetchone()
+            if result:
+                full_name = result[0]
+                self.logger.info(f"Username пользователя найден: user_id={user_id}, full_name={full_name}")
+                return full_name
+            else:
+                self.logger.info(f"Пользователь с user_id={user_id} не найден в базе данных.")
+                return None
+        except sqlite3.Error as error:
+            self.logger.error(f"Ошибка при получении username из базы данных: {error}")
+            raise
+        finally:
+            self.close()
+
     def get_all_user_id(self):
         """
         Возвращает список всех user_id из базы данных.
@@ -298,11 +358,6 @@ class BotDB:
         finally:
             self.close()
 
-    def change_name(self, user_id):
-        #TODO: реализовать функцию изменения имени пользователя по которому к нему обращается бот. Обновляем поля first_name, date_changed
-        #result = self.cursor.execute("UPDATE 'our_users' SET (?) WHERE user_id = (?)", (new_user_name), )
-        pass
-
     def get_info_about_stickers(self, user_id: int):
         """
         Проверяет, получил ли пользователь стикеры.
@@ -605,9 +660,11 @@ class BotDB:
             f"Запуск функции update_username_and_full_name: user_id={user_id}, username={username}, full_name={full_name}")
         try:
             self.connect()
-            self.cursor.execute("UPDATE our_users SET username = ?, full_name = ? WHERE user_id = ?", (username, full_name, user_id,))
+            self.cursor.execute("UPDATE our_users SET username = ?, full_name = ? WHERE user_id = ?",
+                                (username, full_name, user_id,))
             self.conn.commit()
-            self.logger.info(f"Функция update_username_and_full_name. Данные пользователя: user_id={user_id} успешно обновлены")
+            self.logger.info(
+                f"Функция update_username_and_full_name. Данные пользователя: user_id={user_id} успешно обновлены")
             return True
         except sqlite3.Error as error:
             self.logger.error(f"Ошибка в функции update_username_and_full_name: {error}")
@@ -823,18 +880,157 @@ class BotDB:
         finally:
             self.close()
 
+    def get_post_content_from_telegram_by_last_id(self, last_post_id: int):
+        self.logger.info(
+            f"Запуск функции get_post_content_from_telegram_by_last_id, идентификатор поста {last_post_id}")
+        try:
+            self.connect()
+            result = self.cursor.execute("""
+                                                SELECT cpft.content_name, cpft.content_type
+                                                FROM post_from_telegram_suggest pft
+                                                JOIN message_link_to_content mltc 
+                                                ON pft.message_id = mltc.post_id
+                                                JOIN content_post_from_telegram cpft 
+                                                ON cpft.message_id = mltc.message_id
+                                                WHERE pft.helper_text_message_id = ?
+                                                """, (last_post_id,))
+            post_content = result.fetchall()
+            self.logger.info(
+                f"Функция get_post_content_from_telegram_by_last_id получила список контента: {post_content}")
+            return post_content
+        finally:
+            self.close()
+
+    def get_post_ids_from_telegram_by_last_id(self, last_post_id: int):
+        self.logger.info(
+            f"Запуск функции get_post_ids_from_telegram_by_last_id, идентификатор поста {last_post_id}")
+        try:
+            self.connect()
+            result = self.cursor.execute("""
+                                                SELECT mltc.message_id
+                                                FROM post_from_telegram_suggest pft
+                                                JOIN message_link_to_content mltc 
+                                                ON pft.message_id = mltc.post_id
+                                                WHERE pft.helper_text_message_id = ?
+                                                """, (last_post_id,))
+            post_ids = result.fetchall()
+            self.logger.info(f"Функция get_post_ids_from_telegram_by_last_id "
+                             f"получила идентификаторы сообщений: {post_ids}")
+            return post_ids
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции get_post_ids_from_telegram_by_last_id {str(e)}")
+            return False
+        finally:
+            self.close()
+
+    def get_post_text_from_telegram_by_last_id(self, last_post_id: int):
+        self.logger.info(f"Запуск функции get_post_text_from_telegram_by_last_id, идентификатор поста {last_post_id}")
+        try:
+            self.connect()
+            result = self.cursor.execute("SELECT text "
+                                         "FROM post_from_telegram_suggest WHERE helper_text_message_id = ?",
+                                         (last_post_id,))
+            text = result.fetchone()[0]
+            self.logger.info(f"Функция get_post_text_from_telegram_by_last_id получила text")
+            return text
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции get_post_text_from_telegram_by_last_id {str(e)}")
+
+    def get_author_id_by_message_id(self, message_id: int):
+        self.logger.info(f"Запуск функции get_author_id_by_message_id, идентификатор поста {message_id}")
+        try:
+            self.connect()
+            result = self.cursor.execute("SELECT author_id "
+                                         "FROM post_from_telegram_suggest WHERE message_id = ?",
+                                         (message_id,))
+            author_id = result.fetchone()[0]
+            self.logger.info(f"Функция get_author_id_by_message_id получила author_id {author_id}")
+            return author_id
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции get_author_id_by_message_id {str(e)}")
+
+    def get_author_id_by_helper_message_id(self, helper_text_message_id: int):
+        self.logger.info(f"Запуск функции get_author_id_by_helper_message_id, идентификатор поста "
+                         f"{helper_text_message_id}")
+        try:
+            self.connect()
+            result = self.cursor.execute("SELECT author_id "
+                                         "FROM post_from_telegram_suggest WHERE helper_text_message_id = ?",
+                                         (helper_text_message_id,))
+            author_id = result.fetchone()[0]
+            self.logger.info(f"Функция get_author_id_by_helper_message_id получила author_id {author_id}")
+            return author_id
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции get_author_id_by_helper_message_id {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}, "
+            f"content_name={content_name}, content_type={type_content}")
+        try:
+            self.connect()
+            self.cursor.execute(
+                "INSERT INTO message_link_to_content (post_id, message_id)"
+                "VALUES (?, ?)", (post_id, message_id))
+            self.conn.commit()
+            self.cursor.execute(
+                "INSERT INTO content_post_from_telegram (message_id, content_name, content_type)"
+                "VALUES (?, ?, ?)", (message_id, content_name, type_content))
+            self.conn.commit()
+            self.logger.info(f"Функция add_post_content_in_db отработала успешно")
+            return True
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции add_post_content_in_db при добавлении поста в базу данных: {e}")
+            return False
+
+    def add_post_in_db(self, message_id: int, text: str, author_id: int):
+        self.logger.info(
+            f"Запуск функции add_post_in_db: message_id={message_id}, "
+            f"author_id={author_id}")
+        try:
+            today = datetime.now().strftime("%d-%m-%Y %H:%M:%S")
+            self.connect()
+            self.cursor.execute(
+                "INSERT INTO post_from_telegram_suggest (message_id, text, author_id, created_at)"
+                "VALUES (?, ?, ?, ?)", (message_id, text, author_id, today))
+            self.conn.commit()
+            self.logger.info(f"Функция add_post_in_db отработала успешно")
+            return True
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции add_post_in_db при добавлении поста в базу данных: {e}")
+            return False
+
+    def update_helper_message_in_db(self, message_id: int, helper_message_id: int):
+        self.logger.info(
+            f"Запуск функции update_helper_message_in_db: message_id={message_id}, "
+            f"helper_message_id={helper_message_id}")
+        try:
+            self.connect()
+            self.cursor.execute(
+                "UPDATE post_from_telegram_suggest SET helper_text_message_id = ? WHERE message_id = ?",
+                (helper_message_id, message_id,))
+            self.conn.commit()
+            self.logger.info(f"Функция update_helper_message_in_db отработала успешно")
+            return True
+        except Exception as e:
+            self.logger.error(f"Ошибка в функции update_helper_message_in_db при добавлении поста в базу данных: {e}")
+            return False
+
     def add_audio_record(self, file_name, author_id, date_added, listen_count, file_id):
         """Добавляет информацию о войсе юзера в БД"""
         self.logger.info(
-            f"Запуск функции add_audio_record (file_name = {file_name}, author_id = {author_id}, date_added = {date_added}")
+            f"Запуск функции add_audio_record (file_name = {file_name}, author_id = {author_id},"
+            f" date_added = {date_added}")
         try:
             self.connect()
             result = self.cursor.execute(
-                "INSERT INTO `audio_message_reference` (file_name, author_id, date_added, listen_count, file_id) VALUES (?, ?, ?, ?, ?)",
+                "INSERT INTO `audio_message_reference` (file_name, author_id, date_added, listen_count, file_id) "
+                "VALUES (?, ?, ?, ?, ?)",
                 (file_name, author_id, date_added, listen_count, file_id))
             self.conn.commit()
             self.logger.info(
-                f"Аудио успешно добавлено в БД (file_name = {file_name}, author_id = {author_id}, date_added = {date_added}")
+                f"Аудио успешно добавлено в БД (file_name = {file_name}, author_id = {author_id}, "
+                f"date_added = {date_added}")
             return None
         except sqlite3.Error as error:
             print(error)
@@ -871,7 +1067,8 @@ class BotDB:
         try:
             self.connect()
             result = self.cursor.execute(
-                "SELECT `file_id` FROM `audio_message_reference` WHERE `author_id` = ? ORDER BY date_added DESC LIMIT 1",
+                "SELECT `file_id` FROM `audio_message_reference` WHERE `author_id` = ? "
+                "ORDER BY date_added DESC LIMIT 1",
                 (user_id,))
             return result.fetchone()[0]
         except sqlite3.Error as error:
@@ -884,7 +1081,11 @@ class BotDB:
         try:
             self.connect()
             result = self.cursor.execute(
-                "SELECT `file_name` FROM `audio_message_reference` WHERE `author_id` = ? ORDER BY date_added DESC LIMIT 1",
+                "SELECT `file_name` "
+                "FROM `audio_message_reference` "
+                "WHERE `author_id` = ? "
+                "ORDER BY date_added "
+                "DESC LIMIT 1",
                 (user_id,))
             return result.fetchone()[0]
         except sqlite3.Error as error:

helper_bot/handlers/admin/admin_handlers.py

@@ -5,16 +5,14 @@ from aiogram.filters import Command, StateFilter
 from aiogram.fsm.context import FSMContext
 
 from helper_bot.filters.main import ChatTypeFilter
-from helper_bot.keyboards.main import get_reply_keyboard_admin, create_keyboard_with_pagination, \
-    create_keyboard_for_ban_days, create_keyboard_for_approve_ban
+from helper_bot.keyboards.keyboards import get_reply_keyboard_admin, create_keyboard_with_pagination, \
+    create_keyboard_for_ban_days, create_keyboard_for_approve_ban, create_keyboard_for_ban_reason
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 from helper_bot.utils.helper_func import check_access, add_days_to_date, get_banned_users_buttons, get_banned_users_list
 from logs.custom_logger import logger
 
-
 admin_router = Router()
 
-
 bdf = BaseDependencyFactory()
 GROUP_FOR_POST = bdf.settings['Telegram']['group_for_posts']
 GROUP_FOR_MESSAGE = bdf.settings['Telegram']['group_for_message']
@@ -62,6 +60,48 @@ async def get_last_users(message: types.Message):
                          reply_markup=keyboard)
 
 
+@admin_router.message(
+    ChatTypeFilter(chat_type=["private"]),
+    StateFilter("ADMIN"),
+    F.text == 'Бан по нику'
+)
+async def ban_by_nickname(message: types.Message, state: FSMContext):
+    await message.answer('Пришли мне username блокируемого пользователя')
+    await state.set_state('PRE_BAN')
+
+
+@admin_router.message(
+    ChatTypeFilter(chat_type=["private"]),
+    F.text == 'Отменить'
+)
+async def decline_ban(message: types.Message, state: FSMContext):
+    await state.set_data({})
+    await state.set_state("ADMIN")
+    logger.info(f"Отмена процедуры блокировки")
+    markup = get_reply_keyboard_admin()
+    await message.answer('Вернулись в меню', reply_markup=markup)
+
+
+@admin_router.message(
+    ChatTypeFilter(chat_type=["private"]),
+    StateFilter("PRE_BAN")
+)
+async def ban_by_nickname_step_2(message: types.Message, state: FSMContext):
+    logger.info(
+        f"Функция ban_by_nickname_2. Получен никнейм пользователя: {message.text}")
+    user_name = message.text
+    user_id = BotDB.get_user_id_by_username(user_name)
+    await state.update_data(user_id=user_id, user_name=user_name, message_for_user=None,
+                            date_to_unban=None)
+    full_name = BotDB.get_full_name_by_id(user_id)
+    markup = create_keyboard_for_ban_reason()
+    await message.answer(
+        text=f"<b>Выбран пользователь:\nid:</b> {user_id}\n<b>username:</b> {user_name}\n"
+             f"Имя:{full_name}\nВыбери причину бана из списка или напиши ее в чат",
+        reply_markup=markup)
+    await state.set_state('BAN_2')
+
+
 @admin_router.message(
     ChatTypeFilter(chat_type=["private"]),
     StateFilter("ADMIN"),
@@ -116,24 +156,24 @@ async def ban_user_step_3(message: types.Message, state: FSMContext):
 
 @admin_router.message(
     ChatTypeFilter(chat_type=["private"]),
-    StateFilter("BAN_FINAL")
+    StateFilter("BAN_FINAL"),
+    F.text == 'Подтвердить'
 )
 async def approve_ban(message: types.Message, state: FSMContext):
     user_data = await state.get_data()
     logger.info(f"Переход на финальный шаг бана пользователя. Словарь с данными для бана: {user_data})")
-    if message.text == 'Подтвердить':
-        exists = BotDB.check_user_in_blacklist(user_data['user_id'])
-        if exists:
-            await message.reply(f"Пользователь уже был заблокирован ранее.")
-            logger.info(f"Пользователь: {user_data['user_id']} был заблокирован ранее)")
-            await state.set_state('ADMIN')
-        else:
-            BotDB.set_user_blacklist(user_data['user_id'],
-                                     user_data['user_name'],
-                                     user_data['message_for_user'],
-                                     user_data['date_to_unban'])
-            await message.reply(f"Пользователь {user_data['user_name']} успешно заблокирован.")
-            logger.info(f"Пользователь: {user_data['user_id']} успешно заблокирован)")
-            await state.set_state('ADMIN')
+    exists = BotDB.check_user_in_blacklist(user_data['user_id'])
+    if exists:
+        await message.reply(f"Пользователь уже был заблокирован ранее.")
+        logger.info(f"Пользователь: {user_data['user_id']} был заблокирован ранее)")
+        await state.set_state('ADMIN')
+    else:
+        BotDB.set_user_blacklist(user_data['user_id'],
+                                 user_data['user_name'],
+                                 user_data['message_for_user'],
+                                 user_data['date_to_unban'])
+        await message.reply(f"Пользователь {user_data['user_name']} успешно заблокирован.")
+        logger.info(f"Пользователь: {user_data['user_id']} успешно заблокирован)")
+        await state.set_state('ADMIN')
     markup = get_reply_keyboard_admin()
     await message.answer('Вернулись в меню', reply_markup=markup)

helper_bot/handlers/callback/callback_handlers.py

@@ -4,11 +4,12 @@ from aiogram import Router, F
 from aiogram.fsm.context import FSMContext
 from aiogram.types import CallbackQuery
 
-from helper_bot.keyboards.main import create_keyboard_with_pagination, get_reply_keyboard_admin, \
+from helper_bot.keyboards.keyboards import create_keyboard_with_pagination, get_reply_keyboard_admin, \
     create_keyboard_for_ban_reason
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 from helper_bot.utils.helper_func import send_text_message, send_photo_message, get_banned_users_list, \
-    get_banned_users_buttons, delete_user_blacklist, get_help_message_id, send_media_group_message
+    get_banned_users_buttons, delete_user_blacklist, send_media_group_to_channel, \
+    send_video_message, send_video_note_message, send_audio_message, send_voice_message
 from logs.custom_logger import logger
 
 callback_router = Router()
@@ -31,10 +32,17 @@ BotDB = bdf.get_db()
 )
 async def post_for_group(call: CallbackQuery, state: FSMContext):
     logger.info(
-        f'Получен callback-запрос с данными: {call.data} от пользователя {call.from_user.full_name} (ID: {call.from_user.id})')
-    if call.data == 'publish' and call.message.content_type == 'text' and call.message.text != "^":
+        f'Получен callback-запрос с действием: {call.data} от пользователя {call.from_user.full_name} (ID сообщения: {call.message.message_id})')
+    if call.message.content_type == 'text' and call.message.text != "^":
         try:
+            # Пересылаем сообщение в канал
             await send_text_message(MAIN_PUBLIC, call.message, call.message.text)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был выложен🥰')
+
+            # Очищаем предложку и удаляем оттуда пост
             await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
             logger.info(f'Текст сообщения опубликован в канале {MAIN_PUBLIC}.')
             await call.answer(text='Выложено!', show_alert=True, cache_time=3)
@@ -43,9 +51,15 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
                                         text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
             logger.error(f'Ошибка при публикации текста в канал {MAIN_PUBLIC}: {str(e)}')
             await call.answer(text='Что-то пошло не так!', show_alert=True, cache_time=3)
-    elif call.data == 'publish' and call.message.content_type == 'photo':
+    elif call.message.content_type == 'photo':
         try:
             await send_photo_message(MAIN_PUBLIC, call.message, call.message.photo[-1].file_id, call.message.caption)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был выложен🥰')
+
+            # Удаляем пост из предложки
             await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
             logger.info(f'Пост с фото опубликован в канале {MAIN_PUBLIC}.')
             await call.answer(text='Выложено!', show_alert=True, cache_time=3)
@@ -54,15 +68,93 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
                                         text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
             logger.error(f'Ошибка при публикации фотографии в канал {MAIN_PUBLIC}: {str(e)}')
             await call.answer(text='Что-то пошло не так!', show_alert=True, cache_time=3)
-    elif call.data == 'publish' and call.message.text == "^":
-        user_data = await state.get_data()
-        media_group_message_id = get_help_message_id(call.message.message_id, user_data)
-        await call.bot.copy_message(chat_id=MAIN_PUBLIC, from_chat_id=GROUP_FOR_POST,message_id=media_group_message_id, reply_markup=None)
-        await call.bot.delete_message(chat_id=MAIN_PUBLIC, message_id=media_group_message_id)
-        print(user_data['media_group_message_id'])
-        print(user_data['help_message_id'])
+    elif call.message.content_type == 'video':
+        try:
+            await send_video_message(MAIN_PUBLIC, call.message, call.message.video.file_id, call.message.caption)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был выложен🥰')
+
+            await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
+            logger.info(f'Пост с видео опубликован в канале {MAIN_PUBLIC}.')
+            await call.answer(text='Выложено!', show_alert=True, cache_time=3)
+        except Exception as e:
+            await call.bot.send_message(chat_id=IMPORTANT_LOGS,
+                                        text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
+            logger.error(f'Ошибка при публикации видео в канал {MAIN_PUBLIC}: {str(e)}')
+            await call.answer(text='Что-то пошло не так!', show_alert=True, cache_time=3)
+    elif call.message.content_type == 'video_note':
+        try:
+            await send_video_note_message(MAIN_PUBLIC, call.message, call.message.video_note.file_id)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был выложен🥰')
+
+            await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
+            logger.info(f'Пост с кружком опубликован в канале {MAIN_PUBLIC}.')
+            await call.answer(text='Выложено!', show_alert=True, cache_time=3)
+        except Exception as e:
+            await call.bot.send_message(chat_id=IMPORTANT_LOGS,
+                                        text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
+            logger.error(f'Ошибка при публикации кружка в канал {MAIN_PUBLIC}: {str(e)}')
+            await call.answer(text='Что-то пошло не так!', show_alert=True, cache_time=3)
+    elif call.message.content_type == 'audio':
+        try:
+            await send_audio_message(MAIN_PUBLIC, call.message, call.message.audio.file_id, call.message.caption)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был выложен🥰')
+
+            await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
+            logger.info(f'Пост с аудио опубликован в канале {MAIN_PUBLIC}.')
+            await call.answer(text='Выложено!', show_alert=True, cache_time=3)
+        except Exception as e:
+            await call.bot.send_message(chat_id=IMPORTANT_LOGS,
+                                        text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
+            logger.error(f'Ошибка при публикации аудио в канал {MAIN_PUBLIC}: {str(e)}')
+            await call.answer(text='Что-то пошло не так!', show_alert=True, cache_time=3)
+    elif call.message.content_type == 'voice':
+        try:
+            await send_voice_message(MAIN_PUBLIC, call.message, call.message.voice.file_id)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был выложен🥰')
+
+            await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
+            logger.info(f'Пост с войсом опубликован в канале {MAIN_PUBLIC}.')
+            await call.answer(text='Выложено!', show_alert=True, cache_time=3)
+        except Exception as e:
+            await call.bot.send_message(chat_id=IMPORTANT_LOGS,
+                                        text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
+            logger.error(f'Ошибка при публикации войса в канал {MAIN_PUBLIC}: {str(e)}')
+            await call.answer(text='Что-то пошло не так!', show_alert=True, cache_time=3)
+    elif call.message.text == "^":
+        # Получаем контент медиагруппы и текст для публикации
+        post_content = BotDB.get_post_content_from_telegram_by_last_id(call.message.message_id)
+        post_text = BotDB.get_post_text_from_telegram_by_last_id(call.message.message_id)
+
+        # Готовим список для удаления
+        post_ids = BotDB.get_post_ids_from_telegram_by_last_id(call.message.message_id)
+        message_ids = [row[0] for row in post_ids]
+        message_ids.append(call.message.message_id)
+
+        # Выкладываем пост в канал
+        await send_media_group_to_channel(bot=call.bot, chat_id=MAIN_PUBLIC, post_content=post_content,
+                                          post_text=post_text)
+
+        # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+        author_id = BotDB.get_author_id_by_helper_message_id(call.message.message_id)
+        await send_text_message(author_id, call.message, 'Твой пост был выложен🥰')
+
+        # TODO: Удалить фотки с локалки после выкладки?
+        await call.bot.delete_messages(chat_id=GROUP_FOR_POST, message_ids=message_ids)
         await call.answer(text='Выложено!', show_alert=True, cache_time=3)
 
+
 @callback_router.callback_query(
     F.data == "decline"
 )
@@ -70,15 +162,30 @@ async def decline_post_for_group(call: CallbackQuery, state: FSMContext):
     logger.info(
         f'Получен callback-запрос с данными: {call.data} от пользователя {call.from_user.full_name} (ID: {call.from_user.id})')
     try:
-        if call.message.content_type == 'text' and call.message.text != "^" or call.message.content_type == 'photo':
+        if call.message.content_type == 'text' and call.message.text != "^" or call.message.content_type == 'photo' \
+                or call.message.content_type == 'audio' or call.message.content_type == 'voice' \
+                or call.message.content_type == 'video' or call.message.content_type == 'video_note':
             await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был отклонен😔')
+
             logger.info(
                 f'Сообщение отклонено админом {call.from_user.full_name} (ID: {call.from_user.id}).')
             await call.answer(text='Отклонено!', show_alert=True, cache_time=3)
         if call.message.text == '^':
-            user_data = await state.get_data()
-            media_group_message_id = get_help_message_id(call.message.message_id, user_data)
-            await call.bot.delete_message(chat_id=MAIN_PUBLIC, message_id=media_group_message_id)
+            post_ids = BotDB.get_post_ids_from_telegram_by_last_id(call.message.message_id)
+            message_ids = [row[0] for row in post_ids]
+            message_ids.append(call.message.message_id)
+
+            await call.bot.delete_messages(chat_id=GROUP_FOR_POST, message_ids=message_ids)
+
+            # Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
+            author_id = BotDB.get_author_id_by_helper_message_id(call.message.message_id)
+            await send_text_message(author_id, call.message, 'Твой пост был отклонен😔')
+
+            await call.answer(text='Удалено!', show_alert=True, cache_time=3)
     except Exception as e:
         await call.bot.send_message(IMPORTANT_LOGS,
                                     f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
@@ -139,19 +246,19 @@ async def change_page(call: CallbackQuery):
     logger.info(f"Переход на страницу {page_number}")
     if call.message.text == 'Список пользователей которые последними обращались к боту':
         list_users = BotDB.get_last_users_from_db()
-        #TODO: Здесь где-то надо добавить обработку ошибки IndexError: list index out of range
+        # TODO: Здесь где-то надо добавить обработку ошибки IndexError: list index out of range
         keyboard = create_keyboard_with_pagination(int(page_number), len(list_users), list_users,
                                                    'ban')
 
         await call.bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id,
                                                  reply_markup=keyboard)
     else:
-        #Готовим сообщения
+        # Готовим сообщения
         message_user = get_banned_users_list(int(page_number) * 7 - 7)
         await call.bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id,
                                          text=message_user)
 
-        #Готовим клавиатуру
+        # Готовим клавиатуру
         buttons = get_banned_users_buttons()
         keyboard = create_keyboard_with_pagination(int(call.data[5:]), len(buttons), buttons, 'unlock')
         await call.bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id,

helper_bot/handlers/group/group_handlers.py

@@ -2,14 +2,13 @@ from aiogram import Router, types
 from aiogram.fsm.context import FSMContext
 
 from helper_bot.filters.main import ChatTypeFilter
-from helper_bot.keyboards.main import get_reply_keyboard_leave_chat
+from helper_bot.keyboards.keyboards import get_reply_keyboard_leave_chat
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 from helper_bot.utils.helper_func import send_text_message
 from logs.custom_logger import logger
 
 group_router = Router()
 
-
 bdf = BaseDependencyFactory()
 GROUP_FOR_POST = bdf.settings['Telegram']['group_for_posts']
 GROUP_FOR_MESSAGE = bdf.settings['Telegram']['group_for_message']

helper_bot/handlers/private/private_handlers.py

@@ -11,13 +11,14 @@ from aiogram.types import FSInputFile
 
 from helper_bot.filters.main import ChatTypeFilter
 from helper_bot.keyboards import get_reply_keyboard, get_reply_keyboard_for_post
-from helper_bot.keyboards.main import get_reply_keyboard_leave_chat
+from helper_bot.keyboards.keyboards import get_reply_keyboard_leave_chat
 from helper_bot.middlewares.album_middleware import AlbumMiddleware
 from helper_bot.middlewares.blacklist_middleware import BlacklistMiddleware
 from helper_bot.utils import messages
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 from helper_bot.utils.helper_func import get_first_name, get_text_message, send_text_message, send_photo_message, \
-    process_photo_album, send_media_group_message, check_username_and_full_name
+    send_media_group_message_to_private_chat, prepare_media_group_from_middlewares, check_username_and_full_name, \
+    send_video_message, send_video_note_message, send_audio_message, send_voice_message, add_in_db_media
 from logs.custom_logger import logger
 
 private_router = Router()
@@ -64,8 +65,10 @@ async def handle_start_message(message: types.Message, state: FSMContext):
             is_need_update = check_username_and_full_name(user_id, username, full_name)
             if is_need_update:
                 BotDB.update_username_and_full_name(user_id, username, full_name)
-                await message.answer(f"Давно не виделись! Вижу что ты изменился;) Теперь буду звать тебя: {full_name} и ник @{username}")
-                await message.bot.send_message(chat_id=GROUP_FOR_LOGS, text=f'Для пользователя: {user_id} обновлены данные в БД.\nНовое имя: {full_name}\nНовый ник:{username}')
+                await message.answer(
+                    f"Давно не виделись! Вижу что ты изменился;) Теперь буду звать тебя: {full_name} и ник @{username}")
+                await message.bot.send_message(chat_id=GROUP_FOR_LOGS,
+                                               text=f'Для пользователя: {user_id} обновлены данные в БД.\nНовое имя: {full_name}\nНовый ник:{username}')
             sleep(1)
         BotDB.update_date_for_user(date, user_id)
         await state.set_state("START")
@@ -101,6 +104,10 @@ async def handle_start_message(message: types.Message, state: FSMContext):
 )
 async def suggest_post(message: types.Message, state: FSMContext):
     try:
+        user_id = message.from_user.id
+        current_date = datetime.now()
+        date = current_date.strftime("%Y-%m-%d %H:%M:%S")
+        BotDB.update_date_for_user(date, user_id)
         await message.forward(chat_id=GROUP_FOR_LOGS)
         await state.set_state("SUGGEST")
         current_state = await state.get_state()
@@ -127,6 +134,11 @@ async def suggest_post(message: types.Message, state: FSMContext):
 )
 async def end_message(message: types.Message, state: FSMContext):
     try:
+        user_id = message.from_user.id
+        current_date = datetime.now()
+        date = current_date.strftime("%Y-%m-%d %H:%M:%S")
+        BotDB.update_date_for_user(date, user_id)
+        await message.forward(chat_id=GROUP_FOR_LOGS)
         logger.info(
             f"Вызов функции end_message. Пользователь: {message.from_user.id} Имя автора сообщения: {message.from_user.full_name}")
         name_stick_bye = list(Path('Stick').rglob('Universal_*'))
@@ -158,56 +170,165 @@ async def suggest_router(message: types.Message, state: FSMContext, album: list
     logger.info(
         f"Вызов функции suggest_router. Пользователь: {message.from_user.id} Имя автора сообщения: {message.from_user.full_name}")
     try:
+        post_caption = ''
         if message.content_type == 'text':
             lower_text = message.text.lower()
-            post_text, is_anonymous = get_text_message(lower_text, message.from_user.full_name,
-                                                       message.from_user.username)
+            # Получаем текст сообщения и преобразовываем его по правилам
+            post_text = get_text_message(lower_text, message.from_user.full_name,
+                                         message.from_user.username)
+
+            # Получаем клавиатуру для поста
             markup = get_reply_keyboard_for_post()
-            if is_anonymous:
-                await send_text_message(GROUP_FOR_POST, message, post_text, markup)
-            else:
-                await send_text_message(GROUP_FOR_POST, message, post_text, markup)
+
+            # Отправляем сообщение в приватный канал
+            sent_message_id = await send_text_message(GROUP_FOR_POST, message, post_text, markup)
+
+            # Записываем в базу пост
+            BotDB.add_post_in_db(sent_message_id, message.text, message.from_user.id)
+
+            # Отправляем юзеру ответ, что сообщение отравлено и возвращаем его в меню
             markup_for_user = get_reply_keyboard(BotDB, message.from_user.id)
             success_send_message = messages.get_message(get_first_name(message), 'SUCCESS_SEND_MESSAGE')
             await message.answer(success_send_message, reply_markup=markup_for_user)
             await state.set_state("START")
+
         elif message.content_type == 'photo' and message.media_group_id is None:
-            lower_caption = message.caption.lower()
+            if message.caption:
+                lower_caption = message.caption.lower()
+                # Получаем текст сообщения и преобразовываем его по правилам
+                post_caption = get_text_message(lower_caption, message.from_user.full_name,
+                                                message.from_user.username)
             markup = get_reply_keyboard_for_post()
-            post_caption, is_anonymous = get_text_message(lower_caption, message.from_user.full_name,
-                                                          message.from_user.username)
-            #TODO: тут какая-то шляпа
-            if is_anonymous:
-                await send_photo_message(GROUP_FOR_POST, message,
-                                         message.photo[-1].file_id, post_caption, markup)
-            else:
-                await send_photo_message(GROUP_FOR_POST, message,
-                                         message.photo[-1].file_id, post_caption, markup)
+
+            # Отправляем фото и текст в приватный канал
+            sent_message = await send_photo_message(GROUP_FOR_POST, message,
+                                                    message.photo[-1].file_id, post_caption, markup)
+            # Записываем в базу пост и контент
+            BotDB.add_post_in_db(sent_message.message_id, sent_message.caption, message.from_user.id)
+            await add_in_db_media(sent_message)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
             markup_for_user = get_reply_keyboard(BotDB, message.from_user.id)
             success_send_message = messages.get_message(get_first_name(message), 'SUCCESS_SEND_MESSAGE')
             await message.answer(success_send_message, reply_markup=markup_for_user)
             await state.set_state("START")
+
+        elif message.content_type == 'video' and message.media_group_id is None:
+            if message.caption:
+                lower_caption = message.caption.lower()
+                post_caption = get_text_message(lower_caption, message.from_user.full_name,
+                                                message.from_user.username)
+            markup = get_reply_keyboard_for_post()
+            # Получаем текст сообщения и преобразовываем его по правилам
+
+            # Отправляем видео и текст в приватный канал
+            sent_message = await send_video_message(GROUP_FOR_POST, message,
+                                                    message.video.file_id, post_caption, markup)
+
+            # Записываем в базу пост и контент
+            BotDB.add_post_in_db(sent_message.message_id, sent_message.caption, message.from_user.id)
+            await add_in_db_media(sent_message)
+
+            # Записываем в базу пост и контент
+            BotDB.add_post_in_db(sent_message.message_id, sent_message.caption, message.from_user.id)
+            await add_in_db_media(sent_message)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            markup_for_user = get_reply_keyboard(BotDB, message.from_user.id)
+            success_send_message = messages.get_message(get_first_name(message), 'SUCCESS_SEND_MESSAGE')
+            await message.answer(success_send_message, reply_markup=markup_for_user)
+            await state.set_state("START")
+
+        elif message.content_type == 'video_note' and message.media_group_id is None:
+            markup = get_reply_keyboard_for_post()
+
+            # Отправляем видеокружок в приватный канал
+            sent_message = await send_video_note_message(GROUP_FOR_POST, message,
+                                                         message.video_note.file_id, markup)
+
+            # Записываем в базу пост и контент
+            BotDB.add_post_in_db(sent_message.message_id, sent_message.caption, message.from_user.id)
+            await add_in_db_media(sent_message)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            markup_for_user = get_reply_keyboard(BotDB, message.from_user.id)
+            success_send_message = messages.get_message(get_first_name(message), 'SUCCESS_SEND_MESSAGE')
+            await message.answer(success_send_message, reply_markup=markup_for_user)
+            await state.set_state("START")
+
+        elif message.content_type == 'audio' and message.media_group_id is None:
+            if message.caption:
+                lower_caption = message.caption.lower()
+                # Получаем текст сообщения и преобразовываем его по правилам
+                post_caption = get_text_message(lower_caption, message.from_user.full_name,
+                                                message.from_user.username)
+            markup = get_reply_keyboard_for_post()
+
+            # Отправляем аудио и текст в приватный канал
+            sent_message = await send_audio_message(GROUP_FOR_POST, message,
+                                                    message.audio.file_id, post_caption, markup)
+
+            # Записываем в базу пост и контент
+            BotDB.add_post_in_db(sent_message.message_id, sent_message.caption, message.from_user.id)
+            await add_in_db_media(sent_message)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            markup_for_user = get_reply_keyboard(BotDB, message.from_user.id)
+            success_send_message = messages.get_message(get_first_name(message), 'SUCCESS_SEND_MESSAGE')
+            await message.answer(success_send_message, reply_markup=markup_for_user)
+            await state.set_state("START")
+
+        elif message.content_type == 'voice' and message.media_group_id is None:
+            markup = get_reply_keyboard_for_post()
+
+            # Отправляем войс и текст в приватный канал
+            sent_message = await send_voice_message(GROUP_FOR_POST, message,
+                                                    message.voice.file_id, markup)
+
+            # Записываем в базу пост и контент
+            BotDB.add_post_in_db(sent_message.message_id, sent_message.caption, message.from_user.id)
+            await add_in_db_media(sent_message)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            markup_for_user = get_reply_keyboard(BotDB, message.from_user.id)
+            success_send_message = messages.get_message(get_first_name(message), 'SUCCESS_SEND_MESSAGE')
+            await message.answer(success_send_message, reply_markup=markup_for_user)
+            await state.set_state("START")
+
         elif message.media_group_id is not None:
             post_caption = " "
+
+            # Получаем сообщение и проверяем есть ли подпись. Если подпись есть, то преобразуем ее через функцию
             if album[0].caption:
                 lower_caption = album[0].caption.lower()
-                post_caption, is_anonymous = get_text_message(lower_caption, message.from_user.full_name,
-                                                              message.from_user.username)
-            media_group = process_photo_album(album, post_caption)
-            media_group_message_id = await send_media_group_message(GROUP_FOR_POST, message,
-                                           media_group)
+                post_caption = get_text_message(lower_caption, message.from_user.full_name,
+                                                message.from_user.username)
+
+            # Иначе обрабатываем фото и получаем медиагруппу
+            media_group = await prepare_media_group_from_middlewares(album, post_caption)
+
+            # Отправляем медиагруппу в секретный чат
+            media_group_message_id = await send_media_group_message_to_private_chat(GROUP_FOR_POST, message,
+                                                                                    media_group)
             sleep(0.2)
+
+            # Получаем клавиатуру и отправляем еще одно текстовое сообщение с кнопками
             markup = get_reply_keyboard_for_post()
             help_message_id = await send_text_message(GROUP_FOR_POST, message, "^", markup)
-            await state.update_data(media_group_message_id=media_group_message_id, help_message_id=help_message_id)
 
+            # Записываем в state идентификаторы текстового сообщения И последнего сообщения медиагруппы
+            BotDB.update_helper_message_in_db(message_id=media_group_message_id, helper_message_id=help_message_id)
+
+            # Получаем клавиатуру для пользователя, благодарим за пост, и возвращаем в дефолтное сообщение
             markup_for_user = get_reply_keyboard(BotDB, message.from_user.id)
             success_send_message = messages.get_message(get_first_name(message), 'SUCCESS_SEND_MESSAGE')
             await message.answer(success_send_message, reply_markup=markup_for_user)
             await state.set_state("START")
         else:
             await message.bot.send_message(message.chat.id,
-                                           'Я пока не умею работать с таким сообщением. Пришли текст и фото/фоты(ы)')
+                                           'Я пока не умею работать с таким сообщением. '
+                                           'Пришли текст и фото/фоты(ы). А лучше перешли это сообщение админу @kerrad1\n'
+                                           'Мы добавим его к обработке если необходимо')
     except Exception as e:
         await message.bot.send_message(chat_id=IMPORTANT_LOGS,
                                        text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
@@ -242,6 +363,10 @@ async def stickers(message: types.Message, state: FSMContext):
 async def connect_with_admin(message: types.Message, state: FSMContext):
     logger.info(
         f"Вызов функции connect_with_admin. Пользователь: {message.from_user.id} Имя автора сообщения: {message.from_user.full_name}")
+    user_id = message.from_user.id
+    current_date = datetime.now()
+    date = current_date.strftime("%Y-%m-%d %H:%M:%S")
+    BotDB.update_date_for_user(date, user_id)
     admin_message = messages.get_message(get_first_name(message), 'CONNECT_WITH_ADMIN')
     await message.answer(admin_message, parse_mode="html")
     await message.forward(chat_id=GROUP_FOR_LOGS)
@@ -257,6 +382,10 @@ async def connect_with_admin(message: types.Message, state: FSMContext):
     ChatTypeFilter(chat_type=["private"]),
 )
 async def resend_message_in_group_for_message(message: types.Message, state: FSMContext):
+    user_id = message.from_user.id
+    current_date = datetime.now()
+    date = current_date.strftime("%Y-%m-%d %H:%M:%S")
+    BotDB.update_date_for_user(date, user_id)
     logger.info(
         f"Попытка пересылки сообщения в связь с админами. Сообщение: {message.text} Имя автора сообщения: {message.from_user.full_name} Идентификатор сообщения: {message.message_id})")
     await message.forward(chat_id=GROUP_FOR_MESSAGE)
@@ -272,11 +401,3 @@ async def resend_message_in_group_for_message(message: types.Message, state: FSM
     elif user_state == "CHAT":
         markup = get_reply_keyboard_leave_chat()
         await message.answer(question, reply_markup=markup)
-
-# @private_router.message(
-#     ChatTypeFilter(chat_type=["private"])
-# )
-# async def default(message: types.Message, state: FSMContext):
-#     markup = get_reply_keyboard(BotDB, message.from_user.id)
-#     await message.answer('Кажется ты заблудился. Держи клавиатуру, твое состояние сброшено на начало', reply_markup=markup)
-#     await state.set_state("START")

helper_bot/keyboards/__init__.py

@@ -1 +1 @@
-from .main import get_reply_keyboard_for_post, get_reply_keyboard
+from .keyboards import get_reply_keyboard_for_post, get_reply_keyboard

helper_bot/keyboards/keyboards.py

@@ -5,9 +5,10 @@ from aiogram.utils.keyboard import ReplyKeyboardBuilder, InlineKeyboardBuilder
 def get_reply_keyboard_for_post():
     builder = InlineKeyboardBuilder()
     builder.row(types.InlineKeyboardButton(
-        text="Опубликовать", callback_data="publish"),
-        types.InlineKeyboardButton(
-            text="Отклонить", callback_data="decline")
+        text="Опубликовать", callback_data="publish")
+    )
+    builder.row(types.InlineKeyboardButton(
+        text="Отклонить", callback_data="decline")
     )
     markup = builder.as_markup(resize_keyboard=True, one_time_keyboard=True)
     return markup
@@ -34,6 +35,7 @@ def get_reply_keyboard_leave_chat():
 def get_reply_keyboard_admin():
     builder = ReplyKeyboardBuilder()
     builder.add(types.KeyboardButton(text="Бан (Список)"))
+    builder.add(types.KeyboardButton(text="Бан по нику"))
     builder.add(types.KeyboardButton(text="Разбан (список)"))
     builder.add(types.KeyboardButton(text="Вернуться в бота"))
     markup = builder.as_markup(resize_keyboard=True, one_time_keyboard=True)
@@ -59,7 +61,6 @@ def create_keyboard_with_pagination(page: int, total_items: int, array_items: li
 
     # Создаем билдер для клавиатуры
     keyboard = InlineKeyboardBuilder()
-    # TODO: Тут поправить на 9 объектов, а не 7. Клавиатуру переделал
     # Вычисляем стартовый номер для текущей страницы
     start_index = (page - 1) * 9
 
@@ -88,6 +89,9 @@ def create_keyboard_for_ban_reason():
     builder = ReplyKeyboardBuilder()
     builder.add(types.KeyboardButton(text="Спам"))
     builder.add(types.KeyboardButton(text="Заебал стикерами"))
+    builder.row(types.KeyboardButton(text="Реклама здесь: @kerrad1 "))
+    builder.row(types.KeyboardButton(text="Тема с лагерями: https://vk.com/topic-75343895_50049913"))
+    builder.row(types.KeyboardButton(text="Отменить"))
     markup = builder.as_markup(resize_keyboard=True, one_time_keyboard=True)
     return markup
 
@@ -97,7 +101,8 @@ def create_keyboard_for_ban_days():
     builder.add(types.KeyboardButton(text="1"))
     builder.add(types.KeyboardButton(text="7"))
     builder.add(types.KeyboardButton(text="30"))
-    builder.add(types.KeyboardButton(text="Навсегда"))
+    builder.row(types.KeyboardButton(text="Навсегда"))
+    builder.row(types.KeyboardButton(text="Отменить"))
     markup = builder.as_markup(resize_keyboard=True, one_time_keyboard=True)
     return markup
 

helper_bot/utils/base_dependency_factory.py

@@ -4,6 +4,8 @@ import sys
 
 from database.db import BotDB
 
+current_dir = os.getcwd()
+
 
 class BaseDependencyFactory:
     def __init__(self):
@@ -12,7 +14,7 @@ class BaseDependencyFactory:
         self.config = configparser.ConfigParser()
         self.config.read(config_path)
         self.settings = {}
-        self.database = BotDB('database/tg-bot-database')
+        self.database = BotDB(current_dir, 'database/tg-bot-database')
 
         for section in self.config.sections():
             self.settings[section] = {}

helper_bot/utils/helper_func.py

@@ -1,9 +1,11 @@
+import os
 from datetime import datetime, timedelta
 
 from aiogram import types
-from aiogram.types import InputMediaPhoto
+from aiogram.types import InputMediaPhoto, FSInputFile, InputMediaVideo, InputMediaAudio
 
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
+from logs.custom_logger import logger
 
 bdf = BaseDependencyFactory()
 
@@ -24,50 +26,187 @@ def get_text_message(post_text: str, first_name: str, username: str):
         username: Юзернейм автора поста
 
     Returns:
-        Кортеж из двух элементов:
-            - Сформированный текст сообщения.
-            - Флаг, указывающий, является ли пост анонимным (True - анонимный, False - не анонимный).
+        str: - Сформированный текст сообщения.
     """
     if "неанон" in post_text or "не анон" in post_text:
-        is_anonymous = False
-        return f'Пост из ТГ:\n{post_text}\n\nАвтор поста: {first_name} @{username}', is_anonymous
+        return f'Пост из ТГ:\n{post_text}\n\nАвтор поста: {first_name} @{username}'
     elif "анон" in post_text:
-        is_anonymous = True
-        return f'Пост из ТГ:\n{post_text}\n\nПост опубликован анонимно', is_anonymous
+        return f'Пост из ТГ:\n{post_text}\n\nПост опубликован анонимно'
     else:
-        is_anonymous = False
-        return f'Пост из ТГ:\n{post_text}\n\nАвтор поста: {first_name} @{username}', is_anonymous
+        return f'Пост из ТГ:\n{post_text}\n\nАвтор поста: {first_name} @{username}'
 
 
-def process_photo_album(album, post_caption: str = ''):
+async def download_file(message: types.Message, file_id: str):
     """
-    Создает список InputMediaPhoto для альбома.
+    Скачивает файл по file_id из Telegram.
+
+    Args:
+        message: сообщение
+        file_id:  File ID фотографии
+        filename: Имя файла, под которым будет сохранено фото
+
+    Returns:
+        Путь к сохраненному файлу, если файл был скачан успешно, иначе None
+    """
+    try:
+        os.makedirs("files", exist_ok=True)
+        os.makedirs("files/photos", exist_ok=True)
+        os.makedirs("files/videos", exist_ok=True)
+        os.makedirs("files/music", exist_ok=True)
+        os.makedirs("files/voice", exist_ok=True)
+        os.makedirs("files/video_notes", exist_ok=True)
+        file = await message.bot.get_file(file_id)
+        file_path = os.path.join("files", file.file_path)
+        await message.bot.download_file(file_path=file.file_path, destination=file_path)
+        return file_path
+    except Exception as e:
+        logger.error(f"Ошибка скачивания фотографии: {e}")
+        return None
+
+
+async def prepare_media_group_from_middlewares(album, post_caption: str = ''):
+    """
+    Создает MediaGroup.
 
     Args:
         album: Album объект из Telegram API.
         post_caption: Текст подписи к первому фото.
 
     Returns:
-        Список InputMediaPhoto.
+        Список InputMediaPhoto (MediaGroup).
     """
-    photo_media = []
+    media_group = []
+
     for i, message in enumerate(album):
-        if i == 0:
-            photo_media.append(InputMediaPhoto(media=message.photo[-1].file_id, caption=post_caption))
+        if message.photo:
+            file_id = message.photo[-1].file_id
+            media_type = 'photo'
+        elif message.video:
+            file_id = message.video.file_id
+            media_type = 'video'
+        elif message.audio:
+            file_id = message.audio.file_id
+            media_type = 'audio'
         else:
-            photo_media.append(InputMediaPhoto(media=message.photo[-1].file_id))
-    return photo_media
+            # Если нет фото, видео или аудио, пропускаем сообщение
+            continue
+
+        # Формируем объект MediaGroup с учетом типа медиа
+        if i == len(album) - 1:
+            if media_type == 'photo':
+                media_group.append(InputMediaPhoto(media=file_id, caption=post_caption))
+            elif media_type == 'video':
+                media_group.append(InputMediaVideo(media=file_id, caption=post_caption))
+            elif media_type == 'audio':
+                media_group.append(InputMediaAudio(media=file_id, caption=post_caption))
+        else:
+            if media_type == 'photo':
+                media_group.append(InputMediaPhoto(media=file_id))
+            elif media_type == 'video':
+                media_group.append(InputMediaVideo(media=file_id))
+            elif media_type == 'audio':
+                media_group.append(InputMediaAudio(media=file_id))
+
+    return media_group  # Возвращаем MediaGroup
 
 
-async def send_media_group_message(chat_id: int, message: types.Message, media_group: list[InputMediaPhoto]):
+async def add_in_db_media_mediagroup(sent_message):
+    """
+    Идентификатор медиа-группы
+
+    Args:
+        sent_message: sent_message объект из Telegram API
+
+    Returns:
+        Список InputFile (FSInputFile).
+    """
+    media_group_message_id = sent_message[-1].message_id  # Получаем идентификатор медиа-группы
+    for i, message in enumerate(sent_message):
+        if message.photo:
+            file_id = message.photo[-1].file_id
+            file_path = await download_file(message, file_id=file_id)
+            BotDB.add_post_content_in_db(media_group_message_id, message.message_id, file_path, 'photo')
+        elif message.video:
+            file_id = message.video.file_id
+            file_path = await download_file(message, file_id=file_id)
+            BotDB.add_post_content_in_db(media_group_message_id, message.message_id, file_path, 'video')
+        else:
+            # Если нет фото, видео или аудио, или другой контент, пропускаем сообщение
+            continue
+
+
+async def add_in_db_media(sent_message):
+    """
+    Args:
+        sent_message: sent_message объект из Telegram API
+
+    Returns:
+        Список InputFile (FSInputFile).
+    """
+    if sent_message.photo:
+        file_id = sent_message.photo[-1].file_id
+        file_path = await download_file(sent_message, file_id=file_id)
+        BotDB.add_post_content_in_db(sent_message.message_id, sent_message.message_id, file_path, 'photo')
+    elif sent_message.video:
+        file_id = sent_message.video.file_id
+        file_path = await download_file(sent_message, file_id=file_id)
+        BotDB.add_post_content_in_db(sent_message.message_id, sent_message.message_id, file_path, 'video')
+    elif sent_message.voice:
+        file_id = sent_message.voice.file_id
+        file_path = await download_file(sent_message, file_id=file_id)
+        BotDB.add_post_content_in_db(sent_message.message_id, sent_message.message_id, file_path, 'voice')
+    elif sent_message.audio:
+        file_id = sent_message.audio.file_id
+        file_path = await download_file(sent_message, file_id=file_id)
+        BotDB.add_post_content_in_db(sent_message.message_id, sent_message.message_id, file_path, 'audio')
+    elif sent_message.video_note:
+        file_id = sent_message.video_note.file_id
+        file_path = await download_file(sent_message, file_id=file_id)
+        BotDB.add_post_content_in_db(sent_message.message_id, sent_message.message_id, file_path, 'video_note')
+
+
+async def send_media_group_message_to_private_chat(chat_id: int, message: types.Message,
+                                                   media_group: list[InputMediaPhoto]):
     sent_message = await message.bot.send_media_group(
         chat_id=chat_id,
         media=media_group,
     )
+    BotDB.add_post_in_db(sent_message[-1].message_id, sent_message[-1].caption, message.from_user.id)
+    await add_in_db_media_mediagroup(sent_message)
     message_id = sent_message[-1].message_id
     return message_id
 
 
+async def send_media_group_to_channel(bot, chat_id: int, post_content: list[tuple[str]], post_text: str):
+    """
+    Отправляет медиа-группу с подписью к последнему файлу.
+
+    Args:
+        bot: Экземпляр бота aiogram.
+        chat_id: ID чата для отправки.
+        post_content: Список кортежей с путями к файлам.
+        post_text: Текст подписи.
+    """
+    media = []
+    for file_path in post_content:
+        try:
+            file = FSInputFile(path=file_path[0])
+            type = file_path[1]
+            if type == 'video':
+                media.append(types.InputMediaVideo(media=file))
+            if type == 'photo':
+                media.append(types.InputMediaPhoto(media=file))
+        except FileNotFoundError:
+            logger.error(f"Файл не найден: {file_path[0]}")
+            return
+
+    # Добавляем подпись к последнему файлу
+    if media:
+        media[-1].caption = post_text
+
+    await bot.send_media_group(chat_id=chat_id, media=media)
+
+
 async def send_text_message(chat_id, message: types.Message, post_text: str, markup: types.ReplyKeyboardMarkup = None):
     if markup is None:
         sent_message = await message.bot.send_message(
@@ -86,20 +225,90 @@ async def send_text_message(chat_id, message: types.Message, post_text: str, mar
         return message_id
 
 
-async def send_photo_message(chat_id, message: types.Message, photo: str, post_text: str, markup: types.ReplyKeyboardMarkup = None):
+async def send_photo_message(chat_id, message: types.Message, photo: str, post_text: str,
+                             markup: types.ReplyKeyboardMarkup = None):
     if markup is None:
-        await message.bot.send_photo(
+        sent_message = await message.bot.send_photo(
             chat_id=chat_id,
             caption=post_text,
             photo=photo
         )
     else:
-        await message.bot.send_photo(
+        sent_message = await message.bot.send_photo(
             chat_id=chat_id,
             caption=post_text,
             photo=photo,
             reply_markup=markup
         )
+    return sent_message
+
+
+async def send_video_message(chat_id, message: types.Message, video: str, post_text: str = "",
+                             markup: types.ReplyKeyboardMarkup = None):
+    if markup is None:
+        sent_message = await message.bot.send_video(
+            chat_id=chat_id,
+            caption=post_text,
+            video=video
+        )
+    else:
+        sent_message = await message.bot.send_video(
+            chat_id=chat_id,
+            caption=post_text,
+            video=video,
+            reply_markup=markup
+        )
+    return sent_message
+
+
+async def send_video_note_message(chat_id, message: types.Message, video_note: str,
+                                  markup: types.ReplyKeyboardMarkup = None):
+    if markup is None:
+        sent_message = await message.bot.send_video_note(
+            chat_id=chat_id,
+            video_note=video_note
+        )
+    else:
+        sent_message = await message.bot.send_video_note(
+            chat_id=chat_id,
+            video_note=video_note,
+            reply_markup=markup
+        )
+    return sent_message
+
+
+async def send_audio_message(chat_id, message: types.Message, audio: str, post_text: str,
+                             markup: types.ReplyKeyboardMarkup = None):
+    if markup is None:
+        sent_message = await message.bot.send_audio(
+            chat_id=chat_id,
+            caption=post_text,
+            audio=audio
+        )
+    else:
+        sent_message = await message.bot.send_audio(
+            chat_id=chat_id,
+            caption=post_text,
+            audio=audio,
+            reply_markup=markup
+        )
+    return sent_message
+
+
+async def send_voice_message(chat_id, message: types.Message, voice: str,
+                             markup: types.ReplyKeyboardMarkup = None):
+    if markup is None:
+        sent_message = await message.bot.send_voice(
+            chat_id=chat_id,
+            voice=voice
+        )
+    else:
+        sent_message = await message.bot.send_voice(
+            chat_id=chat_id,
+            voice=voice,
+            reply_markup=markup
+        )
+    return sent_message
 
 
 def check_access(user_id: int):
@@ -155,24 +364,6 @@ def get_banned_users_buttons():
     return user_ids
 
 
-def get_help_message_id(media_group_message_id: int, data: dict) -> int:
-    """
-    Получает идентификатор сообщения помощи по идентификатору сообщения группы.
-
-    Args:
-      media_group_message_id: Идентификатор сообщения группы
-      data: Словарь с данными.
-
-    Returns:
-      Идентификатор сообщения помощи.
-    """
-
-    if 'help_message_id' in data and 'media_group_message_id' in data:
-        return data['media_group_message_id']
-    else:
-        return 0
-
-
 def delete_user_blacklist(user_id: int):
     return BotDB.delete_user_blacklist(user_id=user_id)
 
@@ -185,7 +376,6 @@ def check_username_and_full_name(user_id: int, username: str, full_name: str):
 def unban_notifier(self):
     # Получение сегодняшней даты в формате DD-MM-YYYY
     current_date = datetime.now()
-    print('Мы в функции unban_notifier')
     today = current_date.strftime("%d-%m-%Y")
     # Получение списка разблокированных пользователей
     unblocked_users = self.BotDB.get_users_for_unblock_today(today)

helper_bot/utils/state.py

@@ -7,6 +7,7 @@ class StateUser(StatesGroup):
     ADMIN = State()
     CHAT = State()
     PRE_CHAT = State()
+    PRE_BAN = State()
     BAN_2 = State()
     BAN_3 = State()
     BAN_4 = State()

logs/custom_logger.py

@@ -1,7 +1,7 @@
 import datetime
 import os
-from loguru import logger
 
+from loguru import logger
 
 logger = logger.bind(name='main_log')
 
@@ -20,6 +20,5 @@ logger.add(
     filename,
     rotation="00:00",
     retention="5 days",
-    compression="zip",
     format="{time:YYYY-MM-DD at HH:mm:ss} | {level} | {name} | {line} | {message}",
 )

migrations/000_migrations_init.py

@@ -1,17 +1,14 @@
 import os
-from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 
-# Получаем текущий рабочий каталог
-current_dir = os.path.dirname(os.path.abspath(__file__))
+from database.db import BotDB
 
-# Переходим на уровень выше, чтобы выйти из папки migrations/
+# Получаем текущую директорию
+current_dir = os.path.dirname(__file__)
+
+# Переходим на уровень выше
 parent_dir = os.path.dirname(current_dir)
 
-# Строим путь до файла
-tg_bot_database_path = os.path.join(parent_dir, "tg-bot-database")
-
-bdf = BaseDependencyFactory()
-BotDB = bdf.get_db()
+BotDB = BotDB(parent_dir, 'database/tg-bot-database')
 
 
 def get_filename():
@@ -32,5 +29,6 @@ def main():
     BotDB.create_table(migrations_init)
     BotDB.update_version(0, get_filename())
 
+
 if __name__ == "__main__":
     main()

migrations/001_create_new_tables.py

@@ -1,8 +1,14 @@
 import os
-from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 
-bdf = BaseDependencyFactory()
-BotDB = bdf.get_db()
+from database.db import BotDB
+
+# Получаем текущую директорию
+current_dir = os.path.dirname(__file__)
+
+# Переходим на уровень выше
+parent_dir = os.path.dirname(current_dir)
+
+BotDB = BotDB(parent_dir, 'database/tg-bot-database')
 
 
 def get_filename():

migrations/002_create_tables_media_group.py

@@ -0,0 +1,61 @@
+import os
+
+from database.db import BotDB
+
+# Получаем текущую директорию
+current_dir = os.path.dirname(__file__)
+
+# Переходим на уровень выше
+parent_dir = os.path.dirname(current_dir)
+
+BotDB = BotDB(parent_dir, 'database/tg-bot-database')
+
+
+def get_filename():
+    """Возвращает имя файла без расширения."""
+    filename = os.path.basename(__file__)
+    filename = os.path.splitext(filename)[0]
+    return filename
+
+
+def main():
+    # Проверка версии миграций
+    current_version = BotDB.get_current_version()  # Добавьте функцию для получения версии
+
+    # Выполнение миграций и проверка последней версии
+    if current_version < 2:
+        # Скрипты миграции
+        create_table_sql_1 = """
+            CREATE TABLE IF NOT EXISTS "post_from_telegram_suggest"
+                (
+                    message_id             INTEGER not null,
+                    text                   TEXT,
+                    helper_text_message_id INTEGER,
+                    author_id              INTEGER,
+                    created_at             TEXT
+                );
+            """
+        create_table_sql_2 = """
+                    CREATE TABLE IF NOT EXISTS message_link_to_content (
+                    post_id INTEGER NOT NULL,
+                    message_id INTEGER NOT NULL
+                );
+            """
+        create_table_sql_3 = """
+            CREATE TABLE IF NOT EXISTS content_post_from_telegram (
+                    message_id INTEGER NOT NULL,
+                    content_name TEXT NOT NULL,
+                    content_type TEXT
+                );
+        """
+        # Применение миграции
+        BotDB.create_table(create_table_sql_1)
+        BotDB.create_table(create_table_sql_2)
+        BotDB.create_table(create_table_sql_3)
+        filename = get_filename()
+
+        BotDB.update_version(2, filename)
+
+
+if __name__ == "__main__":
+    main()

run_helper.py

@@ -1,7 +1,7 @@
 import asyncio
+
 from helper_bot.main import start_bot
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 
-
 if __name__ == '__main__':
     asyncio.run(start_bot(BaseDependencyFactory()))

tests/test_db.py

@@ -1,13 +1,17 @@
-from datetime import datetime
 import os
 import sqlite3
+from datetime import datetime
+
 import pytest
+
 from database.db import BotDB
 
+
 @pytest.fixture
 def bot():
     """Фикстура для создания объекта BotDB."""
-    return BotDB("test.db")
+    current_dir = os.getcwd()
+    return BotDB(current_dir, "test.db")
 
 
 @pytest.fixture(autouse=True, )
@@ -104,12 +108,12 @@ def setup_db():
         );
     """)
 
-    #blacklist mock data
+    # blacklist mock data
     cursor.execute("INSERT INTO blacklist (user_id, user_name, message_for_user, date_to_unban) VALUES (?, ?, ?, ?)",
                    (user_id, username, message_for_user, next_date))
     cursor.execute("INSERT INTO blacklist (user_id, user_name, message_for_user, date_to_unban) VALUES (?, ?, ?, ?)",
                    (user_id_2, username_2, message_for_user_2, date))
-    #our_users mock data
+    # our_users mock data
     cursor.execute(
         "INSERT INTO our_users (user_id, first_name, full_name, username, date_added, date_changed, has_stickers)"
         " VALUES (?, ?, ?, ?, ?, ?, ?)", (user_id, first_name, full_name, username, date, date, has_stickers)
@@ -118,7 +122,7 @@ def setup_db():
         "INSERT INTO our_users (user_id, first_name, full_name, username, date_added, date_changed, has_stickers)"
         " VALUES (?, ?, ?, ?, ?, ?, ?)", (user_id_2, first_name_2, full_name_2, username_2, date, date, has_stickers_2)
     )
-    #messages mock data
+    # messages mock data
     cursor.execute(
         "INSERT INTO user_messages (message_text, user_id, message_id, date) "
         "VALUES (?, ?, ?, ?)",
@@ -127,7 +131,7 @@ def setup_db():
         "INSERT INTO user_messages (message_text, user_id, message_id, date) "
         "VALUES (?, ?, ?, ?)",
         (message_text_2, user_id_2, message_id_2, date))
-    #mock admins
+    # mock admins
     cursor.execute(
         "INSERT INTO admins (user_id, role) "
         "VALUES (?, ?)",