Merge pull request #119 from TrackLine/main

Add Yoomoney pay service
2024-12-14 19:07:25 +03:00
parent ca87286898 7677ae9193
commit bbb554e4e9
4 changed files with 222 additions and 0 deletions
@@ -7,6 +7,7 @@ from config import (
    ROBOKASSA_ENABLE,
    STARS_ENABLE,
    YOOKASSA_ENABLE,
+    YOOMONEY_ENABLE
 )

 router = Router()
@@ -23,6 +24,13 @@ async def handle_pay(callback_query: CallbackQuery):
                callback_data="pay_yookassa",
            )
        )
+    if YOOMONEY_ENABLE:
+        builder.row(
+            InlineKeyboardButton(
+                text="💳 ЮМани: оплата по карте",
+                callback_data="pay_yoomoney",
+            )
+        )
    if FREEKASSA_ENABLE:
        builder.row(
            InlineKeyboardButton(
@@ -7,6 +7,7 @@ from config import (
    ROBOKASSA_ENABLE,
    STARS_ENABLE,
    YOOKASSA_ENABLE,
+    YOOMONEY_ENABLE,
 )

 from .cryprobot_pay import router as cryprobot_router
@@ -14,11 +15,14 @@ from .freekassa_pay import router as freekassa_router
 from .robokassa_pay import router as robokassa_router
 from .stars_pay import router as stars_router
 from .yookassa_pay import router as yookassa_router
+from .yoomoney_pay import router as yoomoney_router

 router = Router(name="payments_main_router")

 if YOOKASSA_ENABLE:
    router.include_router(yookassa_router)
+if YOOMONEY_ENABLE:
+    router.include_router(yoomoney_router)
 if FREEKASSA_ENABLE:
    router.include_router(freekassa_router)
 if ROBOKASSA_ENABLE:
@@ -0,0 +1,206 @@
+import uuid
+from typing import Any
+
+from aiogram import F, Router, types
+from aiogram.fsm.context import FSMContext
+from aiogram.fsm.state import State, StatesGroup
+from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
+from aiogram.utils.keyboard import InlineKeyboardBuilder
+from aiohttp import web
+from config import YOOMONEY_ENABLE, YOOMONEY_ID, YOOMONEY_SECRET_KEY
+
+from database import (
+    add_connection,
+    add_payment,
+    check_connection_exists,
+    get_key_count,
+    update_balance,
+)
+from handlers.payments.utils import send_payment_success_notification
+from handlers.texts import PAYMENT_OPTIONS
+from logger import logger
+import hashlib
+
+router = Router()
+
+if YOOMONEY_ENABLE:
+    logger.debug(f"Account ID: {YOOMONEY_ID}")
+
+
+class ReplenishBalanceState(StatesGroup):
+    choosing_amount_yoomoney = State()
+    waiting_for_payment_confirmation_yoomoney = State()
+    entering_custom_amount_yoomoney = State()
+
+
+@router.callback_query(F.data == "pay_yoomoney")
+async def process_callback_pay_yoomoney(
+    callback_query: types.CallbackQuery, state: FSMContext, session: Any
+):
+    tg_id = callback_query.message.chat.id
+
+    builder = InlineKeyboardBuilder()
+
+    for i in range(0, len(PAYMENT_OPTIONS), 2):
+        if i + 1 < len(PAYMENT_OPTIONS):
+            builder.row(
+                InlineKeyboardButton(
+                    text=PAYMENT_OPTIONS[i]["text"],
+                    callback_data=f'yoomoney_{PAYMENT_OPTIONS[i]["callback_data"]}',
+                ),
+                InlineKeyboardButton(
+                    text=PAYMENT_OPTIONS[i + 1]["text"],
+                    callback_data=f'yoomoney_{PAYMENT_OPTIONS[i + 1]["callback_data"]}',
+                ),
+            )
+        else:
+            builder.row(
+                InlineKeyboardButton(
+                    text=PAYMENT_OPTIONS[i]["text"],
+                    callback_data=f'yoomoney_{PAYMENT_OPTIONS[i]["callback_data"]}',
+                )
+            )
+    builder.row(
+        InlineKeyboardButton(
+            text="💰 Ввести свою сумму",
+            callback_data="enter_custom_amount_yoomoney",
+        )
+    )
+    builder.row(InlineKeyboardButton(text="⬅️ Назад", callback_data="pay"))
+
+    key_count = await get_key_count(tg_id)
+
+    if key_count == 0:
+        exists = await check_connection_exists(tg_id)
+        if not exists:
+            await add_connection(tg_id, balance=0.0, trial=0, session=session)
+
+    await callback_query.message.answer(
+        text="Выберите сумму пополнения:",
+        reply_markup=builder.as_markup(),
+    )
+
+    await state.set_state(ReplenishBalanceState.choosing_amount_yoomoney)
+
+
+@router.callback_query(F.data.startswith("yoomoney_amount"))
+async def process_amount_selection(
+    callback_query: types.CallbackQuery, state: FSMContext
+):
+    data = callback_query.data.split("|", 1)
+
+    if len(data) != 2:
+        return
+
+    amount_str = data[1]
+    try:
+        amount = int(amount_str)
+    except ValueError:
+        return
+
+    await state.update_data(amount=amount)
+    await state.set_state(
+        ReplenishBalanceState.waiting_for_payment_confirmation_yoomoney
+    )
+
+    # state_data = await state.get_data()
+    #customer_name = callback_query.from_user.full_name
+    customer_id = callback_query.message.chat.id
+    account_id = YOOMONEY_ID
+    payment_url = f"https://yoomoney.ru/quickpay/confirm.xml?receiver={account_id}&quickpay-form=shop&targets=&sum={amount}&paymentType=PC&comment=Пополнение баланса&label={customer_id}"
+
+    confirm_keyboard = InlineKeyboardMarkup(
+            inline_keyboard=[
+                [InlineKeyboardButton(text="Пополнить", url=payment_url)],
+                [InlineKeyboardButton(text="⬅️ Назад", callback_data="pay")],
+            ]
+        )
+
+    await callback_query.message.answer(
+            text=f"Вы выбрали пополнение на {amount} рублей.",
+            reply_markup=confirm_keyboard,
+        )
+
+
+async def yoomoney_webhook(request: web.Request):
+    data = await request.post()
+    logger.debug(f"Webhook event received: {data}")
+
+    user_id_str = data.get("label")
+    amount_str = data.get("withdraw_amount")
+    notification_secret = YOOMONEY_SECRET_KEY
+    sha1_hash = data.get("sha1_hash")
+
+    # Строим строку для хэширования
+    string_to_hash = f"{data.get('notification_type')}&{data.get('operation_id')}&{data.get('amount')}&{data.get('currency')}&{data.get('datetime')}&{data.get('sender')}&{data.get('codepro')}&{notification_secret}&{user_id_str}"
+
+    # Высчитываем хэш
+    calculated_hash = hashlib.sha1(string_to_hash.encode('utf-8')).hexdigest()
+
+    # Проверяем хэш
+    if calculated_hash != sha1_hash:
+        logger.error("Проверка хэша не пройдена")
+        return web.Response(status=400)
+
+    try:
+        user_id = int(user_id_str)
+        amount = float(amount_str)
+        logger.debug(f"Payment succeeded for user_id: {user_id}, amount: {amount}")
+        await add_payment(user_id, amount, "yoomoney")
+        await update_balance(user_id, amount)
+        await send_payment_success_notification(user_id, amount)
+    except ValueError as e:
+        logger.error(f"Ошибка конвертации user_id или amount: {e}")
+        return web.Response(status=400)
+
+    return web.Response(status=200)
+
+
+@router.callback_query(F.data == "enter_custom_amount_yoomoney")
+async def process_enter_custom_amount(
+    callback_query: types.CallbackQuery, state: FSMContext
+):
+    builder = InlineKeyboardBuilder()
+    builder.row(InlineKeyboardButton(text="🔙 Назад", callback_data="pay_yoomoney"))
+
+    await callback_query.message.answer(
+        "Пожалуйста, введите сумму пополнения.",
+        reply_markup=builder.as_markup(),
+    )
+
+    await state.set_state(ReplenishBalanceState.entering_custom_amount_yoomoney)
+
+
+@router.message(ReplenishBalanceState.entering_custom_amount_yoomoney)
+async def process_custom_amount_input(message: types.Message, state: FSMContext):
+    if message.text.isdigit():
+        amount = int(message.text)
+        if amount <= 0:
+            await message.answer(
+                "Сумма должна быть больше нуля. Пожалуйста, введите сумму еще раз:"
+            )
+            return
+
+        await state.update_data(amount=amount)
+        await state.set_state(
+            ReplenishBalanceState.waiting_for_payment_confirmation_yoomoney
+        )
+
+        customer_id = message.chat.id
+        account_id = YOOMONEY_ID
+        payment_url = f"https://yoomoney.ru/quickpay/confirm.xml?receiver={account_id}&quickpay-form=shop&targets=&sum={amount}&paymentType=PC&comment=Пополнение баланса&label={customer_id}"
+
+        confirm_keyboard = InlineKeyboardMarkup(
+                inline_keyboard=[
+                    [InlineKeyboardButton(text="Пополнить", url=payment_url)],
+                    [InlineKeyboardButton(text="⬅️ Назад", callback_data="pay")],
+                ]
+            )
+
+        await message.answer(
+                text=f"Вы выбрали пополнение на {amount} рублей.",
+                reply_markup=confirm_keyboard,
+        ) 
+
+    else:
+        await message.answer("Некорректная сумма. Пожалуйста, введите сумму еще раз:")
@@ -16,6 +16,7 @@ from config import (
    WEBHOOK_PATH,
    WEBHOOK_URL,
    YOOKASSA_ENABLE,
+    YOOMONEY_ENABLE,
 )

 from backup import backup_database
@@ -28,6 +29,7 @@ from handlers.payments.cryprobot_pay import cryptobot_webhook
 from handlers.payments.freekassa_pay import freekassa_webhook
 from handlers.payments.robokassa_pay import robokassa_webhook
 from handlers.payments.yookassa_pay import yookassa_webhook
+from handlers.payments.yoomoney_pay import yoomoney_webhook
 from logger import logger
 from servers import check_servers, sync_servers_with_db

@@ -85,6 +87,8 @@ async def main():
        app.on_shutdown.append(on_shutdown)
        if YOOKASSA_ENABLE:
            app.router.add_post("/yookassa/webhook", yookassa_webhook)
+        if YOOMONEY_ENABLE:
+            app.router.add_post("/yoomoney/webhook", yoomoney_webhook)
        if FREEKASSA_ENABLE:
            app.router.add_post("/freekassa/webhook", freekassa_webhook)
        if CRYPTO_BOT_ENABLE: