Zakhar Izmaylov
129 lines
4.4 KiB
Python
129 lines
4.4 KiB
Python
from datetime import datetime
|
|
|
|
import asyncpg
|
|
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 config import DATABASE_URL
|
|
from database import update_balance
|
|
|
|
|
|
class CouponActivationState(StatesGroup):
|
|
waiting_for_coupon_code = State()
|
|
|
|
|
|
router = Router()
|
|
|
|
|
|
@router.callback_query(F.data == "activate_coupon")
|
|
async def handle_activate_coupon(
|
|
callback_query: types.CallbackQuery, state: FSMContext
|
|
):
|
|
try:
|
|
await callback_query.message.delete()
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при удалении сообщения: {e}")
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
builder.row(
|
|
InlineKeyboardButton(text="⬅️ Вернуться в профиль", callback_data="view_profile")
|
|
)
|
|
|
|
await callback_query.message.answer(
|
|
"<b>Введите код купона:</b>\n\n"
|
|
"Пожалуйста, введите действующий код купона, который вы хотите активировать.",
|
|
parse_mode="HTML",
|
|
reply_markup=builder.as_markup(),
|
|
)
|
|
await state.set_state(CouponActivationState.waiting_for_coupon_code)
|
|
await callback_query.answer()
|
|
|
|
|
|
@router.message(CouponActivationState.waiting_for_coupon_code)
|
|
async def process_coupon_code(message: types.Message, state: FSMContext):
|
|
try:
|
|
await message.delete()
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при удалении сообщения: {e}")
|
|
|
|
coupon_code = message.text.strip()
|
|
activation_result = await activate_coupon(message.from_user.id, coupon_code)
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
builder.row(
|
|
InlineKeyboardButton(text="👤 Личный кабинет", callback_data="view_profile")
|
|
)
|
|
|
|
markup = builder.as_markup()
|
|
|
|
await message.answer(activation_result, reply_markup=markup, parse_mode="HTML")
|
|
await state.clear()
|
|
|
|
|
|
async def activate_coupon(user_id: int, coupon_code: str):
|
|
"""Функция для активации купона"""
|
|
conn = await asyncpg.connect(DATABASE_URL)
|
|
|
|
try:
|
|
coupon_record = await conn.fetchrow(
|
|
"""
|
|
SELECT id, usage_limit, usage_count, is_used, amount
|
|
FROM coupons
|
|
WHERE code = $1 AND (usage_count < usage_limit OR usage_limit = 0) AND is_used = FALSE
|
|
""",
|
|
coupon_code,
|
|
)
|
|
|
|
if not coupon_record:
|
|
return "<b>❌ Купон не найден</b> или его использование ограничено. Пожалуйста, проверьте код и попробуйте снова."
|
|
|
|
usage_exists = await conn.fetchrow(
|
|
"""
|
|
SELECT 1 FROM coupon_usages WHERE coupon_id = $1 AND user_id = $2
|
|
""",
|
|
coupon_record["id"],
|
|
user_id,
|
|
)
|
|
|
|
if usage_exists:
|
|
return "<b>❌ Вы уже активировали этот купон.</b> Купоны могут быть активированы только один раз."
|
|
|
|
coupon_amount = coupon_record["amount"]
|
|
|
|
async with conn.transaction():
|
|
await conn.execute(
|
|
"""
|
|
UPDATE coupons
|
|
SET usage_count = usage_count + 1,
|
|
is_used = CASE WHEN usage_count + 1 >= usage_limit AND usage_limit > 0 THEN TRUE ELSE FALSE END
|
|
WHERE id = $1
|
|
""",
|
|
coupon_record["id"],
|
|
)
|
|
|
|
await conn.execute(
|
|
"""
|
|
INSERT INTO coupon_usages (coupon_id, user_id, used_at)
|
|
VALUES ($1, $2, $3)
|
|
""",
|
|
coupon_record["id"],
|
|
user_id,
|
|
datetime.utcnow(),
|
|
)
|
|
|
|
await update_balance(user_id, coupon_amount)
|
|
return f"<b>✅ Купон успешно активирован!</b>\n\nНа ваш баланс добавлено <b>{coupon_amount} рублей</b>."
|
|
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при активации купона: {e}")
|
|
return (
|
|
"<b>⚠️ Произошла ошибка при активации купона.</b>\nПопробуйте ещё раз позже."
|
|
)
|
|
|
|
finally:
|
|
await conn.close()
|