258 lines
12 KiB
Python
258 lines
12 KiB
Python
import logging
|
|
from aiogram import Router, F, types
|
|
from typing import Optional, Dict, List
|
|
from datetime import datetime
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from config.settings import Settings
|
|
|
|
from db.dal import user_dal, payment_dal, panel_sync_dal
|
|
from db.models import Payment, PanelSyncStatus
|
|
from bot.services.panel_api_service import PanelApiService
|
|
|
|
from bot.keyboards.inline.admin_keyboards import get_back_to_admin_panel_keyboard
|
|
from bot.middlewares.i18n import JsonI18n
|
|
|
|
router = Router(name="admin_statistics_router")
|
|
|
|
|
|
async def show_statistics_handler(callback: types.CallbackQuery,
|
|
i18n_data: dict, settings: Settings,
|
|
session: AsyncSession):
|
|
current_lang = i18n_data.get("current_language", settings.DEFAULT_LANGUAGE)
|
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
|
if not i18n or not callback.message:
|
|
await callback.answer("Error displaying statistics.", show_alert=True)
|
|
return
|
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs)
|
|
|
|
await callback.answer()
|
|
|
|
stats_text_parts = [f"<b>{_('admin_stats_header')}</b>"]
|
|
|
|
# Enhanced user statistics
|
|
user_stats = await user_dal.get_enhanced_user_statistics(session)
|
|
|
|
stats_text_parts.append(
|
|
f"\n<b>👥 {_('admin_enhanced_users_stats_header')}</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"📊 {_('admin_user_stats_total_label')}: <b>{user_stats['total_users']}</b>"
|
|
)
|
|
# Removed: Active today moved to panel stats
|
|
stats_text_parts.append(
|
|
f"💳 {_('admin_user_stats_paid_subs_label')}: <b>{user_stats['paid_subscriptions']}</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"🆓 {_('admin_user_stats_trial_label')}: <b>{user_stats['trial_users']}</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"😴 {_('admin_user_stats_inactive_label')}: <b>{user_stats['inactive_users']}</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"🚫 {_('admin_user_stats_banned_label')}: <b>{user_stats['banned_users']}</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"🎁 {_('admin_user_stats_referral_label')}: <b>{user_stats['referral_users']}</b>"
|
|
)
|
|
|
|
# Panel Statistics - moved above financial
|
|
stats_text_parts.append(f"\n<b>🖥 {_('admin_panel_stats_header')}</b>")
|
|
|
|
try:
|
|
async with PanelApiService(settings) as panel_service:
|
|
# Get system stats
|
|
system_stats = await panel_service.get_system_stats()
|
|
bandwidth_stats = await panel_service.get_bandwidth_stats()
|
|
nodes_stats = await panel_service.get_nodes_statistics()
|
|
|
|
logging.info(f"Panel stats response: system={system_stats}, bandwidth={bandwidth_stats}, nodes={nodes_stats}")
|
|
|
|
if system_stats:
|
|
users = system_stats.get('users', {})
|
|
status_counts = users.get('statusCounts', {})
|
|
online_stats = system_stats.get('onlineStats', {})
|
|
|
|
active_users = status_counts.get('ACTIVE', 0)
|
|
disabled_users = status_counts.get('DISABLED', 0)
|
|
expired_users = status_counts.get('EXPIRED', 0)
|
|
limited_users = status_counts.get('LIMITED', 0)
|
|
total_users = users.get('totalUsers', 0)
|
|
online_now = online_stats.get('onlineNow', 0)
|
|
|
|
stats_text_parts.append(f"🟢 {_('admin_panel_online_label')}: <b>{online_now}</b>")
|
|
stats_text_parts.append(f"📊 {_('admin_panel_active_label')}: <b>{active_users}</b>")
|
|
stats_text_parts.append(f"🔴 {_('admin_panel_disabled_label')}: <b>{disabled_users}</b>")
|
|
stats_text_parts.append(f"⏰ {_('admin_panel_expired_label')}: <b>{expired_users}</b>")
|
|
stats_text_parts.append(f"⚠️ {_('admin_panel_limited_label')}: <b>{limited_users}</b>")
|
|
stats_text_parts.append(f"👥 {_('admin_panel_total_users_label')}: <b>{total_users}</b>")
|
|
|
|
# System resources
|
|
memory = system_stats.get('memory', {})
|
|
if memory:
|
|
memory_total = memory.get('total', 1)
|
|
memory_used = memory.get('used', 0)
|
|
memory_usage = (memory_used / memory_total) * 100 if memory_total > 0 else 0
|
|
stats_text_parts.append(f"💾 {_('admin_panel_memory_usage_label')}: <b>{memory_usage:.1f}%</b>")
|
|
else:
|
|
stats_text_parts.append(f"⚠️ {_('admin_panel_system_stats_error')}")
|
|
|
|
# Bandwidth stats
|
|
if bandwidth_stats:
|
|
week_traffic = bandwidth_stats.get('bandwidthLastSevenDays', {})
|
|
month_traffic = bandwidth_stats.get('bandwidthLast30Days', {})
|
|
# Fallback to the actual key name from API if the above doesn't exist
|
|
if not month_traffic:
|
|
month_traffic = bandwidth_stats.get('bandwidthLastThirtyDays', {})
|
|
|
|
if week_traffic:
|
|
week_total = week_traffic.get('current', '0 B')
|
|
stats_text_parts.append(f"📊 {_('admin_panel_traffic_week_label')}: <b>{week_total}</b>")
|
|
|
|
if month_traffic:
|
|
month_total = month_traffic.get('current', '0 B')
|
|
stats_text_parts.append(f"📊 {_('admin_panel_traffic_month_label')}: <b>{month_total}</b>")
|
|
else:
|
|
stats_text_parts.append(f"⚠️ {_('admin_panel_bandwidth_stats_error')}")
|
|
|
|
# Nodes stats
|
|
if nodes_stats and 'lastSevenDays' in nodes_stats:
|
|
last_seven_days = nodes_stats.get('lastSevenDays', [])
|
|
# Get unique node names from the data
|
|
unique_nodes = set()
|
|
for node_data in last_seven_days:
|
|
unique_nodes.add(node_data.get('nodeName', ''))
|
|
total_nodes_count = len(unique_nodes)
|
|
# Assume all nodes are active since we don't have status info
|
|
stats_text_parts.append(f"🔗 {_('admin_panel_nodes_label')}: <b>{total_nodes_count}/{total_nodes_count}</b>")
|
|
else:
|
|
# Use nodes total from system stats as fallback
|
|
nodes_info = system_stats.get('nodes', {}) if system_stats else {}
|
|
total_online = nodes_info.get('totalOnline', 0)
|
|
stats_text_parts.append(f"🔗 {_('admin_panel_nodes_label')}: <b>{total_online}</b>")
|
|
|
|
except Exception as e:
|
|
logging.error(f"Failed to fetch panel statistics: {e}", exc_info=True)
|
|
stats_text_parts.append(f"❌ {_('admin_panel_stats_fetch_error')}")
|
|
stats_text_parts.append(f"⚠️ {_('admin_panel_stats_error_details')}: {str(e)}")
|
|
|
|
# Financial statistics
|
|
financial_stats = await payment_dal.get_financial_statistics(session)
|
|
|
|
stats_text_parts.append(
|
|
f"\n<b>💰 {_('admin_financial_stats_header')}</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"📅 {_('admin_financial_today_label')}: <b>{financial_stats['today_revenue']:.2f} RUB</b> ({financial_stats['today_payments_count']} {_('admin_financial_payments_label')})"
|
|
)
|
|
stats_text_parts.append(
|
|
f"📅 {_('admin_financial_week_label')}: <b>{financial_stats['week_revenue']:.2f} RUB</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"📅 {_('admin_financial_month_label')}: <b>{financial_stats['month_revenue']:.2f} RUB</b>"
|
|
)
|
|
stats_text_parts.append(
|
|
f"🏆 {_('admin_financial_all_time_label')}: <b>{financial_stats['all_time_revenue']:.2f} RUB</b>"
|
|
)
|
|
|
|
last_payments_models: List[
|
|
Payment] = await payment_dal.get_recent_payment_logs_with_user(session,
|
|
limit=5)
|
|
if last_payments_models:
|
|
stats_text_parts.append(
|
|
f"\n<b>{_('admin_stats_recent_payments_header')}</b>")
|
|
for payment in last_payments_models:
|
|
pending_statuses = [
|
|
"pending",
|
|
"pending_yookassa",
|
|
"pending_freekassa",
|
|
"pending_platega",
|
|
"pending_severpay",
|
|
"pending_cryptopay",
|
|
]
|
|
status_emoji = (
|
|
"✅"
|
|
if payment.status == "succeeded"
|
|
else "⏳"
|
|
if payment.status in pending_statuses
|
|
else "❌"
|
|
)
|
|
|
|
user_info = f"User {payment.user_id}"
|
|
if payment.user and payment.user.username:
|
|
user_info += f" (@{payment.user.username})"
|
|
elif payment.user and payment.user.first_name:
|
|
user_info += f" ({payment.user.first_name})"
|
|
|
|
payment_date_str = payment.created_at.strftime(
|
|
'%Y-%m-%d') if payment.created_at else "N/A"
|
|
|
|
stats_text_parts.append(
|
|
_("admin_stats_payment_item",
|
|
status_emoji=status_emoji,
|
|
amount=payment.amount,
|
|
currency=payment.currency,
|
|
user_info=user_info,
|
|
p_status=payment.status,
|
|
p_date=payment_date_str))
|
|
else:
|
|
stats_text_parts.append(f"\n{_('admin_stats_no_payments_found')}")
|
|
|
|
sync_status_model: Optional[
|
|
PanelSyncStatus] = await panel_sync_dal.get_panel_sync_status(session)
|
|
if sync_status_model and sync_status_model.status != "never_run":
|
|
stats_text_parts.append(
|
|
f"\n<b>{_('admin_stats_last_sync_header')}</b>")
|
|
|
|
sync_time_val = sync_status_model.last_sync_time
|
|
sync_time_str = sync_time_val.strftime(
|
|
'%Y-%m-%d %H:%M:%S UTC') if sync_time_val else "N/A"
|
|
|
|
details_val = sync_status_model.details
|
|
details_str = details_val or "N/A"
|
|
|
|
stats_text_parts.append(
|
|
f" {_('admin_stats_sync_time')}: {sync_time_str}")
|
|
stats_text_parts.append(
|
|
f" {_('admin_stats_sync_status')}: {sync_status_model.status}")
|
|
stats_text_parts.append(
|
|
f" {_('admin_stats_sync_users_processed')}: {sync_status_model.users_processed_from_panel}"
|
|
)
|
|
stats_text_parts.append(
|
|
f" {_('admin_stats_sync_subs_synced')}: {sync_status_model.subscriptions_synced}"
|
|
)
|
|
stats_text_parts.append(
|
|
f" {_('admin_stats_sync_details_label')}: {details_str}")
|
|
else:
|
|
stats_text_parts.append(f"\n{_('admin_sync_status_never_run')}")
|
|
|
|
final_text = "\n".join(stats_text_parts)
|
|
|
|
try:
|
|
await callback.message.edit_text(
|
|
final_text,
|
|
reply_markup=get_back_to_admin_panel_keyboard(current_lang, i18n),
|
|
parse_mode="HTML")
|
|
except Exception as e_edit:
|
|
logging.error(f"Error editing message for statistics: {e_edit}",
|
|
exc_info=True)
|
|
|
|
max_chunk_size = 4000
|
|
for i in range(0, len(final_text), max_chunk_size):
|
|
chunk = final_text[i:i + max_chunk_size]
|
|
is_last_chunk = (i + max_chunk_size) >= len(final_text)
|
|
try:
|
|
await callback.message.answer(
|
|
chunk,
|
|
reply_markup=get_back_to_admin_panel_keyboard(
|
|
current_lang, i18n) if is_last_chunk else None,
|
|
parse_mode="HTML")
|
|
except Exception as e_chunk:
|
|
logging.error(f"Failed to send statistics chunk: {e_chunk}")
|
|
if i == 0:
|
|
await callback.message.answer(
|
|
_("error_displaying_statistics"),
|
|
reply_markup=get_back_to_admin_panel_keyboard(
|
|
current_lang, i18n))
|
|
break
|