Solo_bot/handlers/commands.py

import asyncpg
from aiogram import Router, types
from aiogram.filters import Command
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
from aiogram.types import Message
from loguru import logger

from bot import bot
from config import DATABASE_URL
from handlers.admin.admin import cmd_add_balance
from handlers.keys.key_management import handle_key_name_input
from handlers.payment.yookassa_pay import ReplenishBalanceState, process_custom_amount_input
from handlers.profile import process_callback_view_profile
from handlers.start import start_command
from handlers.texts import TRIAL

router = Router()


class Form(StatesGroup):
    waiting_for_server_selection = State()
    waiting_for_key_name = State()
    viewing_profile = State()
    waiting_for_message = State()


@router.message(Command("backup"))
async def backup_command(message: Message, is_admin: bool):
    if is_admin:
        from backup import backup_database

        await message.answer("Запускаю бэкап базы данных...")
        await backup_database()
        await message.answer("Бэкап завершен и отправлен админу.")


@router.message(Command("start"))
async def handle_start(message: types.Message, state: FSMContext):
    await start_command(message)


@router.message(Command("add_balance"))
async def handle_add_balance(message: types.Message, state: FSMContext):
    await cmd_add_balance(message)


@router.message(Command("menu"))
async def handle_menu(message: types.Message, state: FSMContext):
    await start_command(message)


@router.message(Command("send_trial"))
async def handle_send_trial_command(
    message: types.Message, state: FSMContext, is_admin: bool
):
    if is_admin:
        try:
            conn = await asyncpg.connect(DATABASE_URL)
            try:
                records = await conn.fetch(
                    """
                    SELECT tg_id FROM connections WHERE trial = 0
                """
                )

                if records:
                    for record in records:
                        tg_id = record["tg_id"]
                        trial_message = TRIAL
                        try:
                            await bot.send_message(chat_id=tg_id, text=trial_message)
                        except Exception as e:
                            if "Forbidden: bot was blocked by the user" in str(e):
                                logger.info(
                                    f"Бот заблокирован пользователем с tg_id: {tg_id}"
                                )
                            else:
                                logger.error(
                                    f"Ошибка при отправке сообщения пользователю {tg_id}: {e}"
                                )

                    await message.answer(
                        "Сообщения о пробном периоде отправлены всем пользователям с не использованным ключом."
                    )
                else:
                    await message.answer(
                        "Нет пользователей с не использованными пробными ключами."
                    )

            finally:
                await conn.close()

        except Exception as e:
            await message.answer(f"Ошибка при отправке сообщений: {e}")


@router.message(Command("send_to_all"))
async def send_message_to_all_clients(
    message: types.Message, state: FSMContext, is_admin: bool, from_panel=False
):
    if from_panel and is_admin:

        await message.answer(
            "Введите текст сообщения, который вы хотите отправить всем клиентам:"
        )
        await state.set_state(Form.waiting_for_message)


@router.message(Form.waiting_for_message)
async def process_message_to_all(
    message: types.Message, state: FSMContext, is_admin: bool
):
    if is_admin:
        text_message = message.text

        try:
            conn = await asyncpg.connect(DATABASE_URL)
            tg_ids = await conn.fetch("SELECT tg_id FROM connections")

            for record in tg_ids:
                tg_id = record["tg_id"]
                try:
                    await bot.send_message(chat_id=tg_id, text=text_message)
                except Exception as e:
                    logger.error(
                        f"Ошибка при отправке сообщения пользователю {tg_id}: {e}. Пропускаем этого пользователя."
                    )

            await message.answer("Сообщение было отправлено всем клиентам.")
        except Exception as e:
            logger.error(f"Ошибка при подключении к базе данных: {e}")
            await message.answer("Произошла ошибка при отправке сообщения.")
        finally:
            await conn.close()

    await state.clear()


@router.message()
async def handle_text(message: types.Message, state: FSMContext):
    current_state = await state.get_state()

    if message.text in ["/send_to_all"]:
        await send_message_to_all_clients(message, state)
        return

    if message.text == "Мой профиль":
        callback_query = types.CallbackQuery(
            id="1",
            from_user=message.from_user,
            chat_instance="",
            data="view_profile",
            message=message,
        )
        await process_callback_view_profile(callback_query, state)
        return

    if current_state == ReplenishBalanceState.entering_custom_amount.state:
        await process_custom_amount_input(message, state)
        return

    if current_state == Form.waiting_for_key_name.state:
        await handle_key_name_input(message, state)
        return

    if message.text == "/backup":
        await backup_command(message)
        return

    elif current_state is None:
        await start_command(message)