Check

handlers/handlers.py

@@ -3,7 +3,7 @@ from aiogram.filters import Command
 from databases.postgresql import DatabaseManager
 from databases.model import User, Subscription, Transaction, Administrators
 from databases.db_config import get_postgres_session
-from keyboard.keyboards import subhist_keyboard,tarif_confirm_keyboard, popup_keyboard, main_keyboard,faq_keyboard,about_tarifs_keyboard, account_keyboard, buy_keyboard,balance_keyboard,guide_keyboard,tarif_Lark_keyboard,tarif_Lark_pro_keyboard,tranhist_keyboard
+from keyboard.keyboards import subhist_keyboard,confirm_popup_keyboard,tarif_confirm_keyboard, popup_keyboard, main_keyboard,faq_keyboard,about_tarifs_keyboard, account_keyboard, buy_keyboard,balance_keyboard,guide_keyboard,tarif_Lark_keyboard,tarif_Lark_pro_keyboard,tranhist_keyboard
 
 # Инициализируем менеджер базы данных
 db_manager = DatabaseManager(get_postgres_session)
@@ -93,7 +93,7 @@ async def popup_callback_handler(callback: types.CallbackQuery):
 
     if user:
         await callback.message.edit_text(
-            f"Ты думал здесь что то будет?",
+            f"Работает в режиме теста!!!",
             reply_markup=popup_keyboard()
         )
     else:
@@ -242,6 +242,22 @@ async def lark_tariff_callback_handler(callback: types.CallbackQuery):
     keyboard = tarif_confirm_keyboard(tariff_name, tariff_time, tariff_class)
     await callback.message.edit_text(text=text, reply_markup=keyboard)
 
+async def popup_confirm_callback_handler(callback: types.CallbackQuery):
+    """
+    Обработчик подтверждения пополнения баланса.
+    """
+    data = callback.data.split(":")
+    popup_info = data[1]
+    result = await db_manager.update_balance(callback.from_user.id,popup_info)
+    if result == "ERROR":
+        await callback.message.answer(
+            "Произошла ошибка, попробуйте позже или свяжитесь с администрацией."
+        )
+        await callback.answer()
+        return
+    text = f"Вы пополнили свой баланс на {popup_info}. P.S. Мы завтра закрываемся"
+    await callback.message.edit_text(text=text, reply_markup=confirm_popup_keyboard())
+
 async def confirm_callback_handler(callback: types.CallbackQuery):
     """
     Обработчик подтверждения покупки тарифа.
@@ -295,4 +311,5 @@ def register_handlers(dp: Dispatcher):
     dp.callback_query.register(guide_callback_handler, lambda c: c.data == "guide")
     dp.callback_query.register(about_tarifs_callback_handler, lambda c: c.data == "about_tarifs")
     dp.callback_query.register(lark_tariff_callback_handler, lambda c: c.data.startswith("Lark:"))
-    dp.callback_query.register(confirm_callback_handler, lambda c: c.data.startswith("confirm:"))
+    dp.callback_query.register(confirm_callback_handler, lambda c: c.data.startswith("confirm:"))
+    dp.callback_query.register(popup_confirm_callback_handler, lambda c: c.data.startswith("popup:"))