Vladless
108 lines
5.4 KiB
Python
108 lines
5.4 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
|
|
import subprocess
|
|
from backup import backup_database
|
|
from handlers.commands import send_message_to_all_clients
|
|
from aiogram.types import Message
|
|
from aiogram.fsm.context import FSMContext
|
|
|
|
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")],
|
|
[InlineKeyboardButton(text="Отправить сообщение всем клиентам", callback_data="send_to_alls")],
|
|
[InlineKeyboardButton(text="Создать бэкап", callback_data="backups")],
|
|
[InlineKeyboardButton(text="Перезапустить бота", callback_data="restart_bot")]
|
|
])
|
|
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 == "send_to_alls")
|
|
async def handle_send_to_all(callback_query: CallbackQuery, state: FSMContext):
|
|
await send_message_to_all_clients(callback_query.message, state, from_panel=True)
|
|
await callback_query.answer()
|
|
|
|
@router.callback_query(lambda c: c.data == "backups")
|
|
async def handle_backup(message: Message):
|
|
await message.answer("Запускаю бэкап базы данных...")
|
|
await backup_database()
|
|
await message.answer("Бэкап завершен и отправлен админу.")
|
|
|
|
@router.callback_query(lambda c: c.data == "restart_bot")
|
|
async def handle_restart(callback_query: CallbackQuery):
|
|
if callback_query.from_user.id == ADMIN_ID:
|
|
try:
|
|
result = subprocess.run(['sudo', 'systemctl', 'restart', 'bot.service'], check=True, capture_output=True, text=True)
|
|
await callback_query.message.answer("Бот успешно перезапущен.")
|
|
except subprocess.CalledProcessError as e:
|
|
await callback_query.message.answer(f"Бот будет перезапущен через 30 секунд {e.stderr}")
|
|
else:
|
|
await callback_query.answer("У вас нет доступа к этой команде.", show_alert=True)
|
|
|
|
@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)
|