Чет исправил, я забыл

This commit is contained in:
unknown
2025-12-02 14:28:03 +03:00
parent 00edfdbd19
commit 709e8f09eb
3 changed files with 290 additions and 17 deletions

View File

@@ -1,7 +1,12 @@
# Профиль. последнее изменение 24.11.2025
from aiogram import Router, types
from aiogram.types import CallbackQuery
from aiogram import Router, types, F
from aiogram.types import (
CallbackQuery,
LabeledPrice,
Message,
PreCheckoutQuery,
)
import logging
from datetime import datetime
from aiogram.enums.parse_mode import ParseMode
@@ -233,17 +238,54 @@ async def popup_confirm_callback_handler(callback: CallbackQuery):
await callback.answer()
# ===== Telegram Stars =====
@router.callback_query(lambda callback: callback.data.startswith("method_stars_"))
async def method_stars_handler(callback: CallbackQuery):
"""
Заглушка: оплата через Telegram Stars.
Оплата через Telegram Stars.
Формируем invoice прямо из бота, без отдельного биллинга.
"""
amount = callback.data.split("_")[-1]
await callback.message.edit_text(
f"⭐ Оплата через Telegram Stars на {amount} ₽ пока в разработке.\n\n"
"Позже сюда подвяжем реальный платёж.",
)
await callback.answer()
try:
amount_str = callback.data.split("_")[-1]
amount_rub = int(float(amount_str))
except Exception:
await callback.message.answer("Некорректная сумма для оплаты.")
await callback.answer()
return
# Внутренний payload, чтобы при успешной оплате понять, что это пополнение баланса
payload = f"stars_topup:{callback.from_user.id}:{amount_rub}"
# 1 ₽ == 1 Star, если нужна другая конвертация — меняй тут
stars_amount = amount_rub
prices = [
LabeledPrice(
label=f"Пополнение баланса на {amount_rub}",
amount=stars_amount,
)
]
try:
await callback.message.answer_invoice(
title="Пополнение баланса Lark VPN",
description=(
f"Пополнение баланса на {amount_rub} ₽ через Telegram Stars.\n\n"
"После успешной оплаты баланс будет зачислен автоматически."
),
payload=payload,
provider_token="", # для Stars провайдер пустой
currency="XTR",
prices=prices,
)
await callback.answer()
except Exception as e:
logger.exception(f"Ошибка при отправке invoice Telegram Stars: {e}")
await callback.message.answer(
"Не удалось создать счёт в Telegram Stars. Попробуй позже или выбери другой способ оплаты."
)
await callback.answer()
@router.callback_query(lambda callback: callback.data.startswith("method_ykassa_"))
@@ -383,3 +425,87 @@ async def guide_callback_handler(callback: CallbackQuery):
reply_markup=guide_keyboard(),
)
await callback.answer()
# ===== Служебные хендлеры для платежей Telegram Stars =====
@router.pre_checkout_query()
async def pre_checkout_query_handler(pre_checkout_query: PreCheckoutQuery):
"""
Обязательный шаг для Telegram Payments:
подтверждаем pre_checkout_query, иначе платёж не пройдёт.
"""
try:
await pre_checkout_query.bot.answer_pre_checkout_query(
pre_checkout_query.id,
ok=True,
)
except Exception as e:
logger.exception(f"Ошибка при answer_pre_checkout_query: {e}")
async def _process_stars_topup(message: Message):
"""
Логика зачисления средств после успешной оплаты Stars.
"""
sp = message.successful_payment
if not sp:
return
payload = sp.invoice_payload or ""
parts = payload.split(":")
if len(parts) != 3 or parts[0] != "stars_topup":
logger.info(
"successful_payment не относится к пополнению баланса Stars.")
return
_, telegram_id_str, amount_str = parts
try:
amount_rub = int(amount_str)
except ValueError:
# На всякий случай fallback к total_amount
amount_rub = sp.total_amount
data = {
"telegram_id": telegram_id_str,
"amount": amount_rub,
"currency": sp.currency,
"provider": "telegram_stars",
"telegram_payment_charge_id": sp.telegram_payment_charge_id,
}
logger.info(
f"Обработка успешного платежа Telegram Stars: "
f"user={telegram_id_str}, amount={amount_rub}, currency={sp.currency}"
)
result = await call_api("POST", "/user/deposit", data)
if result == "ERROR" or result is None:
await message.answer(
"⭐ Оплата через Telegram Stars прошла, но не удалось автоматически обновить баланс.\n"
"Если баланс не изменился — напиши, пожалуйста, в поддержку и укажи время платежа."
)
return
await message.answer(
f"⭐ Оплата через Telegram Stars успешно проведена.\n"
f"На твой баланс зачислено {amount_rub} ₽."
)
@router.message(F.successful_payment)
async def successful_payment_handler(message: Message):
"""
Глобальный хендлер успешных платежей.
Сейчас используем только для пополнения баланса через Stars.
"""
try:
await _process_stars_topup(message)
except Exception as e:
logger.exception(f"Ошибка при обработке successful_payment: {e}")
await message.answer(
"Оплата прошла, но произошла ошибка при обработке. "
"Если баланс не изменился — напиши, пожалуйста, в поддержку."
)