вова помоги (merge front-misha-ui)

handlers/__init__.py

@@ -4,6 +4,7 @@ from .subscriptions import router as subscriptions_router
 from .support import router as support_router
 from .referrals import router as referrals_router
 
+# Список всех роутеров бота
 routers = [
     start_router,
     profile_router,
@@ -11,3 +12,5 @@ routers = [
     support_router,
     referrals_router,
 ]
+
+__all__ = ["routers"]

handlers/profile.py

@@ -24,7 +24,7 @@ router = Router()
 logger = logging.getLogger(__name__)
 
 
-async def call_api(method, endpoint, data=None,base_url = BASE_URL_FASTAPI):
+async def call_api(method, endpoint, data=None, base_url=BASE_URL_FASTAPI):
     """
     Выполняет HTTP-запрос к FastAPI.
     """
@@ -124,20 +124,11 @@ async def profile_callback_handler(callback: CallbackQuery):
 @router.callback_query(lambda callback: callback.data == "balance")
 async def balance_callback_handler(callback: CallbackQuery):
     """
-    Обработчик callback_query для баланса.
+    При нажатии «Пополнить баланс» показываем выбор суммы пополнения.
     """
-    user_data = await call_api("GET", f"/user/{callback.from_user.id}")
-    if not user_data:
-        await callback.message.answer(
-            "Произошла ошибка, попробуйте позже или свяжитесь с администрацией."
-        )
-        await callback.answer()
-        return
-
     await callback.message.edit_text(
-        f"💰 Текущий баланс: {user_data['balance']} ₽\n\n"
-        "Выбери сумму, на которую хочешь пополнить счёт.",
-        reply_markup=balance_keyboard(),
+        "💳 Выберите сумму пополнения:",
+        reply_markup=popup_keyboard(),
     )
     await callback.answer()
 
@@ -213,17 +204,23 @@ async def tranhist_callback_handler(callback: CallbackQuery):
 async def popup_confirm_callback_handler(callback: CallbackQuery):
     """
     После выбора суммы показываем варианты оплаты.
-    Реального списания пока нет — только выбор метода.
+    Разрешены только суммы 200, 300, 600, 1000 ₽.
     """
     try:
-        data = callback.data.split(":")
-        amount_str = data[1]
-        amount = int(float(amount_str))
+        _, amount_raw = callback.data.split(":", maxsplit=1)
+        amount = int(float(amount_raw))
     except Exception:
         await callback.message.answer("Некорректная сумма пополнения.")
         await callback.answer()
         return
 
+    if amount not in {200, 300, 600, 1000}:
+        await callback.message.answer(
+            "Эта сумма пополнения недоступна. Выбери вариант от 200 до 1000 ₽."
+        )
+        await callback.answer()
+        return
+
     text = (
         f"💰 Сумма пополнения: {amount} ₽\n\n"
         "Выбери способ оплаты:"
@@ -256,28 +253,39 @@ async def method_ykassa_handler(callback: CallbackQuery):
     """
     amount = callback.data.split("_")[-1]
 
+    # Сразу отвечаем на callback, чтобы Telegram не считал запрос "старым"
+    try:
+        await callback.answer()
+    except Exception:
+        # Если по какой-то причине уже отвечали — просто игнорируем
+        pass
+
     # Формируем URL с query parameters вместо JSON body
-    endpoint = f"/billing/payments/init?user_id={callback.from_user.id}&amount={float(amount)}&provider=yookassa"
+    endpoint = (
+        f"/billing/payments/init?"
+        f"user_id={callback.from_user.id}&amount={float(amount)}&provider=yookassa"
+    )
 
     logger.info(f"Отправка запроса на инициализацию платежа: {endpoint}")
 
     # Отправляем POST запрос с пустым телом (параметры в URL)
     result = await call_api("POST", endpoint, None, "http://billing:8000")
 
-    if result == "ERROR":
+    # Биллинг вообще не ответил/упал
+    if result == "ERROR" or not isinstance(result, dict):
         await callback.message.edit_text(
             "❌ Произошла ошибка при создании платежа. Попробуйте позже."
         )
-        await callback.answer()
         return
 
-    # Проверяем успешность операции
+    # Биллинг вернул ошибку (success = False)
     if not result.get("success", False):
-        error_msg = result.get("error", "Неизвестная ошибка")
-        await callback.message.edit_text(
-            f"❌ Ошибка: {error_msg}"
+        error_msg = (
+            result.get("error")
+            or result.get("detail")
+            or "Неизвестная ошибка"
         )
-        await callback.answer()
+        await callback.message.edit_text(f"❌ Ошибка: {error_msg}")
         return
 
     payment_url = result.get("confirmation_url", "#")
@@ -289,16 +297,16 @@ async def method_ykassa_handler(callback: CallbackQuery):
         f"📋 ID платежа: <code>{payment_id}</code>\n\n"
         f"➡️ <a href='{payment_url}'>Перейти к оплате</a>\n\n"
         f"<i>После оплаты нажмите кнопку 'Проверить оплату'</i>",
-        parse_mode="HTML",
+        parse_mode=ParseMode.HTML,
         disable_web_page_preview=True,
         reply_markup=types.InlineKeyboardMarkup(
             inline_keyboard=[[
                 types.InlineKeyboardButton(
                     text="🔄 Проверить оплату",
-                    callback_data=f"check_payment:{payment_id}"
-                )]
-            ]
+                    callback_data=f"check_payment:{payment_id}",
                 )
+            ]]
+        ),
     )
 
     await callback.answer()
@@ -307,30 +315,38 @@ async def method_ykassa_handler(callback: CallbackQuery):
 @router.callback_query(lambda callback: callback.data.startswith("method_crypto_"))
 async def method_crypto_handler(callback: CallbackQuery):
     """
-    Заглушка: оплата через CryptoBot.
+    Оплата через CryptoBot.
     """
     amount = callback.data.split("_")[-1]
-    endpoint = f"/billing/payments/init?user_id={callback.from_user.id}&amount={float(amount)}&provider=cryptobot"
+
+    # Сразу отвечаем на callback, чтобы избежать таймаута
+    try:
+        await callback.answer()
+    except Exception:
+        pass
+
+    endpoint = (
+        f"/billing/payments/init?"
+        f"user_id={callback.from_user.id}&amount={float(amount)}&provider=cryptobot"
+    )
 
     logger.info(f"Отправка запроса на инициализацию платежа: {endpoint}")
 
-    # Отправляем POST запрос с пустым телом (параметры в URL)
     result = await call_api("POST", endpoint, None, "http://billing:8000")
 
-    if result == "ERROR":
+    if result == "ERROR" or not isinstance(result, dict):
         await callback.message.edit_text(
             "❌ Произошла ошибка при создании платежа. Попробуйте позже."
         )
-        await callback.answer()
         return
 
-    # Проверяем успешность операции
     if not result.get("success", False):
-        error_msg = result.get("error", "Неизвестная ошибка")
-        await callback.message.edit_text(
-            f"❌ Ошибка: {error_msg}"
+        error_msg = (
+            result.get("error")
+            or result.get("detail")
+            or "Неизвестная ошибка"
         )
-        await callback.answer()
+        await callback.message.edit_text(f"❌ Ошибка: {error_msg}")
         return
 
     payment_url = result.get("confirmation_url", "#")
@@ -342,16 +358,16 @@ async def method_crypto_handler(callback: CallbackQuery):
         f"📋 ID платежа: <code>{payment_id}</code>\n\n"
         f"➡️ <a href='{payment_url}'>Перейти к оплате</a>\n\n"
         f"<i>После оплаты нажмите кнопку 'Проверить оплату'</i>",
-        parse_mode="HTML",
+        parse_mode=ParseMode.HTML,
         disable_web_page_preview=True,
         reply_markup=types.InlineKeyboardMarkup(
             inline_keyboard=[[
                 types.InlineKeyboardButton(
                     text="🔄 Проверить оплату",
-                    callback_data=f"check_payment:{payment_id}"
-                )]
-            ]
+                    callback_data=f"check_payment:{payment_id}",
                 )
+            ]]
+        ),
     )
 
     await callback.answer()

handlers/referrals.py

@@ -29,6 +29,7 @@ async def referrals_command(message: types.Message):
     """
     Команда /referrals — показывает текст реферальной программы.
     """
+    logger.info(f"Получена команда /referrals от {message.from_user.id}")
     try:
         text = await _build_referral_text(message.bot, message.from_user.id)
         await message.answer(text, parse_mode=ParseMode.HTML)

handlers/start.py

@@ -5,6 +5,8 @@ import logging
 from instences.config import BASE_URL_FASTAPI
 import aiohttp
 from keyboard.keyboards import main_keyboard
+from aiogram.enums.parse_mode import ParseMode
+from .referrals import _build_referral_text
 
 router = Router()
 logger = logging.getLogger(__name__)
@@ -25,7 +27,8 @@ async def call_api(method, endpoint, data=None):
         async with aiohttp.ClientSession() as session:
             async with session.request(method, url, json=data) as response:
                 logger.info(
-                    f"Получен ответ от {url}: статус {response.status}")
+                    f"Получен ответ от {url}: статус {response.status}"
+                )
 
                 if response.status in {200, 201}:
                     result = await response.json()
@@ -35,7 +38,8 @@ async def call_api(method, endpoint, data=None):
                     logger.debug(f"Код {response.status}, возвращаю ничего")
                     return None
                 logger.error(
-                    f"Ошибка в запросе: статус {response.status}, причина {response.reason}")
+                    f"Ошибка в запросе: статус {response.status}, причина {response.reason}"
+                )
                 return "ERROR"
     except Exception as e:
         logger.exception(f"Исключение при выполнении запроса к {url}: {e}")
@@ -43,18 +47,11 @@ async def call_api(method, endpoint, data=None):
 
 
 def _welcome_text(username: str | None) -> str:
-    uname = username or "птенец"
-    return (
-        f"🪽 Добро пожаловать в Lark VPN, {uname}!\n\n"
-        "Здесь ты получаешь стабильный доступ к TikTok, YouTube, Discord, Instagram "
-        "и другим сервисам без ограничений.\n\n"
-        "1️⃣ Пополни баланс\n"
-        "2️⃣ Выбери подходящий тариф\n"
-        "3️⃣ Подключай конфиг на своих устройствах и забудь про блокировки.\n\n"
-        "Следи за акциями и розыгрышами в канале:\n"
-        "https://t.me/+0z5xqn3F1m02OTJi\n\n"
-        "Lark держит связь. 🐣🦅"
-    )
+    """
+    Текст приветствия в /start и в главном меню.
+    Имя пока не используем — оставляем сигнатуру на будущее.
+    """
+    return "🥚 Lark Security\n\nВыберите действие из меню ниже."
 
 
 @router.message(Command("start"))
@@ -72,8 +69,13 @@ async def start_command(message: Message):
         user_data = await call_api("GET", f"/user/{message.from_user.id}")
         if not user_data:
             logger.debug(
-                "Пользователь не найден в базе, создаем новую запись.")
-            await call_api("POST", "/user/create", {"telegram_id": message.from_user.id})
+                "Пользователь не найден в базе, создаем новую запись."
+            )
+            await call_api(
+                "POST",
+                "/user/create",
+                {"telegram_id": message.from_user.id},
+            )
 
         logger.debug("Отправка приветственного сообщения пользователю.")
         await message.answer(
@@ -89,6 +91,28 @@ async def start_command(message: Message):
         await message.answer("Произошла ошибка. Попробуйте позже.")
 
 
+@router.message(Command("referrals"))
+async def referrals_menu_command(message: Message):
+    """
+    Команда /referrals из бокового меню Telegram.
+    Показывает текст реферальной программы.
+    """
+    logger.info(
+        f"Получена команда /referrals от пользователя: "
+        f"{message.from_user.id} ({message.from_user.username})"
+    )
+    try:
+        text = await _build_referral_text(message.bot, message.from_user.id)
+        await message.answer(text, parse_mode=ParseMode.HTML)
+        logger.info("Реферальная программа отправлена пользователю.")
+    except Exception as e:
+        logger.exception(
+            f"Ошибка при обработке команды /referrals для пользователя "
+            f"{message.from_user.id}: {e}"
+        )
+        await message.answer("Произошла ошибка. Попробуйте позже.")
+
+
 @router.callback_query(lambda callback: callback.data == "base")
 async def start_callback_handler(callback: CallbackQuery):
     """

keyboard/keyboards.py

@@ -13,12 +13,7 @@ def main_keyboard():
             callback_data="profile",
         )
     )
-    builder.row(
-        InlineKeyboardButton(
-            text="👥 Реферальная программа",
-            callback_data="referral",
-        )
-    )
+    # ------
     builder.row(
         InlineKeyboardButton(
             text="❔ FAQ ❔",
@@ -37,7 +32,7 @@ def main_keyboard():
 def account_keyboard():
     """
     Клавиатура профиля:
-    только пополнить баланс и история транзакций.
+    пополнить баланс, история транзакций, назад в главное меню.
     """
     builder = InlineKeyboardBuilder()
     builder.row(
@@ -52,6 +47,12 @@ def account_keyboard():
             callback_data="tranhist",
         )
     )
+    builder.row(
+        InlineKeyboardButton(
+            text="🔙 Назад",
+            callback_data="base",
+        )
+    )
     return builder.as_markup()
 
 
@@ -83,25 +84,25 @@ def balance_keyboard():
 
 def popup_keyboard():
     """
-    Суммы пополнения.
+    Суммы пополнения: 200, 300, 600, 1000 ₽.
     """
     builder = InlineKeyboardBuilder()
     builder.row(
         InlineKeyboardButton(text="200 ₽", callback_data="popup:200"),
-        InlineKeyboardButton(text="500 ₽", callback_data="popup:500"),
+    )
+    builder.row(
+        InlineKeyboardButton(text="300 ₽", callback_data="popup:300"),
+    )
+    builder.row(
+        InlineKeyboardButton(text="600 ₽", callback_data="popup:600"),
     )
     builder.row(
         InlineKeyboardButton(text="1000 ₽", callback_data="popup:1000"),
-        InlineKeyboardButton(text="2000 ₽", callback_data="popup:2000"),
-    )
-    builder.row(
-        InlineKeyboardButton(text="3000 ₽", callback_data="popup:3000"),
-        InlineKeyboardButton(text="5000 ₽", callback_data="popup:5000"),
     )
     builder.row(
         InlineKeyboardButton(
             text="🔙 Назад",
-            callback_data="balance",
+            callback_data="profile",  # назад в профиль
         )
     )
     return builder.as_markup()