Vladless
101 lines
4.6 KiB
Python
101 lines
4.6 KiB
Python
import os
|
||
from aiogram import Router
|
||
from aiogram.filters import Command
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
from aiogram.types import (BufferedInputFile, CallbackQuery,
|
||
InlineKeyboardButton, InlineKeyboardMarkup, Message)
|
||
|
||
from bot import bot
|
||
from database import add_referral, check_connection_exists, add_connection
|
||
from config import CHANNEL_URL, SUPPORT_CHAT_URL
|
||
|
||
router = Router()
|
||
|
||
class FeedbackState(StatesGroup):
|
||
waiting_for_feedback = State()
|
||
|
||
async def send_welcome_message(chat_id: int):
|
||
welcome_text = (
|
||
"<b>🎉 SoloNet — твой доступ в свободный интернет! 🌐✨</b>\n\n"
|
||
"<b>Наши преимущества:</b>\n"
|
||
"<blockquote>"
|
||
"🚀 <b>Высокая скорость</b>\n"
|
||
"🔄 <b>Стабильность</b>\n"
|
||
"🌍 <b>Смена локаций</b>\n"
|
||
"💬 <b>Отзывчивая поддержка</b>\n"
|
||
"📱💻 <b>Для телефонов, компьютеров и планшетов</b>\n"
|
||
"💰 <b>Реферальная программа: 25% от покупки</b>\n"
|
||
"</blockquote>"
|
||
"\n🎁 <b>1 день бесплатно</b>\n\n"
|
||
"<i>Перейдите в профиль для продолжения 👇</i>"
|
||
)
|
||
|
||
image_path = os.path.join(os.path.dirname(__file__), 'pic.jpg')
|
||
|
||
if not os.path.isfile(image_path):
|
||
await bot.send_message(chat_id, "Файл изображения не найден.")
|
||
return
|
||
|
||
inline_keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
||
[InlineKeyboardButton(text='👤 Мой профиль', callback_data='view_profile')],
|
||
[InlineKeyboardButton(text='🔒 О VPN', callback_data='about_vpn')],
|
||
[InlineKeyboardButton(text='📞 Поддержка', url=SUPPORT_CHAT_URL)],
|
||
[InlineKeyboardButton(text='📢 Наш канал', url=CHANNEL_URL)]
|
||
])
|
||
|
||
with open(image_path, 'rb') as image_from_buffer:
|
||
await bot.send_photo(
|
||
chat_id,
|
||
BufferedInputFile(image_from_buffer.read(), filename="pic.jpg"),
|
||
caption=welcome_text,
|
||
parse_mode='HTML',
|
||
reply_markup=inline_keyboard
|
||
)
|
||
|
||
@router.message(Command('start'))
|
||
async def start_command(message: Message):
|
||
print(f"Received start command with text: {message.text}")
|
||
if 'referral_' in message.text:
|
||
referrer_tg_id = int(message.text.split('referral_')[1])
|
||
print(f"Referral ID: {referrer_tg_id}")
|
||
if not await check_connection_exists(message.from_user.id):
|
||
await add_connection(message.from_user.id)
|
||
|
||
await add_referral(message.from_user.id, referrer_tg_id)
|
||
|
||
await message.answer("Вас пригласил друг, добро пожаловать!")
|
||
else:
|
||
await message.answer("Вы уже зарегистрированы в системе!")
|
||
await send_welcome_message(message.chat.id)
|
||
|
||
@router.callback_query(lambda c: c.data == 'about_vpn')
|
||
async def handle_about_vpn(callback_query: CallbackQuery):
|
||
await callback_query.message.delete()
|
||
info_message = (
|
||
"*О VPN*\n\n"
|
||
"<b>🚀 Высокоскоростные серверы</b>\n"
|
||
"Мы используем высокоскоростные серверы в различных локациях для обеспечения стабильного и быстрого соединения.\n\n"
|
||
|
||
"<b>🔐 Безопасность данных</b>\n"
|
||
"Для защиты ваших данных мы применяем новейшие протоколы шифрования, которые гарантируют вашу конфиденциальность.\n\n"
|
||
|
||
"<b>🔑 Ваш ключ — ваша безопасность!</b>\n"
|
||
"Не передавайте своё шифрование сторонним лицам, чтобы избежать рисков.\n"
|
||
)
|
||
|
||
button_back = InlineKeyboardButton(text='⬅️ Назад', callback_data='back_to_menu')
|
||
inline_keyboard_back = InlineKeyboardMarkup(inline_keyboard=[[button_back]])
|
||
|
||
await callback_query.message.answer(
|
||
info_message,
|
||
parse_mode='HTML',
|
||
reply_markup=inline_keyboard_back
|
||
)
|
||
await callback_query.answer()
|
||
|
||
@router.callback_query(lambda c: c.data == 'back_to_menu')
|
||
async def handle_back_to_menu(callback_query: CallbackQuery):
|
||
await callback_query.message.delete()
|
||
await send_welcome_message(callback_query.from_user.id)
|
||
await callback_query.answer()
|