update

src/dutylog/application/bot/user_dialogs/admin_dialogs/__init__.py

@@ -1,3 +1,5 @@
-from dutylog.application.bot.user_dialogs.admin_dialogs.admin_menu_dialog import admin_menu_dialog
+from dutylog.application.bot.user_dialogs.admin_dialogs.admin_menu_dialog import (
+    admin_menu_dialog,
+)
 
 __all__ = ["admin_menu_dialog"]

src/dutylog/application/bot/user_dialogs/admin_dialogs/admin_menu_dialog.py

@@ -3,15 +3,24 @@ from aiogram import Bot
 from aiogram.exceptions import TelegramForbiddenError, TelegramBadRequest
 from aiogram_dialog import Dialog, Window, DialogManager
 from aiogram_dialog.widgets.text import Format, Const
-from aiogram_dialog.widgets.kbd import Row, SwitchTo, Button, ScrollingGroup, Select
+from aiogram_dialog.widgets.kbd import Row, SwitchTo, Button, ScrollingGroup, Select, Group
 from aiogram_dialog.widgets.input import MessageInput
 from dishka import FromDishka
 from dishka.integrations.aiogram_dialog import inject
 
 from dutylog.application.bot.user_dialogs.states import AdminMenuSG
-from dutylog.infrastructure.database.repositories.users_repository import UsersRepository
-from dutylog.infrastructure.database.repositories.residents_repository import ResidentsRepository
-from dutylog.infrastructure.database.repositories.rooms_repository import RoomsRepository
+from dutylog.infrastructure.database.repositories.users_repository import (
+    UsersRepository,
+)
+from dutylog.infrastructure.database.repositories.residents_repository import (
+    ResidentsRepository,
+)
+from dutylog.infrastructure.database.repositories.rooms_repository import (
+    RoomsRepository,
+)
+from dutylog.infrastructure.database.repositories.hours_transactions_repository import (
+    HoursTransactionsRepository,
+)
 from dutylog.infrastructure.utils.config import Config
 
 
@@ -28,14 +37,14 @@ async def get_admin_menu_data(
         first_name=event_from_user.first_name,
         last_name=event_from_user.last_name,
     )
-    
+
     is_creator = event_from_user.id == config.bot.creator_id
-    
+
     if is_creator:
         greeting = "👑 <b>Создатель</b>"
     else:
         greeting = "👨‍💼 <b>Администратор</b>"
-    
+
     content = f"""
 {greeting}
 
@@ -43,7 +52,7 @@ async def get_admin_menu_data(
 
 Выберите действие:
 """
-    
+
     return {"content": content}
 
 
@@ -55,14 +64,14 @@ async def get_statistics_data(
 ):
     all_users = await users_repository.get_all_users()
     all_residents = await residents_repository.get_all_residents()
-    
+
     total_users = len(all_users)
     total_residents = len(all_residents)
     busy_residents = len([r for r in all_residents if r.is_busy])
     total_active_hours = sum(r.active_hours for r in all_residents)
     total_inactive_hours = sum(r.inactive_hours for r in all_residents)
     admins_count = len([u for u in all_users if u.is_admin])
-    
+
     stats_text = f"""
 <blockquote>📊 <b>Статистика системы</b></blockquote>
 
@@ -79,7 +88,7 @@ async def get_statistics_data(
 🔴 <b>Всего неактивных часов:</b> <code>{total_inactive_hours}</code> ч
 📊 <b>Общий итог:</b> <code>{total_active_hours + total_inactive_hours}</code> ч
 """
-    
+
     return {"stats_content": stats_text}
 
 
@@ -90,7 +99,7 @@ async def on_broadcast_message(
 ):
     dialog_manager.dialog_data["broadcast_message_id"] = message.message_id
     dialog_manager.dialog_data["broadcast_chat_id"] = message.chat.id
-    
+
     await message.copy_to(message.chat.id)
     await dialog_manager.switch_to(AdminMenuSG.broadcast_confirm)
 
@@ -103,21 +112,21 @@ async def on_broadcast_confirm(
     users_repository: FromDishka[UsersRepository],
 ):
     assert callback.message is not None
-    
+
     bot: Bot = dialog_manager.middleware_data["bot"]
     message_id = dialog_manager.dialog_data["broadcast_message_id"]
     chat_id = dialog_manager.dialog_data["broadcast_chat_id"]
     admin_id = callback.from_user.id
-    
+
     all_users = await users_repository.get_all_users()
-    
+
     success_count = 0
     failed_count = 0
-    
+
     for user in all_users:
         if user.id == admin_id:
             continue
-        
+
         try:
             await bot.copy_message(
                 chat_id=user.id,
@@ -129,9 +138,7 @@ async def on_broadcast_confirm(
             failed_count += 1
         except TelegramBadRequest:
             failed_count += 1
-        except Exception:
-            failed_count += 1
-    
+
     result_text = f"""
 <blockquote>📢 <b>Результаты рассылки</b></blockquote>
 
@@ -139,7 +146,7 @@ async def on_broadcast_confirm(
 ❌ <b>Не удалось отправить:</b> <code>{failed_count}</code>
 📊 <b>Всего пользователей:</b> <code>{len(all_users) - 1}</code>
 """
-    
+
     await callback.message.answer(result_text)
     await callback.message.delete()
     await dialog_manager.start(AdminMenuSG.main)
@@ -160,25 +167,24 @@ async def get_residents_list_data(
     **kwargs,
 ):
     all_residents = await residents_repository.get_all_residents()
-    
+
     residents_with_rooms = []
     for resident in all_residents:
         room = await rooms_repository.get_room_by_id(resident.room)
         room_number = room.number if room else 999999
         residents_with_rooms.append((resident, room_number))
-    
+
     residents_with_rooms.sort(key=lambda x: x[1])
-    
+
     residents_data = []
     for resident, room_number in residents_with_rooms:
         status = "🟢" if resident.is_busy else "⚪️"
         name = resident.real_name if resident.real_name else "Без имени"
-        
-        residents_data.append((
-            f"{name} | Комната {room_number} | {status}",
-            resident.id
-        ))
-    
+
+        residents_data.append(
+            (f"{name} | Комната {room_number} | {status}", resident.id)
+        )
+
     content = f"""
 <blockquote>🏠 <b>Резиденты</b></blockquote>
 
@@ -186,7 +192,7 @@ async def get_residents_list_data(
 
 Выберите резидента для просмотра информации:
 """
-    
+
     return {
         "content": content,
         "residents": residents_data,
@@ -202,28 +208,31 @@ async def get_resident_info_data(
     **kwargs,
 ):
     resident_id = dialog_manager.dialog_data.get("selected_resident_id")
-    
+
     if not resident_id:
         return {"info_content": "Ошибка: резидент не выбран"}
-    
+
     resident = await residents_repository.get_resident_by_id(resident_id)
-    
+
     if not resident:
         return {"info_content": "Ошибка: резидент не найден"}
-    
+
     room = await rooms_repository.get_room_by_id(resident.room)
     room_number = room.number if room else "???"
-    
+
     name = resident.real_name if resident.real_name else "Без имени"
     status = "🟢 Занят" if resident.is_busy else "⚪️ Свободен"
-    
+
     user_info = "Не привязан"
     if resident.user_entity:
         user = await users_repository.get_user_by_id(resident.user_entity)
         if user:
-            username = f"@{user.username}" if user.username else "без username"
+            if user.username:
+                username = f"@{user.username}"
+            else:
+                username = f"ID: {user.id}"
             user_info = f"{user.first_name} ({username})"
-    
+
     info_content = f"""
 <blockquote>👤 <b>Информация о резиденте</b></blockquote>
 
@@ -239,7 +248,7 @@ async def get_resident_info_data(
 🔴 <b>Неактивные часы:</b> <code>{resident.inactive_hours}</code> ч
 📊 <b>Всего часов:</b> <code>{resident.active_hours + resident.inactive_hours}</code> ч
 """
-    
+
     return {
         "info_content": info_content,
         "is_busy": resident.is_busy,
@@ -288,10 +297,10 @@ async def on_logout_resident_confirm(
     residents_repository: FromDishka[ResidentsRepository],
 ):
     resident_id = dialog_manager.dialog_data.get("selected_resident_id")
-    
+
     if resident_id:
         await residents_repository.unbind_user_from_resident(resident_id)
-    
+
     await dialog_manager.switch_to(AdminMenuSG.resident_info)
 
 
@@ -303,6 +312,142 @@ async def on_logout_resident_cancel(
     await dialog_manager.switch_to(AdminMenuSG.resident_info)
 
 
+async def on_add_hours_click(
+    callback: CallbackQuery,
+    button: Button,
+    dialog_manager: DialogManager,
+):
+    await dialog_manager.switch_to(AdminMenuSG.add_hours_select)
+
+
+async def on_remove_hours_click(
+    callback: CallbackQuery,
+    button: Button,
+    dialog_manager: DialogManager,
+):
+    await dialog_manager.switch_to(AdminMenuSG.remove_hours_select)
+
+
+async def get_hours_select_data(**kwargs):
+    hours_options = [
+        (5, "5"), (10, "10"), (15, "15"), (20, "20"),
+        (25, "25"), (30, "30"), (35, "35"), (40, "40"),
+        (45, "45"), (50, "50"), (55, "55"), (60, "60"),
+        (65, "65"), (70, "70"), (75, "75"), (80, "80"),
+    ]
+    return {"hours_options": hours_options}
+
+
+async def on_hours_selected(
+    callback: CallbackQuery,
+    widget: Select,
+    dialog_manager: DialogManager,
+    item_id: str,
+):
+    dialog_manager.dialog_data["selected_hours"] = int(item_id)
+    
+    if dialog_manager.current_context().state == AdminMenuSG.add_hours_select:
+        await dialog_manager.switch_to(AdminMenuSG.add_hours_confirm)
+    else:
+        await dialog_manager.switch_to(AdminMenuSG.remove_hours_confirm)
+
+
+async def on_custom_hours_click(
+    callback: CallbackQuery,
+    button: Button,
+    dialog_manager: DialogManager,
+):
+    if dialog_manager.current_context().state == AdminMenuSG.add_hours_select:
+        await dialog_manager.switch_to(AdminMenuSG.add_hours_custom)
+    else:
+        await dialog_manager.switch_to(AdminMenuSG.remove_hours_custom)
+
+
+async def on_custom_hours_input(
+    message: Message,
+    widget: MessageInput,
+    dialog_manager: DialogManager,
+):
+    if not message.text:
+        await message.answer("⚠️ Пожалуйста, введите число")
+        return
+    
+    try:
+        hours = int(message.text)
+        if hours <= 0:
+            await message.answer("⚠️ Количество часов должно быть положительным числом")
+            return
+        
+        dialog_manager.dialog_data["selected_hours"] = hours
+        
+        if dialog_manager.current_context().state == AdminMenuSG.add_hours_custom:
+            await dialog_manager.switch_to(AdminMenuSG.add_hours_confirm)
+        else:
+            await dialog_manager.switch_to(AdminMenuSG.remove_hours_confirm)
+    except ValueError:
+        await message.answer("⚠️ Пожалуйста, введите корректное число")
+
+
+async def get_hours_confirm_data(
+    dialog_manager: DialogManager,
+    **kwargs,
+):
+    hours = dialog_manager.dialog_data.get("selected_hours", 0)
+    return {"hours": hours}
+
+
+@inject
+async def on_add_hours_confirm(
+    callback: CallbackQuery,
+    button: Button,
+    dialog_manager: DialogManager,
+    transactions_repository: FromDishka[HoursTransactionsRepository],
+):
+    resident_id = dialog_manager.dialog_data.get("selected_resident_id")
+    hours = dialog_manager.dialog_data.get("selected_hours")
+    admin_id = callback.from_user.id
+    
+    if resident_id and hours:
+        await transactions_repository.add_hours(
+            resident_id=resident_id,
+            amount=hours,
+            admin_id=admin_id,
+            is_active=True,
+        )
+    
+    await dialog_manager.switch_to(AdminMenuSG.resident_info)
+
+
+@inject
+async def on_remove_hours_confirm(
+    callback: CallbackQuery,
+    button: Button,
+    dialog_manager: DialogManager,
+    transactions_repository: FromDishka[HoursTransactionsRepository],
+):
+    resident_id = dialog_manager.dialog_data.get("selected_resident_id")
+    hours = dialog_manager.dialog_data.get("selected_hours")
+    admin_id = callback.from_user.id
+    
+    if resident_id and hours:
+        await transactions_repository.remove_hours(
+            resident_id=resident_id,
+            amount=hours,
+            admin_id=admin_id,
+            is_active=True,
+        )
+    
+    await dialog_manager.switch_to(AdminMenuSG.resident_info)
+
+
+async def on_hours_cancel(
+    callback: CallbackQuery,
+    button: Button,
+    dialog_manager: DialogManager,
+):
+    await dialog_manager.switch_to(AdminMenuSG.resident_info)
+
+
 admin_menu_dialog = Dialog(
     Window(
         Format("{content}"),
@@ -311,11 +456,6 @@ admin_menu_dialog = Dialog(
             id="residents_btn",
             state=AdminMenuSG.residents,
         ),
-        SwitchTo(
-            Const("👥 Пользователи"),
-            id="users_btn",
-            state=AdminMenuSG.users,
-        ),
         SwitchTo(
             Const("📊 Статистика"),
             id="stats_btn",
@@ -370,6 +510,18 @@ admin_menu_dialog = Dialog(
     ),
     Window(
         Format("{info_content}"),
+        Row(
+            Button(
+                Const("➕ Добавить часы"),
+                id="add_hours_btn",
+                on_click=on_add_hours_click,
+            ),
+            Button(
+                Const("➖ Отнять часы"),
+                id="remove_hours_btn",
+                on_click=on_remove_hours_click,
+            ),
+        ),
         Button(
             Const("🚪 Разлогинить"),
             id="logout_resident_btn",
@@ -385,7 +537,9 @@ admin_menu_dialog = Dialog(
         getter=get_resident_info_data,
     ),
     Window(
-        Const("<blockquote>⚠️ <b>Подтверждение</b></blockquote>\n\nВы уверены, что хотите разлогинить этого резидента?"),
+        Const(
+            "<blockquote>⚠️ <b>Подтверждение</b></blockquote>\n\nВы уверены, что хотите разлогинить этого резидента?"
+        ),
         Row(
             Button(
                 Const("✅ Да"),
@@ -401,9 +555,108 @@ admin_menu_dialog = Dialog(
         state=AdminMenuSG.resident_logout_confirm,
     ),
     Window(
-        Const("<blockquote>👥 <b>Пользователи</b></blockquote>\n\n<i>Функционал в разработке</i>"),
-        SwitchTo(Const("◀️ Назад"), id="back_from_users", state=AdminMenuSG.main),
-        state=AdminMenuSG.users,
+        Const("<blockquote>➕ <b>Добавить часы</b></blockquote>\n\nВыберите количество часов:"),
+        Group(
+            Select(
+                Format("{item[1]} ч"),
+                id="hours_select_add",
+                item_id_getter=lambda x: x[0],
+                items="hours_options",
+                on_click=on_hours_selected,
+            ),
+            width=4,
+        ),
+        Button(
+            Const("✏️ Ввести свое количество"),
+            id="custom_hours_add_btn",
+            on_click=on_custom_hours_click,
+        ),
+        SwitchTo(
+            Const("◀️ Отмена"),
+            id="cancel_add_hours",
+            state=AdminMenuSG.resident_info,
+        ),
+        state=AdminMenuSG.add_hours_select,
+        getter=get_hours_select_data,
+    ),
+    Window(
+        Const("<blockquote>➖ <b>Отнять часы</b></blockquote>\n\nВыберите количество часов:"),
+        Group(
+            Select(
+                Format("{item[1]} ч"),
+                id="hours_select_remove",
+                item_id_getter=lambda x: x[0],
+                items="hours_options",
+                on_click=on_hours_selected,
+            ),
+            width=4,
+        ),
+        Button(
+            Const("✏️ Ввести свое количество"),
+            id="custom_hours_remove_btn",
+            on_click=on_custom_hours_click,
+        ),
+        SwitchTo(
+            Const("◀️ Отмена"),
+            id="cancel_remove_hours",
+            state=AdminMenuSG.resident_info,
+        ),
+        state=AdminMenuSG.remove_hours_select,
+        getter=get_hours_select_data,
+    ),
+    Window(
+        Const("<blockquote>✏️ <b>Добавить часы</b></blockquote>\n\nВведите количество часов:"),
+        MessageInput(on_custom_hours_input),
+        SwitchTo(
+            Const("◀️ Отмена"),
+            id="cancel_custom_add",
+            state=AdminMenuSG.add_hours_select,
+        ),
+        state=AdminMenuSG.add_hours_custom,
+    ),
+    Window(
+        Const("<blockquote>✏️ <b>Отнять часы</b></blockquote>\n\nВведите количество часов:"),
+        MessageInput(on_custom_hours_input),
+        SwitchTo(
+            Const("◀️ Отмена"),
+            id="cancel_custom_remove",
+            state=AdminMenuSG.remove_hours_select,
+        ),
+        state=AdminMenuSG.remove_hours_custom,
+    ),
+    Window(
+        Format("<blockquote>➕ <b>Подтверждение</b></blockquote>\n\nВы уверены, что хотите добавить <code>{hours}</code> часов?"),
+        Row(
+            Button(
+                Const("✅ Да"),
+                id="confirm_add_hours",
+                on_click=on_add_hours_confirm,
+            ),
+            Button(
+                Const("❌ Нет"),
+                id="cancel_add_hours_confirm",
+                on_click=on_hours_cancel,
+            ),
+        ),
+        state=AdminMenuSG.add_hours_confirm,
+        getter=get_hours_confirm_data,
+    ),
+    Window(
+        Format("<blockquote>➖ <b>Подтверждение</b></blockquote>\n\nВы уверены, что хотите отнять <code>{hours}</code> часов?"),
+        Row(
+            Button(
+                Const("✅ Да"),
+                id="confirm_remove_hours",
+                on_click=on_remove_hours_confirm,
+            ),
+            Button(
+                Const("❌ Нет"),
+                id="cancel_remove_hours_confirm",
+                on_click=on_hours_cancel,
+            ),
+        ),
+        state=AdminMenuSG.remove_hours_confirm,
+        getter=get_hours_confirm_data,
     ),
     Window(
         Format("{stats_content}"),
@@ -412,16 +665,26 @@ admin_menu_dialog = Dialog(
         getter=get_statistics_data,
     ),
     Window(
-        Const("<blockquote>📢 <b>Рассылка</b></blockquote>\n\nОтправьте сообщение, которое хотите разослать всем пользователям:"),
+        Const(
+            "<blockquote>📢 <b>Рассылка</b></blockquote>\n\nОтправьте сообщение, которое хотите разослать всем пользователям:"
+        ),
         MessageInput(on_broadcast_message),
         SwitchTo(Const("◀️ Отмена"), id="cancel_broadcast", state=AdminMenuSG.main),
         state=AdminMenuSG.broadcast,
     ),
     Window(
-        Const("<blockquote>📢 <b>Подтверждение рассылки</b></blockquote>\n\n⚠️ Вы уверены, что хотите отправить это сообщение всем пользователям?"),
+        Const(
+            "<blockquote>📢 <b>Подтверждение рассылки</b></blockquote>\n\n⚠️ Вы уверены, что хотите отправить это сообщение всем пользователям?"
+        ),
         Row(
-            Button(Const("✅ Да"), id="confirm_broadcast", on_click=on_broadcast_confirm),
-            Button(Const("❌ Нет"), id="cancel_broadcast_confirm", on_click=on_broadcast_cancel),
+            Button(
+                Const("✅ Да"), id="confirm_broadcast", on_click=on_broadcast_confirm
+            ),
+            Button(
+                Const("❌ Нет"),
+                id="cancel_broadcast_confirm",
+                on_click=on_broadcast_cancel,
+            ),
         ),
         state=AdminMenuSG.broadcast_confirm,
     ),

src/dutylog/application/bot/user_dialogs/main_menu_dialog.py

@@ -6,10 +6,18 @@ from dishka import FromDishka
 from dishka.integrations.aiogram_dialog import inject
 
 from dutylog.application.bot.user_dialogs.states import MainMenuSG
-from dutylog.infrastructure.database.repositories.users_repository import UsersRepository
-from dutylog.infrastructure.database.repositories.residents_repository import ResidentsRepository
-from dutylog.infrastructure.database.repositories.rooms_repository import RoomsRepository
-from dutylog.infrastructure.database.repositories.hours_transactions_repository import HoursTransactionsRepository
+from dutylog.infrastructure.database.repositories.users_repository import (
+    UsersRepository,
+)
+from dutylog.infrastructure.database.repositories.residents_repository import (
+    ResidentsRepository,
+)
+from dutylog.infrastructure.database.repositories.rooms_repository import (
+    RoomsRepository,
+)
+from dutylog.infrastructure.database.repositories.hours_transactions_repository import (
+    HoursTransactionsRepository,
+)
 from dutylog.infrastructure.utils.config import Config
 
 
@@ -28,27 +36,35 @@ async def get_main_menu_data(
         first_name=event_from_user.first_name,
         last_name=event_from_user.last_name,
     )
-    
+
     is_creator = event_from_user.id == config.bot.creator_id
     is_admin = user.is_admin
-    
+
     if is_creator:
         greeting = "👑 <b>Создатель</b>"
     elif is_admin:
         greeting = "👨‍💼 <b>Администратор</b>"
     else:
-        resident = await residents_repository.get_resident_by_user_id(event_from_user.id)
+        resident = await residents_repository.get_resident_by_user_id(
+            event_from_user.id
+        )
         if resident:
             room = await rooms_repository.get_room_by_id(resident.room)
             room_number = room.number if room else "???"
-            real_name = resident.real_name if resident.real_name else event_from_user.first_name
-            greeting = f"👋 <b>Привет, {real_name}!</b>\n🚪 Комната <code>{room_number}</code>"
+            real_name = (
+                resident.real_name if resident.real_name else event_from_user.first_name
+            )
+            greeting = (
+                f"👋 <b>Привет, {real_name}!</b>\n🚪 Комната <code>{room_number}</code>"
+            )
         else:
             greeting = f"👋 <b>Привет, {event_from_user.first_name}!</b>"
-    
+
     if not is_admin and not is_creator:
-        resident = await residents_repository.get_resident_by_user_id(event_from_user.id)
-        
+        resident = await residents_repository.get_resident_by_user_id(
+            event_from_user.id
+        )
+
         if not resident:
             content = f"""
 {greeting}
@@ -79,7 +95,7 @@ async def get_main_menu_data(
 Добро пожаловать в систему учета дежурств!
 """
         has_resident = False
-    
+
     return {
         "content": content,
         "is_regular_user": not is_admin and not is_creator,
@@ -95,7 +111,7 @@ async def get_history_data(
     **kwargs,
 ):
     resident = await residents_repository.get_resident_by_user_id(event_from_user.id)
-    
+
     if not resident:
         history_text = """
 <blockquote>📜 <b>История операций</b></blockquote>
@@ -105,7 +121,7 @@ async def get_history_data(
     else:
         transactions = await transactions_repository.get_resident_history(resident.id)
         last_10 = transactions[:10]
-        
+
         if not last_10:
             history_text = """
 <blockquote>📜 <b>История операций</b></blockquote>
@@ -120,14 +136,14 @@ async def get_history_data(
                 history_lines.append(
                     f"{emoji} <code>{tx.amount}</code> ч • <i>{date_str}</i>"
                 )
-            
+
             history_text = f"""
 <blockquote>📜 <b>История операций</b></blockquote>
 
 {"".join(f"{line}\n" for line in history_lines)}
 <i>Показаны последние 10 операций</i>
 """
-    
+
     return {"history_content": history_text}
 
 
@@ -177,4 +193,3 @@ main_menu_dialog = Dialog(
         state=MainMenuSG.faq,
     ),
 )
-

src/dutylog/application/bot/user_dialogs/registration_dialog.py

@@ -7,9 +7,15 @@ from dishka import FromDishka
 from dishka.integrations.aiogram_dialog import inject
 
 from dutylog.application.bot.user_dialogs.states import RegistrationSG, MainMenuSG
-from dutylog.infrastructure.database.repositories.floors_repository import FloorsRepository
-from dutylog.infrastructure.database.repositories.rooms_repository import RoomsRepository
-from dutylog.infrastructure.database.repositories.residents_repository import ResidentsRepository
+from dutylog.infrastructure.database.repositories.floors_repository import (
+    FloorsRepository,
+)
+from dutylog.infrastructure.database.repositories.rooms_repository import (
+    RoomsRepository,
+)
+from dutylog.infrastructure.database.repositories.residents_repository import (
+    ResidentsRepository,
+)
 
 
 @inject
@@ -21,21 +27,21 @@ async def get_floors_data(
 ):
     all_floors = await floors_repository.get_all_floors()
     available_residents = await residents_repository.get_available_residents()
-    
+
     if not available_residents:
         return {
             "has_available": False,
             "floors": [],
         }
-    
+
     available_room_ids = {r.room for r in available_residents}
     all_rooms = await rooms_repository.get_all_rooms()
     available_rooms = [r for r in all_rooms if r.id in available_room_ids]
     available_floor_ids = {r.on_floor for r in available_rooms}
-    
+
     available_floors = [f for f in all_floors if f.id in available_floor_ids]
     available_floors.sort(key=lambda f: f.number)
-    
+
     return {
         "has_available": True,
         "floors": [(f.id, f"Этаж {f.number}") for f in available_floors],
@@ -50,14 +56,14 @@ async def get_rooms_data(
     **kwargs,
 ):
     floor_id = dialog_manager.dialog_data["floor_id"]
-    
+
     rooms = await rooms_repository.get_rooms_by_floor(floor_id)
     available_residents = await residents_repository.get_available_residents()
     available_room_ids = {r.room for r in available_residents}
-    
+
     available_rooms = [r for r in rooms if r.id in available_room_ids]
     available_rooms.sort(key=lambda r: r.number)
-    
+
     return {
         "rooms": [(r.id, str(r.number)) for r in available_rooms],
     }
@@ -70,12 +76,14 @@ async def get_residents_data(
     **kwargs,
 ):
     room_id = dialog_manager.dialog_data["room_id"]
-    
+
     residents = await residents_repository.get_residents_by_room(room_id)
     available_residents = [r for r in residents if not r.is_busy]
-    
+
     return {
-        "residents": [(r.id, r.real_name or f"Резидент #{r.id}") for r in available_residents],
+        "residents": [
+            (r.id, r.real_name or f"Резидент #{r.id}") for r in available_residents
+        ],
     }
 
 
@@ -109,17 +117,23 @@ async def on_resident_selected(
 ):
     user_id = callback.from_user.id
     resident_id = int(item_id)
-    
+
     await residents_repository.bind_user_to_resident(resident_id, user_id)
-    
+
     await callback.answer("✅ Регистрация успешна!")
     await dialog_manager.start(MainMenuSG.main)
 
 
 registration_dialog = Dialog(
     Window(
-        Const("<blockquote>🏢 <b>Выбор этажа</b></blockquote>\n\n<blockquote>⚠️ <b>Внимание!</b> Перерегистрацию может выполнить только администратор. Выбирайте внимательно!</blockquote>\n\nВыберите этаж, на котором вы живете:", when="has_available"),
-        Const("<blockquote>⚠️ <b>Нет доступных резидентов</b></blockquote>\n\nВсе резиденты уже заняты.\nОбратитесь к администратору.", when=~F["has_available"]),
+        Const(
+            "<blockquote>🏢 <b>Выбор этажа</b></blockquote>\n\n<blockquote>⚠️ <b>Внимание!</b> Перерегистрацию может выполнить только администратор. Выбирайте внимательно!</blockquote>\n\nВыберите этаж, на котором вы живете:",
+            when="has_available",
+        ),
+        Const(
+            "<blockquote>⚠️ <b>Нет доступных резидентов</b></blockquote>\n\nВсе резиденты уже заняты.\nОбратитесь к администратору.",
+            when=~F["has_available"],
+        ),
         Group(
             Select(
                 Format("{item[1]}"),
@@ -135,7 +149,9 @@ registration_dialog = Dialog(
         getter=get_floors_data,
     ),
     Window(
-        Const("<blockquote>🚪 <b>Выбор комнаты</b></blockquote>\n\nВыберите вашу комнату:"),
+        Const(
+            "<blockquote>🚪 <b>Выбор комнаты</b></blockquote>\n\nВыберите вашу комнату:"
+        ),
         Group(
             Select(
                 Format("{item[1]}"),
@@ -146,12 +162,16 @@ registration_dialog = Dialog(
             ),
             width=3,
         ),
-        SwitchTo(Const("◀️ Назад"), id="back_to_floors", state=RegistrationSG.select_floor),
+        SwitchTo(
+            Const("◀️ Назад"), id="back_to_floors", state=RegistrationSG.select_floor
+        ),
         state=RegistrationSG.select_room,
         getter=get_rooms_data,
     ),
     Window(
-        Const("<blockquote>👤 <b>Выбор резидента</b></blockquote>\n\nВыберите себя из списка:"),
+        Const(
+            "<blockquote>👤 <b>Выбор резидента</b></blockquote>\n\nВыберите себя из списка:"
+        ),
         Group(
             Select(
                 Format("{item[1]}"),
@@ -162,7 +182,9 @@ registration_dialog = Dialog(
             ),
             width=1,
         ),
-        SwitchTo(Const("◀️ Назад"), id="back_to_rooms", state=RegistrationSG.select_room),
+        SwitchTo(
+            Const("◀️ Назад"), id="back_to_rooms", state=RegistrationSG.select_room
+        ),
         state=RegistrationSG.select_resident,
         getter=get_residents_data,
     ),

src/dutylog/application/bot/user_dialogs/states.py

@@ -12,7 +12,12 @@ class AdminMenuSG(StatesGroup):
     residents = State()
     resident_info = State()
     resident_logout_confirm = State()
-    users = State()
+    add_hours_select = State()
+    remove_hours_select = State()
+    add_hours_custom = State()
+    remove_hours_custom = State()
+    add_hours_confirm = State()
+    remove_hours_confirm = State()
     statistics = State()
     broadcast = State()
     broadcast_confirm = State()