Vladless
233 lines
9.0 KiB
Python
233 lines
9.0 KiB
Python
import os
|
|
|
|
from typing import Any
|
|
|
|
from aiogram import F, Router
|
|
from aiogram.fsm.context import FSMContext
|
|
from aiogram.types import CallbackQuery, InlineKeyboardButton
|
|
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from config import (
|
|
CRYPTO_BOT_ENABLE,
|
|
DONATIONS_ENABLE,
|
|
FREEKASSA_ENABLE,
|
|
HELEKET_ENABLE,
|
|
KASSAI_ENABLE,
|
|
ROBOKASSA_ENABLE,
|
|
STARS_ENABLE,
|
|
WATA_INT_ENABLE,
|
|
WATA_RU_ENABLE,
|
|
WATA_SBP_ENABLE,
|
|
YOOKASSA_ENABLE,
|
|
YOOMONEY_ENABLE,
|
|
TRIBUTE_ENABLE,
|
|
TRIBUTE_LINK
|
|
)
|
|
from database import get_last_payments
|
|
from database.models import User
|
|
from handlers.buttons import (
|
|
BALANCE_HISTORY,
|
|
COUPON,
|
|
CRYPTOBOT,
|
|
FREEKASSA,
|
|
HELEKET_CRYPTO,
|
|
KASSAI_CARDS,
|
|
KASSAI_SBP,
|
|
MAIN_MENU,
|
|
PAYMENT,
|
|
ROBOKASSA,
|
|
STARS,
|
|
WATA_INT,
|
|
WATA_RU,
|
|
WATA_SBP,
|
|
YOOKASSA,
|
|
YOOMONEY,
|
|
TRIBUTE,
|
|
)
|
|
from handlers.payments.cryprobot_pay import process_callback_pay_cryptobot
|
|
from handlers.payments.freekassa_pay import process_callback_pay_freekassa
|
|
from handlers.payments.heleket import process_callback_pay_heleket
|
|
from handlers.payments.kassai import process_callback_pay_kassai
|
|
from handlers.payments.robokassa_pay import process_callback_pay_robokassa
|
|
from handlers.payments.stars_pay import process_callback_pay_stars
|
|
from handlers.payments.wata import process_callback_pay_wata
|
|
from handlers.payments.yookassa_pay import process_callback_pay_yookassa
|
|
from handlers.payments.yoomoney_pay import process_callback_pay_yoomoney
|
|
from handlers.payments.tribute_pay import process_callback_pay_tribute
|
|
from handlers.texts import BALANCE_MANAGEMENT_TEXT, PAYMENT_METHODS_MSG
|
|
from hooks.hook_buttons import insert_hook_buttons
|
|
from hooks.hooks import run_hooks
|
|
|
|
from .utils import edit_or_send_message
|
|
|
|
|
|
router = Router()
|
|
|
|
|
|
@router.callback_query(F.data == "pay")
|
|
async def handle_pay(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
payment_handlers = []
|
|
|
|
if YOOKASSA_ENABLE:
|
|
payment_handlers.append(process_callback_pay_yookassa)
|
|
if YOOMONEY_ENABLE:
|
|
payment_handlers.append(process_callback_pay_yoomoney)
|
|
if KASSAI_ENABLE:
|
|
payment_handlers.append(process_callback_pay_kassai)
|
|
if HELEKET_ENABLE:
|
|
payment_handlers.append(process_callback_pay_heleket)
|
|
if WATA_RU_ENABLE or WATA_SBP_ENABLE or WATA_INT_ENABLE:
|
|
payment_handlers.append(process_callback_pay_wata)
|
|
if CRYPTO_BOT_ENABLE:
|
|
payment_handlers.append(process_callback_pay_cryptobot)
|
|
if STARS_ENABLE:
|
|
payment_handlers.append(process_callback_pay_stars)
|
|
if ROBOKASSA_ENABLE:
|
|
payment_handlers.append(process_callback_pay_robokassa)
|
|
if FREEKASSA_ENABLE:
|
|
payment_handlers.append(process_callback_pay_freekassa)
|
|
if TRIBUTE_ENABLE:
|
|
payment_handlers.append(process_callback_pay_tribute)
|
|
|
|
module_buttons = await run_hooks(
|
|
"pay_menu_buttons",
|
|
chat_id=callback_query.from_user.id,
|
|
admin=False,
|
|
session=session
|
|
)
|
|
has_extra_menu_items = bool(module_buttons) or bool(DONATIONS_ENABLE) or bool(TRIBUTE_ENABLE)
|
|
|
|
if len(payment_handlers) == 1 and not has_extra_menu_items:
|
|
return await payment_handlers[0](callback_query, state, session)
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
if YOOKASSA_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=YOOKASSA, callback_data="pay_yookassa"))
|
|
if YOOMONEY_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=YOOMONEY, callback_data="pay_yoomoney"))
|
|
if KASSAI_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=KASSAI_CARDS, callback_data="pay_kassai_cards"))
|
|
builder.row(InlineKeyboardButton(text=KASSAI_SBP, callback_data="pay_kassai_sbp"))
|
|
if HELEKET_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=HELEKET_CRYPTO, callback_data="pay_heleket_crypto"))
|
|
if CRYPTO_BOT_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=CRYPTOBOT, callback_data="pay_cryptobot"))
|
|
if STARS_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=STARS, callback_data="pay_stars"))
|
|
if ROBOKASSA_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=ROBOKASSA, callback_data="pay_robokassa"))
|
|
if FREEKASSA_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=FREEKASSA, callback_data="pay_freekassa"))
|
|
if WATA_RU_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=WATA_RU, callback_data="pay_wata_ru"))
|
|
if WATA_SBP_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=WATA_SBP, callback_data="pay_wata_sbp"))
|
|
if WATA_INT_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=WATA_INT, callback_data="pay_wata_int"))
|
|
if TRIBUTE_ENABLE:
|
|
builder.row(InlineKeyboardButton(text=TRIBUTE, url=TRIBUTE_LINK))
|
|
if DONATIONS_ENABLE:
|
|
builder.row(InlineKeyboardButton(text="💰 Поддержать проект", callback_data="donate"))
|
|
|
|
builder = insert_hook_buttons(builder, module_buttons)
|
|
|
|
builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile"))
|
|
|
|
await edit_or_send_message(
|
|
target_message=callback_query.message,
|
|
text=PAYMENT_METHODS_MSG,
|
|
reply_markup=builder.as_markup(),
|
|
)
|
|
|
|
|
|
@router.callback_query(F.data == "balance")
|
|
async def balance_handler(callback_query: CallbackQuery, session: AsyncSession):
|
|
stmt = select(User.balance).where(User.tg_id == callback_query.from_user.id)
|
|
result = await session.execute(stmt)
|
|
balance = result.scalar_one_or_none() or 0.0
|
|
balance = int(balance)
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
builder.row(InlineKeyboardButton(text=PAYMENT, callback_data="pay"))
|
|
builder.row(InlineKeyboardButton(text=BALANCE_HISTORY, callback_data="balance_history"))
|
|
builder.row(InlineKeyboardButton(text=COUPON, callback_data="activate_coupon"))
|
|
builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile"))
|
|
|
|
text = BALANCE_MANAGEMENT_TEXT.format(balance=balance)
|
|
image_path = os.path.join("img", "pay.jpg")
|
|
|
|
await edit_or_send_message(
|
|
target_message=callback_query.message,
|
|
text=text,
|
|
reply_markup=builder.as_markup(),
|
|
media_path=image_path,
|
|
disable_web_page_preview=False,
|
|
)
|
|
|
|
|
|
@router.callback_query(F.data == "balance_history")
|
|
async def balance_history_handler(callback_query: CallbackQuery, session: Any):
|
|
builder = InlineKeyboardBuilder()
|
|
builder.row(InlineKeyboardButton(text=PAYMENT, callback_data="pay"))
|
|
builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile"))
|
|
|
|
records = await get_last_payments(session, callback_query.from_user.id)
|
|
|
|
if records:
|
|
history_text = "<b>💳 История операций:</b>\n\n<blockquote>"
|
|
for record in records:
|
|
amount = record["amount"]
|
|
payment_system = record["payment_system"]
|
|
status = record["status"]
|
|
date = record["created_at"].strftime("%Y-%m-%d %H:%M:%S")
|
|
history_text += f"Сумма: {amount}₽\nОплата: {payment_system}\nСтатус: {status}\nДата: {date}\n\n"
|
|
history_text += "</blockquote>"
|
|
else:
|
|
history_text = "❌ У вас пока нет операций с балансом."
|
|
|
|
await edit_or_send_message(
|
|
target_message=callback_query.message,
|
|
text=history_text,
|
|
reply_markup=builder.as_markup(),
|
|
media_path=None,
|
|
disable_web_page_preview=False,
|
|
)
|
|
|
|
|
|
@router.callback_query(F.data == "pay_wata_ru")
|
|
async def handle_pay_wata_ru(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
await process_callback_pay_wata(callback_query, state, session, cassa_name="ru")
|
|
|
|
|
|
@router.callback_query(F.data == "pay_wata_sbp")
|
|
async def handle_pay_wata_sbp(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
await process_callback_pay_wata(callback_query, state, session, cassa_name="sbp")
|
|
|
|
|
|
@router.callback_query(F.data == "pay_wata_int")
|
|
async def handle_pay_wata_int(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
await process_callback_pay_wata(callback_query, state, session, cassa_name="int")
|
|
|
|
|
|
@router.callback_query(F.data == "pay_kassai_cards")
|
|
async def handle_pay_kassai_cards(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
await process_callback_pay_kassai(callback_query, state, session, method_name="cards")
|
|
|
|
|
|
@router.callback_query(F.data == "pay_kassai_sbp")
|
|
async def handle_pay_kassai_sbp(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
await process_callback_pay_kassai(callback_query, state, session, method_name="sbp")
|
|
|
|
|
|
@router.callback_query(F.data == "pay_heleket_crypto")
|
|
async def handle_pay_heleket_crypto(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
await process_callback_pay_heleket(callback_query, state, session, method_name="crypto")
|
|
|
|
|
|
@router.callback_query(F.data == "pay_tribute")
|
|
async def handle_pay_tribute(callback_query: CallbackQuery, state: FSMContext, session: AsyncSession):
|
|
await process_callback_pay_tribute(callback_query, state, session)
|