migrate to aiogram 3.10.0

2024-07-14 21:22:33 +03:00
parent 24ac638433
commit 1a0344d0e8
47 changed files with 1145 additions and 846 deletions
--- a/helper_bot/handlers/callback/main.py
+++ b/helper_bot/handlers/callback/main.py
@@ -0,0 +1,164 @@
+import traceback
+
+from aiogram import Router, F, types
+from aiogram.fsm.context import FSMContext
+from aiogram.types import CallbackQuery
+
+from database.db import BotDB
+from helper_bot.keyboards.main 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
+from logs.custom_logger import Logger
+
+callback_router = Router()
+
+#Инициализируем логгер
+callback_logger = Logger(name='callback_logger')
+logger = callback_logger.get_logger()
+
+bdf = BaseDependencyFactory()
+GROUP_FOR_POST = bdf.settings['Telegram']['group_for_posts']
+GROUP_FOR_MESSAGE = bdf.settings['Telegram']['group_for_message']
+MAIN_PUBLIC = bdf.settings['Telegram']['main_public']
+GROUP_FOR_LOGS = bdf.settings['Telegram']['group_for_logs']
+IMPORTANT_LOGS = bdf.settings['Telegram']['important_logs']
+PREVIEW_LINK = bdf.settings['Telegram']['preview_link']
+LOGS = bdf.settings['Settings']['logs']
+TEST = bdf.settings['Settings']['test']
+
+BotDB = BotDB('database/tg-bot-database')
+
+
+@callback_router.callback_query(
+    F.data == "publish"
+)
+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 != "^":
+        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)
+            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.data == 'publish' and call.message.content_type == 'photo':
+        try:
+            print(f'CALLMESSAGE - {call.message.text}')
+            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)
+            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.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.copy_messages(chat_id=MAIN_PUBLIC, from_chat_id=GROUP_FOR_POST, message_ids=[media_group_message_id, media_group_message_id-1])
+        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'])
+        await call.answer(text='Выложено!', show_alert=True, cache_time=3)
+
+@callback_router.callback_query(
+    F.data == "decline"
+)
+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 != "^":
+            await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
+            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)
+    except Exception as e:
+        await call.bot.send_message(IMPORTANT_LOGS,
+                                    f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
+        logger.error(f'Ошибка при удалении сообщения в группе {GROUP_FOR_POST}: {str(e)}')
+        await call.answer(text='Что-то пошло не так!', show_alert=True, cache_time=3)
+
+
+@callback_router.callback_query(
+    F.data.contains('ban')
+)
+async def process_ban_user(call: CallbackQuery, state: FSMContext):
+    user_id = call.data[4:]
+    logger.info(
+        f"Вызов функции process_ban_user. Данные callback: {call.data} пользователь: {user_id}")
+    user_name = BotDB.get_username(user_id=user_id)
+    if user_name:
+        await state.update_data(user_id=user_id, user_name=user_name, message_for_user=None,
+                                date_to_unban=None)
+        markup = create_keyboard_for_ban_reason()
+        await call.message.answer(
+            text=f"Выбран пользователь: {user_id}. Выбери причину бана из списка или напиши ее в чат",
+            reply_markup=markup)
+        await state.set_state('BAN_2')
+    else:
+        markup = get_reply_keyboard_admin()
+        await call.message.answer(text='Пользователь с таким ID не найден в базе', markup=markup)
+        await state.set_state('ADMIN')
+
+
+@callback_router.callback_query(
+    F.data.contains('unlock')
+)
+async def process_unlock_user(call: CallbackQuery):
+    user_id = call.data[7:]
+    delete_user_blacklist(user_id)
+    logger.info(f"Разблокирован пользователь с ID: {user_id}")
+    username = BotDB.get_username(user_id)
+    await call.answer(f'Пользователь разблокирован {username}', show_alert=True)
+
+
+@callback_router.callback_query(
+    F.data == 'return'
+)
+async def return_to_main_menu(call: CallbackQuery):
+    await call.message.delete()
+    logger.info(f"Запуск админ панели для пользователя: {call.message.from_user.id}")
+    markup = get_reply_keyboard_admin()
+    await call.message.answer("Добро пожаловать в админку. Выбери что хочешь:",
+                              reply_markup=markup)
+
+
+@callback_router.callback_query(
+    F.data.contains('page')
+)
+async def change_page(call: CallbackQuery):
+    page_number = int(call.data[5:])
+    logger.info(f"Переход на страницу {page_number}")
+    if call.message.text == 'Список пользователей которые последними обращались к боту':
+        list_users = BotDB.get_last_users_from_db()
+        #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,
+                                                 reply_markup=keyboard)