Initial commit

src/trudex/application/bot/creator_dialogs/main_menu.py

@@ -1,182 +1,21 @@
-from aiogram.types import CallbackQuery, Message
-from aiogram_dialog import Dialog, DialogManager, Window
-from aiogram_dialog.widgets.input import MessageInput
-from aiogram_dialog.widgets.kbd import Back, Button, Cancel, Column, Row, ScrollingGroup, Select, SwitchTo
-from aiogram_dialog.widgets.text import Const, Format
-from dishka import FromDishka
-from dishka.integrations.aiogram import CONTAINER_NAME
-from dishka.integrations.aiogram_dialog import inject
+from aiogram.types import CallbackQuery
+from aiogram_dialog import Dialog, DialogManager, Window, StartMode
+from aiogram_dialog.widgets.kbd import Button, Column
+from aiogram_dialog.widgets.text import Const
 
-from trudex.application.bot.creator_dialogs.states import CreatorMenuSG
-from trudex.infrastructure.database.dao.user import UserDAO
-from trudex.infrastructure.utils.broadcast import broadcast_message
+from trudex.application.bot.creator_dialogs.states import CreatorMenuSG, CreatorUsersSG, CreatorTestsSG, CreatorBroadcastSG
 
 
-@inject
-async def get_users_data(user_dao: FromDishka[UserDAO], **_kwargs):
-    users = await user_dao.get_all()
-    
-    return {
-        "users": [
-            (f"{u.first_name} (@{u.username or 'нет'})", u.id)
-            for u in users
-        ],
-        "count": len(users),
-    }
-
-
-@inject
-async def get_user_detail_data(dialog_manager: DialogManager, user_dao: FromDishka[UserDAO], **_kwargs):
-    user_id = dialog_manager.dialog_data.get("selected_user_id")
-    if not user_id:
-        return {"user_info": "Пользователь не выбран", "is_admin": True, "show_make_admin": False}
-    
-    user = await user_dao.get_by_id(user_id)
-    if not user:
-        return {"user_info": "Пользователь не найден", "is_admin": True, "show_make_admin": False}
-    
-    username_str = f"@{user.username}" if user.username else "—"
-    last_name_str = user.last_name or "—"
-    group_str = str(user.group) if user.group else "—"
-    admin_status = "✅ Да" if user.is_admin else "❌ Нет"
-    
-    user_info = (
-        f"<b>👤 Информация о пользователе</b>\n\n"
-        f"<b>ID:</b> <code>{user.id}</code>\n"
-        f"<b>Имя:</b> {user.first_name}\n"
-        f"<b>Фамилия:</b> {last_name_str}\n"
-        f"<b>Username:</b> {username_str}\n"
-        f"<b>Группа:</b> {group_str}\n"
-        f"<b>Администратор:</b> {admin_status}"
-    )
-    
-    return {
-        "user_info": user_info,
-        "is_admin": user.is_admin,
-        "show_make_admin": not user.is_admin,
-    }
-
-
-@inject
-async def get_confirm_data(dialog_manager: DialogManager, user_dao: FromDishka[UserDAO], **_kwargs):
-    user_id = dialog_manager.dialog_data.get("selected_user_id")
-    if not user_id:
-        return {"user_info": "Пользователь не выбран"}
-    
-    user = await user_dao.get_by_id(user_id)
-    if not user:
-        return {"user_info": "Пользователь не найден"}
-    
-    username_str = f"@{user.username}" if user.username else "—"
-    return {
-        "user_info": f"<b>{user.first_name}</b>\n{username_str}\nID: <code>{user.id}</code>"
-    }
-
-
-async def on_user_selected(_callback: CallbackQuery, _widget: Select, manager: DialogManager, item_id: str):
-    manager.dialog_data["selected_user_id"] = int(item_id)
-    await manager.switch_to(CreatorMenuSG.user_detail)
-
-
-async def on_input_mode(_callback: CallbackQuery, _button: Button, manager: DialogManager):
-    await manager.switch_to(CreatorMenuSG.users_input)
-
-
-async def on_user_input(message: Message, _widget: MessageInput, manager: DialogManager):
-    container = manager.middleware_data[CONTAINER_NAME]
-    user_dao = await container.get(UserDAO)
-    
-    text = (message.text or "").strip()
-    
-    user = None
-    if text.startswith("@"):
-        username = text[1:]
-        all_users = await user_dao.get_all()
-        user = next((u for u in all_users if u.username == username), None)
-    elif text.isdigit():
-        user = await user_dao.get_by_id(int(text))
-    
-    if not user:
-        await message.answer("❌ Пользователь не найден в базе данных.")
-        return
-    
-    manager.dialog_data["selected_user_id"] = user.id
-    await manager.switch_to(CreatorMenuSG.user_detail)
-
-
-async def on_make_admin_clicked(_callback: CallbackQuery, _button: Button, manager: DialogManager):
-    await manager.switch_to(CreatorMenuSG.make_admin_confirm)
-
-
-async def on_confirm_yes(_callback: CallbackQuery, _button: Button, manager: DialogManager):
-    container = manager.middleware_data[CONTAINER_NAME]
-    user_dao = await container.get(UserDAO)
-    
-    user_id = manager.dialog_data.get("selected_user_id")
-    if not user_id:
-        await _callback.answer("Ошибка: пользователь не выбран")
-        return
-    
-    await user_dao.update(user_id=user_id, is_admin=True)
-    await _callback.answer("✅ Пользователь назначен администратором")
-    await manager.switch_to(CreatorMenuSG.user_detail)
-
-
-async def on_confirm_no(_callback: CallbackQuery, _button: Button, manager: DialogManager):
-    await _callback.answer("Отменено")
-    await manager.switch_to(CreatorMenuSG.user_detail)
-
-
-async def on_broadcast_input(message: Message, _widget: MessageInput, manager: DialogManager):
-    manager.dialog_data["broadcast_message_id"] = message.message_id
-    manager.dialog_data["broadcast_chat_id"] = message.chat.id
-    await manager.switch_to(CreatorMenuSG.broadcast_confirm)
-
-
-@inject
-async def on_broadcast_confirm(_callback: CallbackQuery, _button: Button, manager: DialogManager, user_dao: FromDishka[UserDAO]):
-    message_id = manager.dialog_data.get("broadcast_message_id")
-    chat_id = manager.dialog_data.get("broadcast_chat_id")
-    
-    if not message_id or not chat_id or not _callback.message:
-        await _callback.answer("Ошибка: сообщение не найдено")
-        return
-    
-    await _callback.message.answer("⏳ Рассылка началась...")
-    
-    bot = _callback.bot
-    if not bot:
-        await _callback.answer("Ошибка: бот не найден")
-        return
-    
-    stats = await broadcast_message(bot, message_id, chat_id, user_dao)
-    
-    stats_text = (
-        f"✅ <b>Рассылка завершена</b>\n\n"
-        f"Всего пользователей: {stats.total}\n"
-        f"Успешно отправлено: {stats.success}\n"
-        f"Не удалось отправить: {stats.failed}"
-    )
-    
-    await _callback.message.answer(stats_text)
-    await manager.done()
-
-
-async def on_broadcast_cancel(_callback: CallbackQuery, _button: Button, manager: DialogManager):
-    await _callback.answer("Рассылка отменена")
-    await manager.switch_to(CreatorMenuSG.main)
-
-
-async def on_tests_clicked(_callback: CallbackQuery, _button: Button, _manager: DialogManager) -> None:
-    await _callback.answer("Тесты")
+async def on_tests_clicked(_callback: CallbackQuery, _button: Button, manager: DialogManager) -> None:
+    await manager.start(CreatorTestsSG.tests_list, mode=StartMode.RESET_STACK)
 
 
 async def on_users_clicked(_callback: CallbackQuery, _button: Button, manager: DialogManager) -> None:
-    await manager.switch_to(CreatorMenuSG.users_list)
+    await manager.start(CreatorUsersSG.users_list, mode=StartMode.RESET_STACK)
 
 
 async def on_broadcast_clicked(_callback: CallbackQuery, _button: Button, manager: DialogManager) -> None:
-    await manager.switch_to(CreatorMenuSG.broadcast_input)
+    await manager.start(CreatorBroadcastSG.broadcast_input, mode=StartMode.RESET_STACK)
 
 
 creator_menu_dialog = Dialog(
@@ -189,64 +28,4 @@ creator_menu_dialog = Dialog(
         ),
         state=CreatorMenuSG.main,
     ),
-    Window(
-        Format("<b>👥 Пользователи</b>\n\nВсего: {count}"),
-        ScrollingGroup(
-            Select(
-                Format("{item[0]}"),
-                id="user_select",
-                item_id_getter=lambda x: x[1],
-                items="users",
-                on_click=on_user_selected,
-            ),
-            id="users_scroll",
-            width=1,
-            height=7,
-        ),
-        Column(
-            Button(Const("✏️ Ввести ID/Username"), id="input_mode", on_click=on_input_mode),
-            Cancel(Const("◀️ Назад")),
-        ),
-        state=CreatorMenuSG.users_list,
-        getter=get_users_data,
-    ),
-    Window(
-        Const("<b>Введите ID или @username пользователя:</b>"),
-        MessageInput(on_user_input),
-        SwitchTo(Const("◀️ Назад"), id="back_to_list", state=CreatorMenuSG.users_list),
-        state=CreatorMenuSG.users_input,
-    ),
-    Window(
-        Format("{user_info}"),
-        Column(
-            Button(Const("👑 Сделать администратором"), id="make_admin", on_click=on_make_admin_clicked, when="show_make_admin"),
-            SwitchTo(Const("◀️ Назад"), id="back_to_list", state=CreatorMenuSG.users_list),
-        ),
-        state=CreatorMenuSG.user_detail,
-        getter=get_user_detail_data,
-    ),
-    Window(
-        Const("<b>⚠️ Подтверждение</b>\n\nВы уверены, что хотите назначить этого пользователя администратором?\n"),
-        Format("{user_info}"),
-        Row(
-            Button(Const("✅ Да"), id="confirm_yes", on_click=on_confirm_yes),
-            Button(Const("❌ Нет"), id="confirm_no", on_click=on_confirm_no),
-        ),
-        state=CreatorMenuSG.make_admin_confirm,
-        getter=get_confirm_data,
-    ),
-    Window(
-        Const("<b>📢 Рассылка</b>\n\nОтправьте сообщение, которое хотите разослать всем пользователям:"),
-        MessageInput(on_broadcast_input),
-        SwitchTo(Const("◀️ Отмена"), id="cancel_broadcast", state=CreatorMenuSG.main),
-        state=CreatorMenuSG.broadcast_input,
-    ),
-    Window(
-        Const("<b>⚠️ Подтверждение рассылки</b>\n\nВы уверены, что хотите отправить это сообщение всем пользователям?"),
-        Row(
-            Button(Const("✅ Да"), id="broadcast_confirm", on_click=on_broadcast_confirm),
-            Button(Const("❌ Нет"), id="broadcast_cancel", on_click=on_broadcast_cancel),
-        ),
-        state=CreatorMenuSG.broadcast_confirm,
-    ),
 )