dinlo
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
"""Обработчик команды /start и главного меню."""
|
||
|
||
from aiogram import Router
|
||
from aiogram.filters import CommandStart, Command
|
||
from aiogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
|
||
|
||
from config import settings
|
||
|
||
router = Router()
|
||
|
||
|
||
def get_main_keyboard(user_id: int = None) -> InlineKeyboardMarkup:
|
||
"""Создать главное меню."""
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
||
[
|
||
InlineKeyboardButton(text="🎨 Генерация (txt2img)", callback_data="gen_txt2img"),
|
||
],
|
||
[
|
||
InlineKeyboardButton(text="⚙️ Профили", callback_data="profiles_menu"),
|
||
InlineKeyboardButton(text="🗑️ Настройки хранения", callback_data="storage_settings"),
|
||
],
|
||
[
|
||
InlineKeyboardButton(text="📊 Статус SD API", callback_data="check_sd_status"),
|
||
InlineKeyboardButton(text="ℹ️ Помощь", callback_data="help"),
|
||
],
|
||
])
|
||
|
||
# Добавляем кнопку админ-панели для админа
|
||
if user_id and user_id == settings.ADMIN_ID:
|
||
keyboard.inline_keyboard.insert(0, [
|
||
InlineKeyboardButton(text="🛡️ Админ-панель", callback_data="admin_panel"),
|
||
])
|
||
|
||
return keyboard
|
||
|
||
|
||
@router.message(CommandStart())
|
||
async def cmd_start(message: Message):
|
||
"""Обработка команды /start."""
|
||
text = (
|
||
f"👋 <b>Привет, {message.from_user.first_name}!</b>\n\n"
|
||
"Я бот для генерации изображений через Stable Diffusion.\n"
|
||
"Выберите действие из меню ниже:"
|
||
)
|
||
await message.answer(text, reply_markup=get_main_keyboard(message.from_user.id), parse_mode="HTML")
|
||
|
||
|
||
@router.message(Command("menu"))
|
||
async def cmd_menu(message: Message):
|
||
"""Обработка команды /menu."""
|
||
await message.answer("📋 Главное меню:", reply_markup=get_main_keyboard(message.from_user.id))
|