hexdev
287f8eebda
- Implemented SubscriptionRepository with methods to get, create, and update subscriptions. - Added BalanceTXRepository for creating and retrieving balance transactions. - Introduced a test script for simulating Pally webhook for local testing.
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from aiogram import F, Router
|
|
from aiogram.fsm.context import FSMContext
|
|
from aiogram.types import CallbackQuery, Message
|
|
|
|
from bot.keyboards.client import payment_gateways, return_to_menu
|
|
from bot.states.buy import BillingStorage
|
|
from bot.states.deposit import DepositStorage
|
|
from bot.texts import (
|
|
CHECKOUT_PAYMENT_CHOICE,
|
|
DEPOSIT_AMOUNT,
|
|
DEPOSIT_AMOUNT_FALLBACK,
|
|
)
|
|
from schemas.billing import BillingContext
|
|
from schemas.common import GeneralMessageContext
|
|
|
|
router = Router()
|
|
|
|
|
|
@router.callback_query(F.data == "deposit")
|
|
async def deposit_init(cb: CallbackQuery, state: FSMContext):
|
|
await state.clear()
|
|
|
|
await cb.message.edit_text(DEPOSIT_AMOUNT, reply_markup=return_to_menu)
|
|
|
|
ctx = GeneralMessageContext(cb.message)
|
|
await state.set_state(DepositStorage.amount)
|
|
await state.set_data({"ctx": ctx})
|
|
|
|
|
|
@router.message(DepositStorage.amount)
|
|
async def deposit_amount(msg: Message, state: FSMContext):
|
|
data = await state.get_data()
|
|
await state.clear()
|
|
await msg.delete()
|
|
|
|
ctx: GeneralMessageContext | None = data.get("ctx")
|
|
if not isinstance(ctx, GeneralMessageContext) or not msg.text.isdigit():
|
|
await msg.answer(DEPOSIT_AMOUNT_FALLBACK, reply_markup=return_to_menu)
|
|
await state.set_state(DepositStorage.amount)
|
|
await state.set_data(data)
|
|
return
|
|
|
|
await ctx.msg.edit_text(CHECKOUT_PAYMENT_CHOICE, reply_markup=payment_gateways)
|
|
await state.set_state(BillingStorage.pending)
|
|
billing_context = BillingContext(amount=int(msg.text))
|
|
await state.set_data({"ctx": billing_context})
|