|
|
|
@@ -0,0 +1,228 @@
|
|
|
|
|
from aiogram import F, Router, types
|
|
|
|
|
from aiogram.fsm.context import FSMContext
|
|
|
|
|
from aiogram.fsm.state import State, StatesGroup
|
|
|
|
|
from aiogram.types import InlineKeyboardButton
|
|
|
|
|
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
|
|
|
|
from loguru import logger
|
|
|
|
|
from telegram import LabeledPrice
|
|
|
|
|
|
|
|
|
|
from bot import bot
|
|
|
|
|
from config import RUB_TO_XTR
|
|
|
|
|
from database import add_connection, check_connection_exists, get_key_count, update_balance
|
|
|
|
|
from handlers.profile import process_callback_view_profile
|
|
|
|
|
from handlers.texts import PAYMENT_OPTIONS
|
|
|
|
|
|
|
|
|
|
router = Router()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class ReplenishBalanceState(StatesGroup):
|
|
|
|
|
choosing_amount = State()
|
|
|
|
|
waiting_for_payment_confirmation = State()
|
|
|
|
|
entering_custom_amount = State()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def send_message_with_deletion(
|
|
|
|
|
chat_id, text, reply_markup=None, state=None, message_key="last_message_id"
|
|
|
|
|
):
|
|
|
|
|
if state:
|
|
|
|
|
try:
|
|
|
|
|
state_data = await state.get_data()
|
|
|
|
|
previous_message_id = state_data.get(message_key)
|
|
|
|
|
|
|
|
|
|
if previous_message_id:
|
|
|
|
|
await bot.delete_message(
|
|
|
|
|
chat_id=chat_id, message_id=previous_message_id
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
sent_message = await bot.send_message(
|
|
|
|
|
chat_id=chat_id, text=text, reply_markup=reply_markup
|
|
|
|
|
)
|
|
|
|
|
await state.update_data({message_key: sent_message.message_id})
|
|
|
|
|
|
|
|
|
|
except Exception as e:
|
|
|
|
|
logger.error(f"Ошибка при удалении/отправке сообщения: {e}")
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
return sent_message
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.callback_query(F.data == "replenish_balance")
|
|
|
|
|
async def process_callback_replenish_balance(
|
|
|
|
|
callback_query: types.CallbackQuery, state: FSMContext
|
|
|
|
|
):
|
|
|
|
|
tg_id = callback_query.from_user.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=PAYMENT_OPTIONS[i]["callback_data"],
|
|
|
|
|
),
|
|
|
|
|
InlineKeyboardButton(
|
|
|
|
|
text=PAYMENT_OPTIONS[i + 1]["text"],
|
|
|
|
|
callback_data=PAYMENT_OPTIONS[i + 1]["callback_data"],
|
|
|
|
|
),
|
|
|
|
|
)
|
|
|
|
|
else:
|
|
|
|
|
builder.row(
|
|
|
|
|
InlineKeyboardButton(
|
|
|
|
|
text=PAYMENT_OPTIONS[i]["text"],
|
|
|
|
|
callback_data=PAYMENT_OPTIONS[i]["callback_data"],
|
|
|
|
|
)
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
await bot.delete_message(
|
|
|
|
|
chat_id=tg_id, message_id=callback_query.message.message_id
|
|
|
|
|
)
|
|
|
|
|
except Exception as e:
|
|
|
|
|
logger.error(f"Не удалось удалить сообщение: {e}")
|
|
|
|
|
|
|
|
|
|
await bot.send_message(
|
|
|
|
|
chat_id=tg_id,
|
|
|
|
|
text="Выберите сумму пополнения:",
|
|
|
|
|
reply_markup=builder.as_markup(),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
await state.set_state(ReplenishBalanceState.choosing_amount)
|
|
|
|
|
await callback_query.answer()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.callback_query(F.data == "back_to_profile")
|
|
|
|
|
async def back_to_profile_handler(
|
|
|
|
|
callback_query: types.CallbackQuery, state: FSMContext
|
|
|
|
|
):
|
|
|
|
|
await process_callback_view_profile(callback_query, state)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.callback_query(F.data.startswith("amount|"))
|
|
|
|
|
async def process_amount_selection(
|
|
|
|
|
callback_query: types.CallbackQuery, state: FSMContext
|
|
|
|
|
):
|
|
|
|
|
data = callback_query.data.split("|", 1)
|
|
|
|
|
|
|
|
|
|
if len(data) != 2:
|
|
|
|
|
await send_message_with_deletion(
|
|
|
|
|
callback_query.from_user.id,
|
|
|
|
|
"Неверные данные для выбора суммы.",
|
|
|
|
|
state=state,
|
|
|
|
|
message_key="amount_error_message_id",
|
|
|
|
|
)
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
amount_str = data[1]
|
|
|
|
|
try:
|
|
|
|
|
amount = int(amount_str)
|
|
|
|
|
except ValueError:
|
|
|
|
|
await send_message_with_deletion(
|
|
|
|
|
callback_query.from_user.id,
|
|
|
|
|
"Некорректная сумма.",
|
|
|
|
|
state=state,
|
|
|
|
|
message_key="amount_error_message_id",
|
|
|
|
|
)
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
await state.update_data(amount=amount)
|
|
|
|
|
await state.set_state(ReplenishBalanceState.waiting_for_payment_confirmation)
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
|
builder.row(
|
|
|
|
|
InlineKeyboardButton(text="Пополнить", pay=True),
|
|
|
|
|
InlineKeyboardButton(text="⬅️ Назад", callback_data="back_to_profile"),
|
|
|
|
|
)
|
|
|
|
|
await callback_query.message.answer_invoice(
|
|
|
|
|
title=f"Вы выбрали пополнение на {amount//RUB_TO_XTR} рублей.",
|
|
|
|
|
description=f"Вы выбрали пополнение на {amount//RUB_TO_XTR} рублей.",
|
|
|
|
|
prices=[LabeledPrice(label="XTR", amount=amount)],
|
|
|
|
|
provider_token="",
|
|
|
|
|
payload=f"{amount}_stars",
|
|
|
|
|
currency="XTR",
|
|
|
|
|
reply_markup=builder.as_markup(),
|
|
|
|
|
)
|
|
|
|
|
except Exception as e:
|
|
|
|
|
logger.error(f"Ошибка при создании платежа: {e}")
|
|
|
|
|
await callback_query.message.answer("Произошла ошибка при создании платежа.")
|
|
|
|
|
await callback_query.answer()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def send_payment_success_notification(user_id: int, amount: float):
|
|
|
|
|
try:
|
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
|
builder.row(
|
|
|
|
|
InlineKeyboardButton(text="Перейти в профиль", callback_data="view_profile")
|
|
|
|
|
)
|
|
|
|
|
await bot.send_message(
|
|
|
|
|
chat_id=user_id,
|
|
|
|
|
text=f"Ваш баланс успешно пополнен на {amount} рублей. Спасибо за оплату!",
|
|
|
|
|
reply_markup=builder.as_markup(),
|
|
|
|
|
)
|
|
|
|
|
except Exception as e:
|
|
|
|
|
logger.error(f"Ошибка при отправке уведомления пользователю {user_id}: {e}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.callback_query(F.data == "enter_custom_amount")
|
|
|
|
|
async def process_enter_custom_amount(
|
|
|
|
|
callback_query: types.CallbackQuery, state: FSMContext
|
|
|
|
|
):
|
|
|
|
|
await callback_query.message.edit_text(text="Введите сумму пополнения:")
|
|
|
|
|
await state.set_state(ReplenishBalanceState.entering_custom_amount)
|
|
|
|
|
await callback_query.answer()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.message(State(ReplenishBalanceState.entering_custom_amount))
|
|
|
|
|
async def process_custom_amount_input(message: types.Message, state: FSMContext):
|
|
|
|
|
if message.text.isdigit():
|
|
|
|
|
amount = int(message.text)
|
|
|
|
|
if amount // RUB_TO_XTR <= 0:
|
|
|
|
|
await message.answer(
|
|
|
|
|
f"Сумма должна быть больше {RUB_TO_XTR}. Пожалуйста, введите сумму еще раз:"
|
|
|
|
|
)
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
await state.update_data(amount=amount)
|
|
|
|
|
await state.set_state(ReplenishBalanceState.waiting_for_payment_confirmation)
|
|
|
|
|
try:
|
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
|
|
|
builder.row(
|
|
|
|
|
InlineKeyboardButton(text="Пополнить", pay=True),
|
|
|
|
|
InlineKeyboardButton(text="⬅️ Назад", callback_data="back_to_profile"),
|
|
|
|
|
)
|
|
|
|
|
await message.answer_invoice(
|
|
|
|
|
title=f"Вы выбрали пополнение на {amount//RUB_TO_XTR} рублей.",
|
|
|
|
|
description=f"Вы выбрали пополнение на {amount//RUB_TO_XTR} рублей.",
|
|
|
|
|
prices=[LabeledPrice(label="XTR", amount=amount)],
|
|
|
|
|
provider_token="",
|
|
|
|
|
payload=f"{amount}_stars",
|
|
|
|
|
currency="XTR",
|
|
|
|
|
reply_markup=builder.as_markup(),
|
|
|
|
|
)
|
|
|
|
|
except Exception as e:
|
|
|
|
|
logger.error(f"Ошибка при создании платежа: {e}")
|
|
|
|
|
await message.answer("Произошла ошибка при создании платежа.")
|
|
|
|
|
else:
|
|
|
|
|
await message.answer("Некорректная сумма. Пожалуйста, введите сумму еще раз:")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.message(F.successful_payment)
|
|
|
|
|
async def on_successful_payment(
|
|
|
|
|
message: types.Message,
|
|
|
|
|
):
|
|
|
|
|
try:
|
|
|
|
|
user_id = int(message.from_user.id)
|
|
|
|
|
amount = float(message.successful_payment.total_amount)
|
|
|
|
|
logger.debug(f"Payment succeeded for user_id: {user_id}, amount: {amount}")
|
|
|
|
|
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}")
|
Vladislav Lisitsyn