mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-02-25 22:00:54 +00:00
9a2aea038a
- Add pyproject.toml with uv and ruff configuration - Pin Python version to 3.13 via .python-version - Add Makefile commands: lint, format, fix - Apply ruff formatting to entire codebase - Remove unused imports (base64 in yookassa/simple_subscription) - Update .gitignore for new config files
409 lines
16 KiB
Python
409 lines
16 KiB
Python
import logging
|
||
|
||
from aiogram import Dispatcher, F, types
|
||
from sqlalchemy import func, select
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.database.crud.subscription import (
|
||
get_all_subscriptions,
|
||
get_expired_subscriptions,
|
||
get_expiring_subscriptions,
|
||
get_subscriptions_statistics,
|
||
)
|
||
from app.database.models import User
|
||
from app.utils.decorators import admin_required, error_handler
|
||
from app.utils.formatters import format_datetime
|
||
|
||
|
||
def get_country_flag(country_name: str) -> str:
|
||
flags = {
|
||
'USA': '🇺🇸',
|
||
'United States': '🇺🇸',
|
||
'US': '🇺🇸',
|
||
'Germany': '🇩🇪',
|
||
'DE': '🇩🇪',
|
||
'Deutschland': '🇩🇪',
|
||
'Netherlands': '🇳🇱',
|
||
'NL': '🇳🇱',
|
||
'Holland': '🇳🇱',
|
||
'United Kingdom': '🇬🇧',
|
||
'UK': '🇬🇧',
|
||
'GB': '🇬🇧',
|
||
'Japan': '🇯🇵',
|
||
'JP': '🇯🇵',
|
||
'France': '🇫🇷',
|
||
'FR': '🇫🇷',
|
||
'Canada': '🇨🇦',
|
||
'CA': '🇨🇦',
|
||
'Russia': '🇷🇺',
|
||
'RU': '🇷🇺',
|
||
'Singapore': '🇸🇬',
|
||
'SG': '🇸🇬',
|
||
}
|
||
return flags.get(country_name, '🌍')
|
||
|
||
|
||
async def get_users_by_countries(db: AsyncSession) -> dict:
|
||
try:
|
||
result = await db.execute(
|
||
select(User.preferred_location, func.count(User.id))
|
||
.where(User.preferred_location.isnot(None))
|
||
.group_by(User.preferred_location)
|
||
)
|
||
|
||
stats = {}
|
||
for location, count in result.fetchall():
|
||
if location:
|
||
stats[location] = count
|
||
|
||
return stats
|
||
except Exception as e:
|
||
logger.error(f'Ошибка получения статистики по странам: {e}')
|
||
return {}
|
||
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def show_subscriptions_menu(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||
stats = await get_subscriptions_statistics(db)
|
||
|
||
text = f"""
|
||
📱 <b>Управление подписками</b>
|
||
|
||
📊 <b>Статистика:</b>
|
||
- Всего: {stats['total_subscriptions']}
|
||
- Активных: {stats['active_subscriptions']}
|
||
- Платных: {stats['paid_subscriptions']}
|
||
- Триальных: {stats['trial_subscriptions']}
|
||
|
||
📈 <b>Продажи:</b>
|
||
- Сегодня: {stats['purchased_today']}
|
||
- За неделю: {stats['purchased_week']}
|
||
- За месяц: {stats['purchased_month']}
|
||
|
||
Выберите действие:
|
||
"""
|
||
|
||
keyboard = [
|
||
[
|
||
types.InlineKeyboardButton(text='📋 Список подписок', callback_data='admin_subs_list'),
|
||
types.InlineKeyboardButton(text='⏰ Истекающие', callback_data='admin_subs_expiring'),
|
||
],
|
||
[
|
||
types.InlineKeyboardButton(text='📊 Статистика', callback_data='admin_subs_stats'),
|
||
types.InlineKeyboardButton(text='🌍 География', callback_data='admin_subs_countries'),
|
||
],
|
||
[types.InlineKeyboardButton(text='⬅️ Назад', callback_data='admin_panel')],
|
||
]
|
||
|
||
await callback.message.edit_text(text, reply_markup=types.InlineKeyboardMarkup(inline_keyboard=keyboard))
|
||
await callback.answer()
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def show_subscriptions_list(callback: types.CallbackQuery, db_user: User, db: AsyncSession, page: int = 1):
|
||
subscriptions, total_count = await get_all_subscriptions(db, page=page, limit=10)
|
||
total_pages = (total_count + 9) // 10
|
||
|
||
if not subscriptions:
|
||
text = '📱 <b>Список подписок</b>\n\n❌ Подписки не найдены.'
|
||
else:
|
||
text = '📱 <b>Список подписок</b>\n\n'
|
||
text += f'📊 Всего: {total_count} | Страница: {page}/{total_pages}\n\n'
|
||
|
||
for i, sub in enumerate(subscriptions, 1 + (page - 1) * 10):
|
||
user_info = (
|
||
(f'ID{sub.user.telegram_id}' if sub.user.telegram_id else sub.user.email or f'#{sub.user.id}')
|
||
if sub.user
|
||
else 'Неизвестно'
|
||
)
|
||
sub_type = '🎁' if sub.is_trial else '💎'
|
||
status = '✅ Активна' if sub.is_active else '❌ Неактивна'
|
||
|
||
text += f'{i}. {sub_type} {user_info}\n'
|
||
text += f' {status} | До: {format_datetime(sub.end_date)}\n'
|
||
if sub.device_limit > 0:
|
||
text += f' 📱 Устройств: {sub.device_limit}\n'
|
||
text += '\n'
|
||
|
||
keyboard = []
|
||
|
||
if total_pages > 1:
|
||
nav_row = []
|
||
if page > 1:
|
||
nav_row.append(types.InlineKeyboardButton(text='⬅️', callback_data=f'admin_subs_list_page_{page - 1}'))
|
||
|
||
nav_row.append(types.InlineKeyboardButton(text=f'{page}/{total_pages}', callback_data='current_page'))
|
||
|
||
if page < total_pages:
|
||
nav_row.append(types.InlineKeyboardButton(text='➡️', callback_data=f'admin_subs_list_page_{page + 1}'))
|
||
|
||
keyboard.append(nav_row)
|
||
|
||
keyboard.extend(
|
||
[
|
||
[types.InlineKeyboardButton(text='🔄 Обновить', callback_data='admin_subs_list')],
|
||
[types.InlineKeyboardButton(text='⬅️ Назад', callback_data='admin_subscriptions')],
|
||
]
|
||
)
|
||
|
||
await callback.message.edit_text(text, reply_markup=types.InlineKeyboardMarkup(inline_keyboard=keyboard))
|
||
await callback.answer()
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def show_expiring_subscriptions(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||
expiring_3d = await get_expiring_subscriptions(db, 3)
|
||
expiring_1d = await get_expiring_subscriptions(db, 1)
|
||
expired = await get_expired_subscriptions(db)
|
||
|
||
text = f"""
|
||
⏰ <b>Истекающие подписки</b>
|
||
|
||
📊 <b>Статистика:</b>
|
||
- Истекают через 3 дня: {len(expiring_3d)}
|
||
- Истекают завтра: {len(expiring_1d)}
|
||
- Уже истекли: {len(expired)}
|
||
|
||
<b>Истекают через 3 дня:</b>
|
||
"""
|
||
|
||
for sub in expiring_3d[:5]:
|
||
user_info = (
|
||
(f'ID{sub.user.telegram_id}' if sub.user.telegram_id else sub.user.email or f'#{sub.user.id}')
|
||
if sub.user
|
||
else 'Неизвестно'
|
||
)
|
||
sub_type = '🎁' if sub.is_trial else '💎'
|
||
text += f'{sub_type} {user_info} - {format_datetime(sub.end_date)}\n'
|
||
|
||
if len(expiring_3d) > 5:
|
||
text += f'... и еще {len(expiring_3d) - 5}\n'
|
||
|
||
text += '\n<b>Истекают завтра:</b>\n'
|
||
for sub in expiring_1d[:5]:
|
||
user_info = (
|
||
(f'ID{sub.user.telegram_id}' if sub.user.telegram_id else sub.user.email or f'#{sub.user.id}')
|
||
if sub.user
|
||
else 'Неизвестно'
|
||
)
|
||
sub_type = '🎁' if sub.is_trial else '💎'
|
||
text += f'{sub_type} {user_info} - {format_datetime(sub.end_date)}\n'
|
||
|
||
if len(expiring_1d) > 5:
|
||
text += f'... и еще {len(expiring_1d) - 5}\n'
|
||
|
||
keyboard = [
|
||
[types.InlineKeyboardButton(text='📨 Отправить напоминания', callback_data='admin_send_expiry_reminders')],
|
||
[types.InlineKeyboardButton(text='🔄 Обновить', callback_data='admin_subs_expiring')],
|
||
[types.InlineKeyboardButton(text='⬅️ Назад', callback_data='admin_subscriptions')],
|
||
]
|
||
|
||
await callback.message.edit_text(text, reply_markup=types.InlineKeyboardMarkup(inline_keyboard=keyboard))
|
||
await callback.answer()
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def show_subscriptions_stats(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||
stats = await get_subscriptions_statistics(db)
|
||
|
||
expiring_3d = await get_expiring_subscriptions(db, 3)
|
||
expiring_7d = await get_expiring_subscriptions(db, 7)
|
||
expired = await get_expired_subscriptions(db)
|
||
|
||
text = f"""
|
||
📊 <b>Детальная статистика подписок</b>
|
||
|
||
<b>📱 Общая информация:</b>
|
||
• Всего подписок: {stats['total_subscriptions']}
|
||
• Активных: {stats['active_subscriptions']}
|
||
• Неактивных: {stats['total_subscriptions'] - stats['active_subscriptions']}
|
||
|
||
<b>💎 По типам:</b>
|
||
• Платных: {stats['paid_subscriptions']}
|
||
• Триальных: {stats['trial_subscriptions']}
|
||
|
||
<b>📈 Продажи:</b>
|
||
• Сегодня: {stats['purchased_today']}
|
||
• За неделю: {stats['purchased_week']}
|
||
• За месяц: {stats['purchased_month']}
|
||
|
||
<b>⏰ Истечение:</b>
|
||
• Истекают через 3 дня: {len(expiring_3d)}
|
||
• Истекают через 7 дней: {len(expiring_7d)}
|
||
• Уже истекли: {len(expired)}
|
||
|
||
<b>💰 Конверсия:</b>
|
||
• Из триала в платную: {stats.get('trial_to_paid_conversion', 0)}%
|
||
• Продлений: {stats.get('renewals_count', 0)}
|
||
"""
|
||
|
||
keyboard = [
|
||
# [
|
||
# types.InlineKeyboardButton(text="📊 Экспорт данных", callback_data="admin_subs_export"),
|
||
# types.InlineKeyboardButton(text="📈 Графики", callback_data="admin_subs_charts")
|
||
# ],
|
||
# [types.InlineKeyboardButton(text="🔄 Обновить", callback_data="admin_subs_stats")],
|
||
[types.InlineKeyboardButton(text='⬅️ Назад', callback_data='admin_subscriptions')]
|
||
]
|
||
|
||
await callback.message.edit_text(text, reply_markup=types.InlineKeyboardMarkup(inline_keyboard=keyboard))
|
||
await callback.answer()
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def show_countries_management(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||
try:
|
||
from app.services.remnawave_service import RemnaWaveService
|
||
|
||
remnawave_service = RemnaWaveService()
|
||
|
||
nodes_data = await remnawave_service.get_all_nodes()
|
||
squads_data = await remnawave_service.get_all_squads()
|
||
|
||
text = '🌍 <b>Управление странами</b>\n\n'
|
||
|
||
if nodes_data:
|
||
text += '<b>Доступные серверы:</b>\n'
|
||
countries = {}
|
||
|
||
for node in nodes_data:
|
||
country_code = node.get('country_code', 'XX')
|
||
country_name = country_code
|
||
|
||
if country_name not in countries:
|
||
countries[country_name] = []
|
||
countries[country_name].append(node)
|
||
|
||
for country, nodes in countries.items():
|
||
active_nodes = len([n for n in nodes if n.get('is_connected') and n.get('is_node_online')])
|
||
total_nodes = len(nodes)
|
||
|
||
country_flag = get_country_flag(country)
|
||
text += f'{country_flag} {country}: {active_nodes}/{total_nodes} серверов\n'
|
||
|
||
total_users_online = sum(n.get('users_online', 0) or 0 for n in nodes)
|
||
if total_users_online > 0:
|
||
text += f' 👥 Пользователей онлайн: {total_users_online}\n'
|
||
else:
|
||
text += '❌ Не удалось загрузить данные о серверах\n'
|
||
|
||
if squads_data:
|
||
text += f'\n<b>Всего сквадов:</b> {len(squads_data)}\n'
|
||
|
||
total_members = sum(squad.get('members_count', 0) for squad in squads_data)
|
||
text += f'<b>Участников в сквадах:</b> {total_members}\n'
|
||
|
||
text += '\n<b>Сквады:</b>\n'
|
||
for squad in squads_data[:5]:
|
||
name = squad.get('name', 'Неизвестно')
|
||
members = squad.get('members_count', 0)
|
||
inbounds = squad.get('inbounds_count', 0)
|
||
text += f'• {name}: {members} участников, {inbounds} inbound(s)\n'
|
||
|
||
if len(squads_data) > 5:
|
||
text += f'... и еще {len(squads_data) - 5} сквадов\n'
|
||
|
||
user_stats = await get_users_by_countries(db)
|
||
if user_stats:
|
||
text += '\n<b>Пользователи по регионам:</b>\n'
|
||
for country, count in user_stats.items():
|
||
country_flag = get_country_flag(country)
|
||
text += f'{country_flag} {country}: {count} пользователей\n'
|
||
|
||
except Exception as e:
|
||
logger.error(f'Ошибка получения данных о странах: {e}')
|
||
text = f"""
|
||
🌍 <b>Управление странами</b>
|
||
|
||
❌ <b>Ошибка загрузки данных</b>
|
||
Не удалось получить информацию о серверах.
|
||
|
||
Проверьте подключение к RemnaWave API.
|
||
|
||
<b>Детали ошибки:</b> {e!s}
|
||
"""
|
||
|
||
keyboard = [
|
||
[types.InlineKeyboardButton(text='🔄 Обновить', callback_data='admin_subs_countries')],
|
||
[
|
||
types.InlineKeyboardButton(text='📊 Статистика нод', callback_data='admin_rw_nodes'),
|
||
types.InlineKeyboardButton(text='🔧 Сквады', callback_data='admin_rw_squads'),
|
||
],
|
||
[types.InlineKeyboardButton(text='⬅️ Назад', callback_data='admin_subscriptions')],
|
||
]
|
||
|
||
await callback.message.edit_text(text, reply_markup=types.InlineKeyboardMarkup(inline_keyboard=keyboard))
|
||
await callback.answer()
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def send_expiry_reminders(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||
await callback.message.edit_text(
|
||
'📨 Отправка напоминаний...\n\nПодождите, это может занять время.', reply_markup=None
|
||
)
|
||
|
||
expiring_subs = await get_expiring_subscriptions(db, 1)
|
||
sent_count = 0
|
||
|
||
for subscription in expiring_subs:
|
||
if subscription.user:
|
||
try:
|
||
user = subscription.user
|
||
# Skip email-only users (no telegram_id)
|
||
if not user.telegram_id:
|
||
logger.debug(f'Пропуск email-пользователя {user.id} при отправке напоминания')
|
||
continue
|
||
|
||
days_left = max(1, subscription.days_left)
|
||
|
||
reminder_text = f"""
|
||
⚠️ <b>Подписка истекает!</b>
|
||
|
||
Ваша подписка истекает через {days_left} день(а).
|
||
|
||
Не забудьте продлить подписку, чтобы не потерять доступ к серверам.
|
||
|
||
💎 Продлить подписку можно в главном меню.
|
||
"""
|
||
|
||
await callback.bot.send_message(chat_id=user.telegram_id, text=reminder_text)
|
||
sent_count += 1
|
||
|
||
except Exception as e:
|
||
logger.error(f'Ошибка отправки напоминания пользователю {subscription.user_id}: {e}')
|
||
|
||
await callback.message.edit_text(
|
||
f'✅ Напоминания отправлены: {sent_count} из {len(expiring_subs)}',
|
||
reply_markup=types.InlineKeyboardMarkup(
|
||
inline_keyboard=[[types.InlineKeyboardButton(text='⬅️ Назад', callback_data='admin_subs_expiring')]]
|
||
),
|
||
)
|
||
await callback.answer()
|
||
|
||
|
||
@admin_required
|
||
@error_handler
|
||
async def handle_subscriptions_pagination(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||
page = int(callback.data.split('_')[-1])
|
||
await show_subscriptions_list(callback, db_user, db, page)
|
||
|
||
|
||
def register_handlers(dp: Dispatcher):
|
||
dp.callback_query.register(show_subscriptions_menu, F.data == 'admin_subscriptions')
|
||
dp.callback_query.register(show_subscriptions_list, F.data == 'admin_subs_list')
|
||
dp.callback_query.register(show_expiring_subscriptions, F.data == 'admin_subs_expiring')
|
||
dp.callback_query.register(show_subscriptions_stats, F.data == 'admin_subs_stats')
|
||
dp.callback_query.register(show_countries_management, F.data == 'admin_subs_countries')
|
||
dp.callback_query.register(send_expiry_reminders, F.data == 'admin_send_expiry_reminders')
|
||
|
||
dp.callback_query.register(handle_subscriptions_pagination, F.data.startswith('admin_subs_list_page_'))
|