all type message can process

helper_bot/handlers/callback/callback_handlers.py

@@ -8,7 +8,8 @@ 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_to_channel
+    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()
@@ -54,6 +55,50 @@ 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.message.content_type == 'video':
+        try:
+            await send_video_message(MAIN_PUBLIC, call.message, call.message.video.file_id, call.message.caption)
+            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)
+            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)
+            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)
+            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)
@@ -151,7 +196,7 @@ 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')
 

helper_bot/handlers/private/private_handlers.py

@@ -17,7 +17,8 @@ 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, \
-    send_media_group_message_to_private_chat, prepare_media_group_from_middlewares, 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
 from logs.custom_logger import logger
 
 private_router = Router()
@@ -129,6 +130,7 @@ async def suggest_post(message: types.Message, state: FSMContext):
 )
 async def end_message(message: types.Message, state: FSMContext):
     try:
+        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_*'))
@@ -160,8 +162,8 @@ 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':
-            print(message.content_type)
             lower_text = message.text.lower()
             # Получаем текст сообщения и преобразовываем его по правилам
             post_text = get_text_message(lower_text, message.from_user.full_name,
@@ -180,7 +182,6 @@ async def suggest_router(message: types.Message, state: FSMContext, album: list
             await state.set_state("START")
 
         elif message.content_type == 'photo' and message.media_group_id is None:
-            print(message.content_type)
             lower_caption = message.caption.lower()
             markup = get_reply_keyboard_for_post()
             # Получаем текст сообщения и преобразовываем его по правилам
@@ -197,12 +198,69 @@ async def suggest_router(message: types.Message, state: FSMContext, album: list
             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()
+            # Получаем текст сообщения и преобразовываем его по правилам
+
+            # Отправляем видео и текст в приватный канал
+            await send_video_message(GROUP_FOR_POST, message,
+                                     message.video.file_id, post_caption, markup)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            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()
+
+            # Отправляем видеокружок в приватный канал
+            await send_video_note_message(GROUP_FOR_POST, message,
+                                          message.video_note.file_id, markup)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            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:
+            lower_caption = message.caption.lower()
+            markup = get_reply_keyboard_for_post()
+            # Получаем текст сообщения и преобразовываем его по правилам
+            post_caption = get_text_message(lower_caption, message.from_user.full_name,
+                                            message.from_user.username)
+
+            # Отправляем видео и текст в приватный канал
+            await send_audio_message(GROUP_FOR_POST, message,
+                                     message.audio.file_id, post_caption, markup)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            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()
+
+            # Отправляем видео и текст в приватный канал
+            await send_voice_message(GROUP_FOR_POST, message,
+                                     message.voice.file_id, markup)
+
+            # Отправляем юзеру ответ и возвращаем его в меню
+            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 message.content_type != 'photo':
-                await message.bot.send_message(message.chat.id,
-                                               'Я пока не умею работать с таким сообщением. '
-                                               'Пришли текст и фото/фоты(ы)')
 
             # Получаем сообщение и проверяем есть ли подпись. Если подпись есть, то преобразуем ее через функцию
             if album[0].caption:
@@ -214,7 +272,8 @@ async def suggest_router(message: types.Message, state: FSMContext, album: list
             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)
+            media_group_message_id = await send_media_group_message_to_private_chat(GROUP_FOR_POST, message,
+                                                                                    media_group)
             sleep(0.2)
 
             # Получаем клавиатуру и отправляем еще одно текстовое сообщение с кнопками
@@ -230,9 +289,10 @@ async def suggest_router(message: types.Message, state: FSMContext, album: list
             await message.answer(success_send_message, reply_markup=markup_for_user)
             await state.set_state("START")
         else:
-            print(f'Тип: {message.content_type}')
             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()}")