60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
from aiogram import Router, types
|
|
from aiogram.filters import Command
|
|
from aiogram.enums.parse_mode import ParseMode
|
|
import logging
|
|
|
|
router = Router()
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def _build_referral_text(bot, user_id: int) -> str:
|
|
me = await bot.get_me()
|
|
bot_username = me.username or "LarkVPN_bot"
|
|
|
|
link = f"https://t.me/{bot_username}?start=ref_{user_id}"
|
|
|
|
text = (
|
|
"👥 <b>Реферальная программа</b>\n\n"
|
|
"Зови друзей в Lark VPN и получай бонусы на баланс.\n\n"
|
|
f"🔗 Твоя ссылка:\n<code>{link}</code>\n\n"
|
|
"👤 Приглашено: —\n"
|
|
"💰 Начислено бонусов: — ₽\n\n"
|
|
"Бонусы падают автоматически, когда приглашённые пополняют баланс."
|
|
)
|
|
return text
|
|
|
|
|
|
@router.message(Command("referrals"))
|
|
async def referrals_command(message: types.Message):
|
|
"""
|
|
Команда /referrals — показывает текст реферальной программы.
|
|
"""
|
|
logger.info(f"Получена команда /referrals от {message.from_user.id}")
|
|
try:
|
|
text = await _build_referral_text(message.bot, message.from_user.id)
|
|
await message.answer(text, parse_mode=ParseMode.HTML)
|
|
except Exception as e:
|
|
logger.exception(f"Ошибка в обработчике /referrals: {e}")
|
|
await message.answer("Произошла ошибка. Попробуй позже.")
|
|
|
|
|
|
@router.callback_query(lambda callback: callback.data == "referral")
|
|
async def referrals_callback(callback: types.CallbackQuery):
|
|
"""
|
|
Кнопка «Реферальная программа» в главном меню.
|
|
"""
|
|
try:
|
|
text = await _build_referral_text(
|
|
callback.message.bot,
|
|
callback.from_user.id,
|
|
)
|
|
await callback.message.edit_text(
|
|
text,
|
|
parse_mode=ParseMode.HTML,
|
|
)
|
|
except Exception as e:
|
|
logger.exception(f"Ошибка в обработчике callback 'referral': {e}")
|
|
await callback.message.answer("Произошла ошибка. Попробуй позже.")
|
|
finally:
|
|
await callback.answer()
|