add mediaGroup func

helper_bot/handlers/callback/callback_handlers.py

@@ -8,7 +8,7 @@ from helper_bot.keyboards.main import create_keyboard_with_pagination, get_reply
     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, get_help_message_id, send_media_group_with_caption
 from logs.custom_logger import logger
 
 callback_router = Router()
@@ -31,8 +31,8 @@ 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)
             await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
@@ -43,7 +43,7 @@ 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)
             await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
@@ -54,18 +54,18 @@ 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 == "^":
-        print('Попали куда надо')
+    elif call.message.text == "^":
         user_data = await state.get_data()
-        print(f'В CALLBACK ГЕТ ДАТА: {user_data}')
-        media_group_message_id = get_help_message_id(call.message.message_id, user_data)
-        print(f'ПОЛУЧАЮ МЕССАГ ГРУПП АЙДИ ИЗ ГЕТ ДАТЫ: {media_group_message_id}')
-        print(f'ИДЕНТИФИКАТОР СООБЩЕНИЯ У КОТОРОГО ЖМУ КОЛЛБЭК: {call.message.message_id}')
-        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)
+        # Получил идентификатор последнего предложенного сообщения из кэша
+        help_message = get_help_message_id(call.message.message_id, user_data)
+        media_group_id = help_message-1
+        post_content = BotDB.get_post_content_from_telegram_by_last_id(media_group_id)
+        post_text = BotDB.get_post_text_from_telegram_by_last_id(media_group_id)
+        await send_media_group_with_caption(bot=call.bot, chat_id=MAIN_PUBLIC, post_content=post_content, post_text=post_text)
         await call.answer(text='Выложено!', show_alert=True, cache_time=3)
 
 
+
 @callback_router.callback_query(
     F.data == "decline"
 )

helper_bot/handlers/private/private_handlers.py

@@ -17,9 +17,10 @@ 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, process_photo_album, check_username_and_full_name
 from logs.custom_logger import logger
 
+
 private_router = Router()
 
 private_router.message.middleware(AlbumMiddleware())
@@ -189,20 +190,28 @@ async def suggest_router(message: types.Message, state: FSMContext, album: list
             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)
-            print(f'Отправил в чат предложки сообщения, media_group_message_id:{media_group_message_id}\n\n')
+
+            #Иначе обрабатываем фото и получаем медиагруппу
+            media_group = await process_photo_album(album, post_caption)
+
+            #Отправляем медиагруппу в секретный чат
+            media_group_message_id = await send_media_group_message(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)
+
+            #Записываем в state идентификаторы текстового сообщения И последнего сообщения медиагруппы
             await state.update_data(media_group_message_id=media_group_message_id, help_message_id=help_message_id)
-            d = await state.get_data()
-            print(f'ЗАПИСАЛ В state.update_data {d}')
+
+            #Получаем клавиатуру для пользователя, благодарим за пост, и возвращаем в дефолтное сообщение
             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)

helper_bot/utils/helper_func.py

@@ -1,9 +1,10 @@
 from datetime import datetime, timedelta
 
 from aiogram import types
-from aiogram.types import InputMediaPhoto
+from aiogram.types import InputMediaPhoto, FSInputFile
 
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
+from logs.custom_logger import logger
 
 bdf = BaseDependencyFactory()
 
@@ -39,7 +40,29 @@ def get_text_message(post_text: str, first_name: str, username: str):
         return f'Пост из ТГ:\n{post_text}\n\nАвтор поста: {first_name} @{username}', is_anonymous
 
 
-def process_photo_album(album, post_caption: str = ''):
+async def download_photo(message: types.Message, file_id: str):
+    """
+    Скачивает фото по file_id из Telegram.
+
+    Args:
+        message: сообщение
+        file_id:  File ID фотографии.
+        filename: Имя файла, под которым будет сохранено фото.
+
+    Returns:
+        Путь к сохраненному файлу, если файл был скачан успешно, иначе None.
+    """
+    try:
+        file = await message.bot.get_file(file_id)
+        file_path = file.file_path
+        await message.bot.download_file(file_path=file_path, destination=file_path)
+        return file_path
+    except Exception as e:
+        logger.error(f"Ошибка скачивания фотографии: {e}")
+        return None
+
+
+async def process_photo_album(album, post_caption: str = ''):
     """
     Создает список InputMediaPhoto для альбома.
 
@@ -51,13 +74,41 @@ def process_photo_album(album, post_caption: str = ''):
         Список InputMediaPhoto.
     """
     photo_media = []
+    #Циклом проходимся по собранному миддлварью объекту album
     for i, message in enumerate(album):
-        print(f'message_id в функции process_photo_album: {message.message_id}\n\n')
-        if i == 0:
+        file_id = message.photo[-1].file_id
+        file_path = await download_photo(message, file_id=file_id)
+        # Если это последняя фото в массиве, то добавляем подпись. Остальные фото просто преобразуем в InputMediaPhoto
+        # и формируем объект MediaGroup
+        if i == len(album) - 1:
             photo_media.append(InputMediaPhoto(media=message.photo[-1].file_id, caption=post_caption))
         else:
             photo_media.append(InputMediaPhoto(media=message.photo[-1].file_id))
-    return photo_media
+    return photo_media  # Возвращаем идентификатор медиа-группы
+
+
+async def add_in_db_media(sent_message, post_caption: str = ''):
+    """
+    Идентификатор медиа-группы
+
+    Args:
+        sent_message: sent_message объект из Telegram API.
+        post_caption: Текст подписи к первому фото.
+
+    Returns:
+        Список InputMediaPhoto.
+    """
+    media_group_message_id = sent_message[0].message_id  # Получаем идентификатор медиа-группы
+    for i, message in enumerate(sent_message):
+        file_id = message.photo[-1].file_id
+        file_path = await download_photo(message, file_id=file_id)
+        if i == 0:
+            BotDB.add_post_from_telegram_in_db(media_group_message_id, message.message_id, 0, file_path, post_caption)
+        elif i == len(sent_message) - 1:
+            BotDB.add_post_from_telegram_in_db(media_group_message_id, message.message_id, 1, file_path, post_caption)
+        else:
+            BotDB.add_post_from_telegram_in_db(media_group_message_id, message.message_id, 0, file_path, post_caption)
+    return media_group_message_id
 
 
 async def send_media_group_message(chat_id: int, message: types.Message, media_group: list[InputMediaPhoto]):
@@ -65,12 +116,37 @@ async def send_media_group_message(chat_id: int, message: types.Message, media_g
         chat_id=chat_id,
         media=media_group,
     )
-    print(f'ПОЛНЫЙ ОБЪЕКТ ОТПРАВЛЕННЫХ СООБЩЕНИЙ : {sent_message}\n\n')
+    await add_in_db_media(sent_message, post_caption=sent_message[-1].caption)
     message_id = sent_message[-1].message_id
-    print(f'ИДЕНТИФИКАТОР ПОСЛЕДНГО СООБЩЕНИЯ: {message_id}')
     return message_id
 
 
+async def send_media_group_with_caption(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])
+            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(
@@ -160,7 +236,7 @@ def get_banned_users_buttons():
 
 def get_help_message_id(media_group_message_id: int, data: dict) -> int:
     """
-    Получает идентификатор сообщения помощи по идентификатору сообщения группы.
+    Получает идентификатор текстового сообщения по идентификатору сообщения группы.
 
     Args:
       media_group_message_id: Идентификатор сообщения группы
@@ -169,9 +245,8 @@ def get_help_message_id(media_group_message_id: int, data: dict) -> int:
     Returns:
       Идентификатор сообщения помощи.
     """
-
-    if 'help_message_id' in data and 'media_group_message_id' in data:
-        return data['media_group_message_id']
+    if 'help_message_id' in data:
+        return data['help_message_id']  # Возвращаем help_message_id
     else:
         return 0
 
@@ -188,7 +263,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)