Egor
365 lines
14 KiB
Python
365 lines
14 KiB
Python
"""Функции работы с YooKassa вынесены в dedicated mixin.
|
|
|
|
Такое разделение облегчает поддержку и делает очевидным, какая часть
|
|
отвечает за конкретного провайдера.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from datetime import datetime
|
|
from importlib import import_module
|
|
from typing import Any, Dict, Optional, TYPE_CHECKING
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database.models import PaymentMethod, TransactionType
|
|
from app.utils.user_utils import format_referrer_info
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
if TYPE_CHECKING:
|
|
from app.database.models import YooKassaPayment
|
|
|
|
|
|
class YooKassaPaymentMixin:
|
|
"""Mixin с операциями по созданию и подтверждению платежей YooKassa."""
|
|
|
|
async def create_yookassa_payment(
|
|
self,
|
|
db: AsyncSession,
|
|
user_id: int,
|
|
amount_kopeks: int,
|
|
description: str,
|
|
receipt_email: Optional[str] = None,
|
|
receipt_phone: Optional[str] = None,
|
|
metadata: Optional[Dict[str, Any]] = None,
|
|
) -> Optional[Dict[str, Any]]:
|
|
"""Создаёт обычный платёж в YooKassa и сохраняет локальную запись."""
|
|
if not getattr(self, "yookassa_service", None):
|
|
logger.error("YooKassa сервис не инициализирован")
|
|
return None
|
|
|
|
payment_module = import_module("app.services.payment_service")
|
|
|
|
try:
|
|
amount_rubles = amount_kopeks / 100
|
|
|
|
payment_metadata = metadata.copy() if metadata else {}
|
|
payment_metadata.update(
|
|
{
|
|
"user_id": str(user_id),
|
|
"amount_kopeks": str(amount_kopeks),
|
|
"type": "balance_topup",
|
|
}
|
|
)
|
|
|
|
yookassa_response = await self.yookassa_service.create_payment(
|
|
amount=amount_rubles,
|
|
currency="RUB",
|
|
description=description,
|
|
metadata=payment_metadata,
|
|
receipt_email=receipt_email,
|
|
receipt_phone=receipt_phone,
|
|
)
|
|
|
|
if not yookassa_response or yookassa_response.get("error"):
|
|
logger.error(
|
|
"Ошибка создания платежа YooKassa: %s", yookassa_response
|
|
)
|
|
return None
|
|
|
|
yookassa_created_at: Optional[datetime] = None
|
|
if yookassa_response.get("created_at"):
|
|
try:
|
|
dt_with_tz = datetime.fromisoformat(
|
|
yookassa_response["created_at"].replace("Z", "+00:00")
|
|
)
|
|
yookassa_created_at = dt_with_tz.replace(tzinfo=None)
|
|
except Exception as error:
|
|
logger.warning("Не удалось распарсить created_at: %s", error)
|
|
yookassa_created_at = None
|
|
|
|
local_payment = await payment_module.create_yookassa_payment(
|
|
db=db,
|
|
user_id=user_id,
|
|
yookassa_payment_id=yookassa_response["id"],
|
|
amount_kopeks=amount_kopeks,
|
|
currency="RUB",
|
|
description=description,
|
|
status=yookassa_response["status"],
|
|
confirmation_url=yookassa_response.get("confirmation_url"),
|
|
metadata_json=payment_metadata,
|
|
payment_method_type=None,
|
|
yookassa_created_at=yookassa_created_at,
|
|
test_mode=yookassa_response.get("test_mode", False),
|
|
)
|
|
|
|
logger.info(
|
|
"Создан платеж YooKassa %s на %s₽ для пользователя %s",
|
|
yookassa_response["id"],
|
|
amount_rubles,
|
|
user_id,
|
|
)
|
|
|
|
return {
|
|
"local_payment_id": local_payment.id,
|
|
"yookassa_payment_id": yookassa_response["id"],
|
|
"confirmation_url": yookassa_response.get("confirmation_url"),
|
|
"amount_kopeks": amount_kopeks,
|
|
"amount_rubles": amount_rubles,
|
|
"status": yookassa_response["status"],
|
|
"created_at": local_payment.created_at,
|
|
}
|
|
|
|
except Exception as error:
|
|
logger.error("Ошибка создания платежа YooKassa: %s", error)
|
|
return None
|
|
|
|
async def create_yookassa_sbp_payment(
|
|
self,
|
|
db: AsyncSession,
|
|
user_id: int,
|
|
amount_kopeks: int,
|
|
description: str,
|
|
receipt_email: Optional[str] = None,
|
|
receipt_phone: Optional[str] = None,
|
|
metadata: Optional[Dict[str, Any]] = None,
|
|
) -> Optional[Dict[str, Any]]:
|
|
"""Создаёт платёж по СБП через YooKassa."""
|
|
if not getattr(self, "yookassa_service", None):
|
|
logger.error("YooKassa сервис не инициализирован")
|
|
return None
|
|
|
|
payment_module = import_module("app.services.payment_service")
|
|
|
|
try:
|
|
amount_rubles = amount_kopeks / 100
|
|
|
|
payment_metadata = metadata.copy() if metadata else {}
|
|
payment_metadata.update(
|
|
{
|
|
"user_id": str(user_id),
|
|
"amount_kopeks": str(amount_kopeks),
|
|
"type": "balance_topup_sbp",
|
|
}
|
|
)
|
|
|
|
yookassa_response = (
|
|
await self.yookassa_service.create_sbp_payment(
|
|
amount=amount_rubles,
|
|
currency="RUB",
|
|
description=description,
|
|
metadata=payment_metadata,
|
|
receipt_email=receipt_email,
|
|
receipt_phone=receipt_phone,
|
|
)
|
|
)
|
|
|
|
if not yookassa_response or yookassa_response.get("error"):
|
|
logger.error(
|
|
"Ошибка создания платежа YooKassa СБП: %s",
|
|
yookassa_response,
|
|
)
|
|
return None
|
|
|
|
local_payment = await payment_module.create_yookassa_payment(
|
|
db=db,
|
|
user_id=user_id,
|
|
yookassa_payment_id=yookassa_response["id"],
|
|
amount_kopeks=amount_kopeks,
|
|
currency="RUB",
|
|
description=description,
|
|
status=yookassa_response["status"],
|
|
confirmation_url=yookassa_response.get("confirmation_url"),
|
|
metadata_json=payment_metadata,
|
|
payment_method_type="bank_card",
|
|
yookassa_created_at=None,
|
|
test_mode=yookassa_response.get("test_mode", False),
|
|
)
|
|
|
|
logger.info(
|
|
"Создан платеж YooKassa СБП %s на %s₽ для пользователя %s",
|
|
yookassa_response["id"],
|
|
amount_rubles,
|
|
user_id,
|
|
)
|
|
|
|
confirmation_token = (
|
|
yookassa_response.get("confirmation", {}) or {}
|
|
).get("confirmation_token")
|
|
|
|
return {
|
|
"local_payment_id": local_payment.id,
|
|
"yookassa_payment_id": yookassa_response["id"],
|
|
"confirmation_url": yookassa_response.get("confirmation_url"),
|
|
"confirmation_token": confirmation_token,
|
|
"amount_kopeks": amount_kopeks,
|
|
"amount_rubles": amount_rubles,
|
|
"status": yookassa_response["status"],
|
|
"created_at": local_payment.created_at,
|
|
}
|
|
|
|
except Exception as error:
|
|
logger.error("Ошибка создания платежа YooKassa СБП: %s", error)
|
|
return None
|
|
|
|
async def _process_successful_yookassa_payment(
|
|
self,
|
|
db: AsyncSession,
|
|
payment: "YooKassaPayment",
|
|
) -> bool:
|
|
"""Переносит успешный платёж YooKassa в транзакции и начисляет баланс пользователю."""
|
|
try:
|
|
payment_module = import_module("app.services.payment_service")
|
|
|
|
payment_description = getattr(payment, "description", "YooKassa платеж")
|
|
|
|
transaction = await payment_module.create_transaction(
|
|
db=db,
|
|
user_id=payment.user_id,
|
|
type=TransactionType.DEPOSIT,
|
|
amount_kopeks=payment.amount_kopeks,
|
|
description=f"Пополнение через YooKassa: {payment_description}",
|
|
payment_method=PaymentMethod.YOOKASSA,
|
|
external_id=payment.yookassa_payment_id,
|
|
is_completed=True,
|
|
)
|
|
|
|
await payment_module.link_yookassa_payment_to_transaction(
|
|
db,
|
|
payment.yookassa_payment_id,
|
|
transaction.id,
|
|
)
|
|
|
|
user = await payment_module.get_user_by_id(db, payment.user_id)
|
|
if user:
|
|
old_balance = getattr(user, "balance_kopeks", 0)
|
|
was_first_topup = not getattr(user, "has_made_first_topup", False)
|
|
|
|
user.balance_kopeks += payment.amount_kopeks
|
|
user.updated_at = datetime.utcnow()
|
|
|
|
promo_group = getattr(user, "promo_group", None)
|
|
subscription = getattr(user, "subscription", None)
|
|
referrer_info = format_referrer_info(user)
|
|
topup_status = ("🆕 Первое пополнение" if was_first_topup else "🔄 Пополнение")
|
|
|
|
await db.commit()
|
|
|
|
try:
|
|
from app.services.referral_service import process_referral_topup
|
|
|
|
await process_referral_topup(
|
|
db,
|
|
user.id,
|
|
payment.amount_kopeks,
|
|
getattr(self, "bot", None),
|
|
)
|
|
except Exception as error:
|
|
logger.error(
|
|
"Ошибка обработки реферального пополнения YooKassa: %s",
|
|
error,
|
|
)
|
|
|
|
if was_first_topup and not getattr(user, "has_made_first_topup", False):
|
|
user.has_made_first_topup = True
|
|
await db.commit()
|
|
|
|
await db.refresh(user)
|
|
|
|
if getattr(self, "bot", None):
|
|
try:
|
|
from app.services.admin_notification_service import (
|
|
AdminNotificationService,
|
|
)
|
|
|
|
notification_service = AdminNotificationService(self.bot)
|
|
await notification_service.send_balance_topup_notification(
|
|
user,
|
|
transaction,
|
|
old_balance,
|
|
topup_status=topup_status,
|
|
referrer_info=referrer_info,
|
|
subscription=subscription,
|
|
promo_group=promo_group,
|
|
db=db,
|
|
)
|
|
except Exception as error:
|
|
logger.error(
|
|
"Ошибка отправки уведомления админам о YooKassa пополнении: %s",
|
|
error,
|
|
)
|
|
|
|
if getattr(self, "bot", None):
|
|
try:
|
|
await self._send_payment_success_notification(
|
|
user.telegram_id,
|
|
payment.amount_kopeks,
|
|
user=user,
|
|
db=db,
|
|
payment_method_title="Банковская карта (YooKassa)",
|
|
)
|
|
except Exception as error:
|
|
logger.error(
|
|
"Ошибка отправки уведомления о платеже: %s", error
|
|
)
|
|
|
|
logger.info(
|
|
"Успешно обработан платеж YooKassa %s: пользователь %s получил %s₽",
|
|
payment.yookassa_payment_id,
|
|
payment.user_id,
|
|
payment.amount_kopeks / 100,
|
|
)
|
|
|
|
return True
|
|
|
|
except Exception as error:
|
|
logger.error(
|
|
"Ошибка обработки успешного платежа YooKassa %s: %s",
|
|
payment.yookassa_payment_id,
|
|
error,
|
|
)
|
|
return False
|
|
|
|
async def process_yookassa_webhook(
|
|
self,
|
|
db: AsyncSession,
|
|
event: Dict[str, Any],
|
|
) -> bool:
|
|
"""Обрабатывает входящий webhook YooKassa и синхронизирует состояние платежа."""
|
|
event_object = event.get("object", {})
|
|
yookassa_payment_id = event_object.get("id")
|
|
|
|
if not yookassa_payment_id:
|
|
logger.warning("Webhook без payment id: %s", event)
|
|
return False
|
|
|
|
payment_module = import_module("app.services.payment_service")
|
|
|
|
payment = await payment_module.get_yookassa_payment_by_id(db, yookassa_payment_id)
|
|
if not payment:
|
|
logger.warning(
|
|
"Локальный платеж для YooKassa id %s не найден", yookassa_payment_id
|
|
)
|
|
return False
|
|
|
|
payment.status = event_object.get("status", payment.status)
|
|
payment.confirmation_url = event_object.get("confirmation_url")
|
|
|
|
current_paid = getattr(payment, "paid", False)
|
|
payment.paid = event_object.get("paid", current_paid)
|
|
|
|
await db.commit()
|
|
await db.refresh(payment)
|
|
|
|
if payment.status == "succeeded" and payment.paid:
|
|
return await self._process_successful_yookassa_payment(db, payment)
|
|
|
|
logger.info(
|
|
"Webhook YooKassa обновил платеж %s до статуса %s",
|
|
yookassa_payment_id,
|
|
payment.status,
|
|
)
|
|
return True
|