add answer for user

2024-07-21 13:24:18 +03:00
parent 0704e6b3fe
commit 33fa84943d
9 changed files with 296 additions and 176 deletions
--- a/helper_bot/handlers/callback/callback_handlers.py
+++ b/helper_bot/handlers/callback/callback_handlers.py
@@ -4,7 +4,7 @@ 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, \
@@ -35,7 +35,14 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
        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)
@@ -47,6 +54,12 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
    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)
@@ -58,6 +71,11 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
    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)
@@ -69,6 +87,11 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
    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)
@@ -80,6 +103,11 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
    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)
@@ -91,6 +119,11 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
    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)
@@ -112,6 +145,11 @@ async def post_for_group(call: CallbackQuery, state: FSMContext):
        # Выкладываем пост в канал
        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)
@@ -124,8 +162,15 @@ 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)
@@ -135,6 +180,11 @@ async def decline_post_for_group(call: CallbackQuery, state: FSMContext):
            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,