telegram-helper-bot/helper_bot/handlers/callback/main.py

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"Выбран пользователь:\nid: {user_id}\nusername:{user_name}. Выбери причину бана из списка или напиши ее в чат",
            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:]
    user_name = BotDB.get_username(user_id=user_id)
    delete_user_blacklist(user_id)
    logger.info(f"Разблокирован пользователь с ID: {user_id}\nusername:{user_name}")
    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)