103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
|
from aiogram import Router, F
|
|||
|
|
from aiogram.filters import Command
|
|||
|
|
from aiogram.types import Message, CallbackQuery, InlineKeyboardMarkup, InlineKeyboardButton
|
|||
|
|
|
|||
|
|
from config.settings import settings
|
|||
|
|
from database.db import async_session_maker
|
|||
|
|
from database.models import AIProvider
|
|||
|
|
from services.ai_model_service import AIModelService
|
|||
|
|
|
|||
|
|
router = Router()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_admin_ids() -> set:
|
|||
|
|
"""Получить множество ID админов"""
|
|||
|
|
if not settings.admin_ids:
|
|||
|
|
return set()
|
|||
|
|
return set(int(x.strip()) for x in settings.admin_ids.split(",") if x.strip())
|
|||
|
|
|
|||
|
|
|
|||
|
|
def is_admin(user_id: int) -> bool:
|
|||
|
|
"""Проверить, является ли пользователь админом"""
|
|||
|
|
return user_id in get_admin_ids()
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_model_keyboard() -> InlineKeyboardMarkup:
|
|||
|
|
"""Создать клавиатуру выбора AI модели"""
|
|||
|
|
async with async_session_maker() as session:
|
|||
|
|
models = await AIModelService.get_all_models(session)
|
|||
|
|
|
|||
|
|
keyboard = []
|
|||
|
|
for model in models:
|
|||
|
|
marker = "✓ " if model.is_active else ""
|
|||
|
|
provider_emoji = "🟢" if model.provider == AIProvider.openai else "🔵"
|
|||
|
|
keyboard.append([InlineKeyboardButton(
|
|||
|
|
text=f"{marker}{provider_emoji} {model.display_name}",
|
|||
|
|
callback_data=f"admin_model_{model.id}"
|
|||
|
|
)])
|
|||
|
|
|
|||
|
|
keyboard.append([InlineKeyboardButton(text="❌ Закрыть", callback_data="admin_close")])
|
|||
|
|
return InlineKeyboardMarkup(inline_keyboard=keyboard)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.message(Command("admin"))
|
|||
|
|
async def cmd_admin(message: Message):
|
|||
|
|
"""Админская панель"""
|
|||
|
|
if not is_admin(message.from_user.id):
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# Убеждаемся что дефолтные модели созданы
|
|||
|
|
async with async_session_maker() as session:
|
|||
|
|
await AIModelService.ensure_default_models(session)
|
|||
|
|
active_model = await AIModelService.get_active_model(session)
|
|||
|
|
|
|||
|
|
active_name = active_model.display_name if active_model else "Не выбрана"
|
|||
|
|
|
|||
|
|
text = (
|
|||
|
|
"🔧 <b>Админ-панель</b>\n\n"
|
|||
|
|
f"🤖 Текущая AI модель: <b>{active_name}</b>\n\n"
|
|||
|
|
"Выберите модель для генерации:"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
keyboard = await get_model_keyboard()
|
|||
|
|
await message.answer(text, reply_markup=keyboard)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.callback_query(F.data.startswith("admin_model_"))
|
|||
|
|
async def set_admin_model(callback: CallbackQuery):
|
|||
|
|
"""Установить AI модель"""
|
|||
|
|
if not is_admin(callback.from_user.id):
|
|||
|
|
await callback.answer("⛔ Доступ запрещен", show_alert=True)
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
model_id = int(callback.data.split("_")[-1])
|
|||
|
|
|
|||
|
|
async with async_session_maker() as session:
|
|||
|
|
success = await AIModelService.set_active_model(session, model_id)
|
|||
|
|
|
|||
|
|
if success:
|
|||
|
|
active_model = await AIModelService.get_active_model(session)
|
|||
|
|
await callback.answer(f"✅ Модель изменена: {active_model.display_name}")
|
|||
|
|
|
|||
|
|
text = (
|
|||
|
|
"🔧 <b>Админ-панель</b>\n\n"
|
|||
|
|
f"🤖 Текущая AI модель: <b>{active_model.display_name}</b>\n\n"
|
|||
|
|
"Выберите модель для генерации:"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
await callback.answer("❌ Ошибка при смене модели", show_alert=True)
|
|||
|
|
text = "🔧 <b>Админ-панель</b>\n\n❌ Ошибка при смене модели"
|
|||
|
|
|
|||
|
|
keyboard = await get_model_keyboard()
|
|||
|
|
await callback.message.edit_text(text, reply_markup=keyboard)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.callback_query(F.data == "admin_close")
|
|||
|
|
async def admin_close(callback: CallbackQuery):
|
|||
|
|
"""Закрыть админ-панель"""
|
|||
|
|
if not is_admin(callback.from_user.id):
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
await callback.message.delete()
|
|||
|
|
await callback.answer()
|