mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-03-03 20:34:10 +00:00
132 lines
4.5 KiB
Python
132 lines
4.5 KiB
Python
import logging
|
|
from aiogram import Dispatcher, types, F
|
|
from aiogram.fsm.context import FSMContext
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.config import settings
|
|
from app.database.crud.user import get_user_by_telegram_id, update_user
|
|
from app.keyboards.inline import get_main_menu_keyboard
|
|
from app.localization.texts import get_texts
|
|
from app.database.models import User
|
|
from app.utils.user_utils import mark_user_as_had_paid_subscription
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def show_main_menu(
|
|
callback: types.CallbackQuery,
|
|
db_user: User,
|
|
db: AsyncSession
|
|
):
|
|
|
|
texts = get_texts(db_user.language)
|
|
|
|
from datetime import datetime
|
|
db_user.last_activity = datetime.utcnow()
|
|
await db.commit()
|
|
|
|
has_active_subscription = bool(db_user.subscription)
|
|
subscription_is_active = False
|
|
|
|
if db_user.subscription:
|
|
subscription_is_active = db_user.subscription.is_active
|
|
|
|
menu_text = texts.MAIN_MENU.format(
|
|
user_name=db_user.full_name,
|
|
balance=texts.format_price(db_user.balance_kopeks),
|
|
subscription_status=_get_subscription_status(db_user, texts)
|
|
)
|
|
|
|
await callback.message.edit_text(
|
|
menu_text,
|
|
reply_markup=get_main_menu_keyboard(
|
|
language=db_user.language,
|
|
is_admin=settings.is_admin(db_user.telegram_id),
|
|
has_had_paid_subscription=db_user.has_had_paid_subscription,
|
|
has_active_subscription=has_active_subscription,
|
|
subscription_is_active=subscription_is_active
|
|
)
|
|
)
|
|
await callback.answer()
|
|
|
|
async def mark_user_as_had_paid_subscription(
|
|
db: AsyncSession,
|
|
user: User
|
|
) -> None:
|
|
if not user.has_had_paid_subscription:
|
|
user.has_had_paid_subscription = True
|
|
user.updated_at = datetime.utcnow()
|
|
await db.commit()
|
|
logger.info(f"🎯 Пользователь {user.telegram_id} отмечен как имевший платную подписку")
|
|
|
|
|
|
async def show_service_rules(
|
|
callback: types.CallbackQuery,
|
|
db_user: User,
|
|
db: AsyncSession
|
|
):
|
|
from app.database.crud.rules import get_current_rules_content
|
|
|
|
rules_text = await get_current_rules_content(db, db_user.language)
|
|
|
|
if not rules_text:
|
|
texts = get_texts(db_user.language)
|
|
rules_text = texts._get_default_rules(db_user.language) if hasattr(texts, '_get_default_rules') else """
|
|
📋 <b>Правила использования сервиса</b>
|
|
|
|
1. Запрещается использование сервиса для незаконной деятельности
|
|
2. Запрещается нарушение авторских прав
|
|
3. Запрещается спам и рассылка вредоносного ПО
|
|
4. Запрещается использование сервиса для DDoS атак
|
|
5. Один аккаунт - один пользователь
|
|
6. Возврат средств производится только в исключительных случаях
|
|
7. Администрация оставляет за собой право заблокировать аккаунт при нарушении правил
|
|
|
|
<b>Принимая правила, вы соглашаетесь соблюдать их.</b>
|
|
"""
|
|
|
|
await callback.message.edit_text(
|
|
f"📋 <b>Правила сервиса</b>\n\n{rules_text}",
|
|
reply_markup=types.InlineKeyboardMarkup(inline_keyboard=[
|
|
[types.InlineKeyboardButton(text="⬅️ Назад", callback_data="back_to_menu")]
|
|
])
|
|
)
|
|
await callback.answer()
|
|
|
|
|
|
async def handle_back_to_menu(
|
|
callback: types.CallbackQuery,
|
|
state: FSMContext,
|
|
db_user: User,
|
|
db: AsyncSession
|
|
):
|
|
|
|
await state.clear()
|
|
|
|
await show_main_menu(callback, db_user, db)
|
|
|
|
|
|
def _get_subscription_status(user: User, texts) -> str:
|
|
if not user.subscription:
|
|
return texts.SUBSCRIPTION_NONE
|
|
|
|
if user.subscription.is_trial:
|
|
return f"{texts.SUBSCRIPTION_TRIAL} (до {user.subscription.end_date.strftime('%d.%m.%Y')})"
|
|
elif user.subscription.is_active:
|
|
days_left = user.subscription.days_left
|
|
return f"{texts.SUBSCRIPTION_ACTIVE} ({days_left} дн.)"
|
|
else:
|
|
return texts.SUBSCRIPTION_EXPIRED
|
|
|
|
|
|
def register_handlers(dp: Dispatcher):
|
|
|
|
dp.callback_query.register(
|
|
handle_back_to_menu,
|
|
F.data == "back_to_menu"
|
|
)
|
|
|
|
dp.callback_query.register(
|
|
show_service_rules,
|
|
F.data == "menu_rules"
|
|
) |