Vladless
78 lines
3.7 KiB
Python
78 lines
3.7 KiB
Python
from aiogram import Router, types
|
|
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup, CallbackQuery
|
|
from aiogram.filters import Command
|
|
from aiogram.fsm.state import StatesGroup, State
|
|
from config import ADMIN_ID, DATABASE_URL
|
|
import asyncpg
|
|
from datetime import datetime
|
|
from bot import bot
|
|
|
|
router = Router()
|
|
|
|
class UserEditorState(StatesGroup):
|
|
waiting_for_tg_id = State()
|
|
displaying_user_info = State()
|
|
|
|
@router.message(Command('admin'))
|
|
async def handle_admin_command(message: types.Message):
|
|
if message.from_user.id != ADMIN_ID:
|
|
await message.reply("У вас нет доступа к этой команде.")
|
|
return
|
|
|
|
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
|
[InlineKeyboardButton(text="Статистика пользователей", callback_data="user_stats")],
|
|
[InlineKeyboardButton(text="Редактор пользователей", callback_data="user_editor")]
|
|
])
|
|
await message.reply("Панель администратора", reply_markup=keyboard)
|
|
|
|
@router.callback_query(lambda c: c.data == "user_stats")
|
|
async def user_stats_menu(callback_query: CallbackQuery):
|
|
conn = await asyncpg.connect(DATABASE_URL)
|
|
try:
|
|
total_users = await conn.fetchval("SELECT COUNT(*) FROM connections")
|
|
total_keys = await conn.fetchval("SELECT COUNT(*) FROM keys")
|
|
total_referrals = await conn.fetchval("SELECT COUNT(*) FROM referrals")
|
|
|
|
active_keys = await conn.fetchval("SELECT COUNT(*) FROM keys WHERE expiry_time > $1", int(datetime.utcnow().timestamp() * 1000))
|
|
expired_keys = total_keys - active_keys
|
|
|
|
stats_message = (
|
|
f"🔹 <b>Общая статистика пользователей:</b>\n"
|
|
f"• Всего пользователей: <b>{total_users}</b>\n"
|
|
f"• Всего ключей: <b>{total_keys}</b>\n"
|
|
f"• Всего рефералов: <b>{total_referrals}</b>\n"
|
|
f"• Активные ключи: <b>{active_keys}</b>\n"
|
|
f"• Истекшие ключи: <b>{expired_keys}</b>"
|
|
)
|
|
|
|
back_button = InlineKeyboardButton(text="Назад", callback_data="back_to_admin_menu")
|
|
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
|
[back_button]
|
|
])
|
|
|
|
await callback_query.message.edit_text(stats_message, reply_markup=keyboard, parse_mode="HTML")
|
|
finally:
|
|
await conn.close()
|
|
|
|
await callback_query.answer()
|
|
|
|
@router.callback_query(lambda c: c.data == "user_editor")
|
|
async def user_editor_menu(callback_query: CallbackQuery):
|
|
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
|
[InlineKeyboardButton(text="Поиск по имени ключа", callback_data="search_by_key_name")],
|
|
[InlineKeyboardButton(text="Поиск по tg_id", callback_data="search_by_tg_id")],
|
|
[InlineKeyboardButton(text="Назад", callback_data="back_to_admin_menu")] # Back button
|
|
])
|
|
await callback_query.message.edit_text("Выберите метод поиска:", reply_markup=keyboard)
|
|
|
|
@router.callback_query(lambda c: c.data == "back_to_admin_menu")
|
|
async def back_to_admin_menu(callback_query: CallbackQuery):
|
|
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
|
[InlineKeyboardButton(text="Статистика пользователей", callback_data="user_stats")],
|
|
[InlineKeyboardButton(text="Редактор пользователей", callback_data="user_editor")]
|
|
])
|
|
await callback_query.message.edit_text("Панель администратора", reply_markup=keyboard)
|
|
|
|
async def handle_error(tg_id, callback_query, message):
|
|
await bot.edit_message_text(message, chat_id=tg_id, message_id=callback_query.message.message_id)
|