вова помоги

handlers/profile.py

@@ -1,3 +1,5 @@
+# Профиль. последнее изменение 24.11.2025
+
 from aiogram import Router, types
 from aiogram.types import CallbackQuery
 import logging
@@ -13,6 +15,7 @@ from keyboard.keyboards import (
     confirm_popup_keyboard,
     guide_keyboard,
     balance_keyboard,
+    payment_methods_keyboard,
 )
 
 locale.setlocale(locale.LC_TIME, "ru_RU.UTF-8")
@@ -21,18 +24,24 @@ router = Router()
 logger = logging.getLogger(__name__)
 
 
-async def call_api(method, endpoint, data=None):
+async def call_api(method, endpoint, data=None, base_url=BASE_URL_FASTAPI):
     """
     Выполняет HTTP-запрос к FastAPI.
     """
-    url = f"{BASE_URL_FASTAPI}{endpoint}"
+    url = f"{base_url}{endpoint}"
     logger.info(f"Инициализация запроса: {method} {url} с данными {data}")
 
     try:
         async with aiohttp.ClientSession() as session:
-            async with session.request(method, url, json=data) as response:
+            async with session.request(
+                method,
+                url,
+                json=data,
+                headers={"Content-Type": "application/json"}
+            ) as response:
                 logger.info(
-                    f"Получен ответ от {url}: статус {response.status}")
+                    f"Получен ответ от {url}: статус {response.status}"
+                )
 
                 if response.status in {200, 201}:
                     result = await response.json()
@@ -54,7 +63,7 @@ async def call_api(method, endpoint, data=None):
 async def profile_callback_handler(callback: CallbackQuery):
     """
     Профиль пользователя.
-    Логика работы с API сохранена, переписан только текст/визуал.
+    Логика работы с API сохранена, изменён только текст/визуал.
     """
     try:
         user_data = await call_api("GET", f"/user/{callback.from_user.id}")
@@ -65,55 +74,46 @@ async def profile_callback_handler(callback: CallbackQuery):
             await callback.answer()
             return
 
-        sub_data = await call_api("GET", f"/subscription/{user_data['telegram_id']}/last")
+        # Последняя подписка пользователя
+        sub_data = await call_api(
+            "GET", f"/subscription/{user_data['telegram_id']}/last"
+        )
         if sub_data == "ERROR" or not isinstance(sub_data, dict):
             sub_data = None
 
-        balance_text = f"💰 Баланс: {user_data['balance']} ₽"
+        username = callback.from_user.username or "-"
+        balance = user_data.get("balance", 0)
 
-        username = callback.from_user.username or "пользователь"
+        # Статус подписки: Активна / Нет активных
+        sub_status = "⚫ Нет активных"
 
-        # Нет подписки
-        if not sub_data:
-            text = (
-                f"📜 Профиль {username}\n\n"
-                f"{balance_text}\n\n"
-                "У тебя пока нет активной подписки.\n"
-                "Пополняй баланс и подключай тариф Lark, чтобы получить статус 🐣 или 🦅."
-            )
-        else:
-            expiry_date = sub_data.get("end_date")
-            formatted_date = (
-                datetime.fromisoformat(expiry_date).strftime("%d %B %Y г.")
-                if expiry_date
-                else None
-            )
-            is_expired = (
-                datetime.fromisoformat(expiry_date) < datetime.now()
-                if expiry_date
-                else True
-            )
-            status_icon = "✖️" if is_expired else "☑️"
-            profile_status = "🦅" if "Pro" in sub_data.get("plan", "") else "🐣"
-
-            if not is_expired and formatted_date:
-                sub_text = (
-                    f"Подписка активна до {formatted_date} {status_icon}\n"
-                    f"Текущий статус: {profile_status}"
-                )
+        if sub_data:
+            expiry_date = sub_data.get("expiry_date")
+            if expiry_date:
+                try:
+                    is_expired = datetime.fromisoformat(
+                        expiry_date) < datetime.now()
+                except ValueError:
+                    is_expired = True
             else:
-                sub_text = (
-                    f"Подписка неактивна {status_icon}\n"
-                    "Ты можешь продлить её в разделе подписок."
-                )
+                is_expired = True
 
-            text = (
-                f"📜 Профиль {username} {profile_status}\n\n"
-                f"{sub_text}\n\n"
-                f"{balance_text}"
-            )
+            if not is_expired:
+                sub_status = "🟢 Активна"
 
-        await callback.message.edit_text(text, reply_markup=account_keyboard())
+        text = (
+            "🥚 <b>Профиль</b>\n\n"
+            f"Пользователь: @{username}\n"
+            f"Баланс: {balance} ₽\n"
+            f"Статус подписки: {sub_status}\n\n"
+            "Выберите действие:"
+        )
+
+        await callback.message.edit_text(
+            text,
+            parse_mode=ParseMode.HTML,
+            reply_markup=account_keyboard(),
+        )
     except Exception as e:
         logger.exception(f"Ошибка в обработчике профиля: {e}")
         await callback.message.answer("Произошла ошибка. Попробуйте позже.")
@@ -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()
 
@@ -175,7 +166,9 @@ async def tranhist_callback_handler(callback: CallbackQuery):
         return
 
     try:
-        transactions = await call_api("GET", f"/user/{user_data['telegram_id']}/transactions")
+        transactions = await call_api(
+            "GET", f"/user/{user_data['telegram_id']}/transactions"
+        )
         if not transactions:
             await callback.message.edit_text(
                 "У вас нет транзакций.", reply_markup=tranhist_keyboard()
@@ -200,7 +193,9 @@ async def tranhist_callback_handler(callback: CallbackQuery):
         )
     except Exception as e:
         logger.error(f"Ошибка обработки транзакций: {e}")
-        await callback.message.edit_text("Произошла ошибка. Попробуйте позже.")
+        await callback.message.edit_text(
+            "Произошла ошибка. Попробуйте позже."
+        )
     finally:
         await callback.answer()
 
@@ -208,25 +203,173 @@ async def tranhist_callback_handler(callback: CallbackQuery):
 @router.callback_query(lambda callback: callback.data.startswith("popup:"))
 async def popup_confirm_callback_handler(callback: CallbackQuery):
     """
-    Обработчик подтверждения пополнения баланса.
+    После выбора суммы показываем варианты оплаты.
+    Разрешены только суммы 200, 300, 600, 1000 ₽.
     """
-    data = callback.data.split(":")
-    popup_info = data[1]
-    result = await call_api(
-        "POST",
-        f"/user/{callback.from_user.id}/balance/{float(popup_info)}",
-    )
-    if result == "ERROR":
+    try:
+        _, 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"✅ Баланс пополнен на {popup_info} ₽. Спасибо, что остаёшься с Lark."
-    await callback.message.edit_text(
-        text=text, reply_markup=confirm_popup_keyboard()
+    text = (
+        f"💰 Сумма пополнения: {amount} ₽\n\n"
+        "Выбери способ оплаты:"
     )
+
+    await callback.message.edit_text(
+        text=text,
+        reply_markup=payment_methods_keyboard(amount),
+    )
+    await callback.answer()
+
+
+@router.callback_query(lambda callback: callback.data.startswith("method_stars_"))
+async def method_stars_handler(callback: CallbackQuery):
+    """
+    Заглушка: оплата через Telegram Stars.
+    """
+    amount = callback.data.split("_")[-1]
+    await callback.message.edit_text(
+        f"⭐ Оплата через Telegram Stars на {amount} ₽ пока в разработке.\n\n"
+        "Позже сюда подвяжем реальный платёж.",
+    )
+    await callback.answer()
+
+
+@router.callback_query(lambda callback: callback.data.startswith("method_ykassa_"))
+async def method_ykassa_handler(callback: CallbackQuery):
+    """
+    Обработчик оплаты через YooKassa.
+    """
+    amount = callback.data.split("_")[-1]
+
+    # Сразу отвечаем на callback, чтобы Telegram не считал запрос "старым"
+    try:
+        await callback.answer()
+    except Exception:
+        # Если по какой-то причине уже отвечали — просто игнорируем
+        pass
+
+    # Формируем URL с query parameters вместо JSON body
+    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" or not isinstance(result, dict):
+        await callback.message.edit_text(
+            "❌ Произошла ошибка при создании платежа. Попробуйте позже."
+        )
+        return
+
+    # Биллинг вернул ошибку (success = False)
+    if not result.get("success", False):
+        error_msg = (
+            result.get("error")
+            or result.get("detail")
+            or "Неизвестная ошибка"
+        )
+        await callback.message.edit_text(f"❌ Ошибка: {error_msg}")
+        return
+
+    payment_url = result.get("confirmation_url", "#")
+    payment_id = result.get("payment_id", "")
+
+    await callback.message.edit_text(
+        f"💵 <b>Оплата через YooKassa</b>\n\n"
+        f"💰 Сумма: <code>{amount}</code> руб\n"
+        f"📋 ID платежа: <code>{payment_id}</code>\n\n"
+        f"➡️ <a href='{payment_url}'>Перейти к оплате</a>\n\n"
+        f"<i>После оплаты нажмите кнопку 'Проверить оплату'</i>",
+        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}",
+                )
+            ]]
+        ),
+    )
+
+    await callback.answer()
+
+
+@router.callback_query(lambda callback: callback.data.startswith("method_crypto_"))
+async def method_crypto_handler(callback: CallbackQuery):
+    """
+    Оплата через CryptoBot.
+    """
+    amount = callback.data.split("_")[-1]
+
+    # Сразу отвечаем на 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}")
+
+    result = await call_api("POST", endpoint, None, "http://billing:8000")
+
+    if result == "ERROR" or not isinstance(result, dict):
+        await callback.message.edit_text(
+            "❌ Произошла ошибка при создании платежа. Попробуйте позже."
+        )
+        return
+
+    if not result.get("success", False):
+        error_msg = (
+            result.get("error")
+            or result.get("detail")
+            or "Неизвестная ошибка"
+        )
+        await callback.message.edit_text(f"❌ Ошибка: {error_msg}")
+        return
+
+    payment_url = result.get("confirmation_url", "#")
+    payment_id = result.get("payment_id", "")
+
+    await callback.message.edit_text(
+        f"💵 <b>Оплата через Сryptobot</b>\n\n"
+        f"💰 Сумма: <code>{amount}</code> руб\n"
+        f"📋 ID платежа: <code>{payment_id}</code>\n\n"
+        f"➡️ <a href='{payment_url}'>Перейти к оплате</a>\n\n"
+        f"<i>После оплаты нажмите кнопку 'Проверить оплату'</i>",
+        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}",
+                )
+            ]]
+        ),
+    )
+
     await callback.answer()