55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
from aiogram import Router, types
|
|
from aiogram.filters import CommandStart, Command
|
|
from aiogram.types import LabeledPrice, PreCheckoutQuery
|
|
|
|
from bot.handlers.user.model_processing import PayCallBack
|
|
from bot.handlers.user.page_processing import get_page_content
|
|
from bot.kbs.inline import PageCallBack
|
|
from config import shop_api_token
|
|
|
|
user = Router()
|
|
|
|
@user.message(CommandStart())
|
|
async def start(message: types.Message):
|
|
text, reply_markup = await get_page_content(page=0,name="main")
|
|
|
|
await message.answer(text, reply_markup=reply_markup)
|
|
|
|
|
|
@user.callback_query(PageCallBack.filter())
|
|
async def user_pages(callback: types.CallbackQuery, callback_data: PageCallBack):
|
|
|
|
text, reply_markup = await get_page_content(
|
|
page=callback_data.page,
|
|
name=callback_data.page_name,
|
|
)
|
|
|
|
await callback.message.edit_text(text, reply_markup=reply_markup)
|
|
await callback.answer()
|
|
|
|
@user.callback_query(PayCallBack.filter())
|
|
async def buy(callback: types.CallbackQuery, callback_data: PayCallBack):
|
|
|
|
await callback.message.bot.send_invoice(
|
|
chat_id=callback.from_user.id,
|
|
description=f'{callback_data.tokens} токенов за {callback_data.amount} RUB для модели {callback_data.name}',
|
|
title=f'{callback_data.option_of_model} подписка на {callback_data.name}',
|
|
payload=f'sub_{callback_data.name}',
|
|
provider_token=shop_api_token,
|
|
start_parameter='test',
|
|
currency="RUB",
|
|
prices=[LabeledPrice(
|
|
label=f'Оплата токенов для {callback_data.option_of_model}',
|
|
amount=int(callback_data.amount)*100,
|
|
)],
|
|
)
|
|
|
|
@user.pre_checkout_query()
|
|
async def procces_pre_checkout_query(pre_checkout_query: PreCheckoutQuery):
|
|
await pre_checkout_query.answer(ok=True)
|
|
|
|
|
|
|
|
|
|
|