Zakhar Izmaylov
a2c71a49f2
- Consolidated and simplified code formatting by removing unnecessary line breaks and improving readability in various functions. - Updated the handling of backup file sending in backup.py for better clarity. - Streamlined error handling and logging messages in several handlers to enhance consistency. - Adjusted the Makefile to exclude specific files during Ruff checks and formatting. - Made minor adjustments to function signatures and parameter handling for improved clarity and consistency. This commit enhances code maintainability and readability without altering functionality.
94 lines
3.4 KiB
Python
94 lines
3.4 KiB
Python
from datetime import datetime
|
|
from typing import Any
|
|
|
|
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 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):
|
|
builder = InlineKeyboardBuilder()
|
|
builder.row(InlineKeyboardButton(text="👤 Личный кабинет", callback_data="profile"))
|
|
|
|
await callback_query.message.answer(
|
|
"<b>🎫 Введите код купона:</b>\n\n"
|
|
"📝 Пожалуйста, введите действующий код купона, который вы хотите активировать. 🔑",
|
|
reply_markup=builder.as_markup(),
|
|
)
|
|
await state.set_state(CouponActivationState.waiting_for_coupon_code)
|
|
|
|
|
|
@router.message(CouponActivationState.waiting_for_coupon_code)
|
|
async def process_coupon_code(message: types.Message, state: FSMContext, session: Any):
|
|
coupon_code = message.text.strip()
|
|
activation_result = await activate_coupon(message.chat.id, coupon_code, session)
|
|
|
|
builder = InlineKeyboardBuilder()
|
|
builder.row(InlineKeyboardButton(text="👤 Личный кабинет", callback_data="profile"))
|
|
|
|
await message.answer(activation_result, reply_markup=builder.as_markup())
|
|
await state.clear()
|
|
|
|
|
|
async def activate_coupon(user_id: int, coupon_code: str, session: Any):
|
|
coupon_record = await session.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 session.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 session.transaction():
|
|
await session.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 session.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> 💰."
|