update

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

@@ -38,13 +38,9 @@ from dutylog.application.bot.admin_dialogs.rooms_management import (
     room_history_window,
     room_delete_confirm_window,
     room_add_hours_select_window,
-    room_remove_hours_select_window,
     room_add_hours_custom_window,
-    room_remove_hours_custom_window,
     room_add_hours_remark_window,
-    room_remove_hours_remark_window,
     room_add_hours_confirm_window,
-    room_remove_hours_confirm_window,
     create_room_select_floor_window,
     create_room_input_window,
     create_room_confirm_window,
@@ -106,13 +102,9 @@ admin_menu_dialog = Dialog(
     room_history_window,
     room_delete_confirm_window,
     room_add_hours_select_window,
-    room_remove_hours_select_window,
     room_add_hours_custom_window,
-    room_remove_hours_custom_window,
     room_add_hours_remark_window,
-    room_remove_hours_remark_window,
     room_add_hours_confirm_window,
-    room_remove_hours_confirm_window,
     create_room_select_floor_window,
     create_room_input_window,
     create_room_confirm_window,

src/dutylog/application/bot/admin_dialogs/rooms_management.py

@@ -257,14 +257,6 @@ async def on_room_add_hours_click(
     await dialog_manager.switch_to(AdminMenuSG.room_add_hours_select)
 
 
-async def on_room_remove_hours_click(
-    callback: CallbackQuery,
-    button: Button,
-    dialog_manager: DialogManager,
-):
-    await dialog_manager.switch_to(AdminMenuSG.room_remove_hours_select)
-
-
 async def get_room_hours_select_data(**kwargs):
     hours_options = [
         (5, "5"), (10, "10"), (15, "15"), (20, "20"),
@@ -282,11 +274,7 @@ async def on_room_hours_selected(
     item_id: str,
 ):
     dialog_manager.dialog_data["selected_hours"] = int(item_id)
-    
-    if dialog_manager.current_context().state == AdminMenuSG.room_add_hours_select:
-        await dialog_manager.switch_to(AdminMenuSG.room_add_hours_remark)
-    else:
-        await dialog_manager.switch_to(AdminMenuSG.room_remove_hours_remark)
+    await dialog_manager.switch_to(AdminMenuSG.room_add_hours_remark)
 
 
 async def on_room_custom_hours_click(
@@ -294,10 +282,7 @@ async def on_room_custom_hours_click(
     button: Button,
     dialog_manager: DialogManager,
 ):
-    if dialog_manager.current_context().state == AdminMenuSG.room_add_hours_select:
-        await dialog_manager.switch_to(AdminMenuSG.room_add_hours_custom)
-    else:
-        await dialog_manager.switch_to(AdminMenuSG.room_remove_hours_custom)
+    await dialog_manager.switch_to(AdminMenuSG.room_add_hours_custom)
 
 
 async def on_room_custom_hours_input(
@@ -316,11 +301,7 @@ async def on_room_custom_hours_input(
             return
         
         dialog_manager.dialog_data["selected_hours"] = hours
-        
-        if dialog_manager.current_context().state == AdminMenuSG.room_add_hours_custom:
-            await dialog_manager.switch_to(AdminMenuSG.room_add_hours_remark)
-        else:
-            await dialog_manager.switch_to(AdminMenuSG.room_remove_hours_remark)
+        await dialog_manager.switch_to(AdminMenuSG.room_add_hours_remark)
     except ValueError:
         await message.answer("⚠️ Пожалуйста, введите корректное число")
 
@@ -335,10 +316,7 @@ async def on_room_remark_input(
     else:
         dialog_manager.dialog_data["remark"] = None
     
-    if dialog_manager.current_context().state == AdminMenuSG.room_add_hours_remark:
-        await dialog_manager.switch_to(AdminMenuSG.room_add_hours_confirm)
-    else:
-        await dialog_manager.switch_to(AdminMenuSG.room_remove_hours_confirm)
+    await dialog_manager.switch_to(AdminMenuSG.room_add_hours_confirm)
 
 
 async def on_room_skip_remark(
@@ -347,11 +325,7 @@ async def on_room_skip_remark(
     dialog_manager: DialogManager,
 ):
     dialog_manager.dialog_data["remark"] = None
-    
-    if dialog_manager.current_context().state == AdminMenuSG.room_add_hours_remark:
-        await dialog_manager.switch_to(AdminMenuSG.room_add_hours_confirm)
-    else:
-        await dialog_manager.switch_to(AdminMenuSG.room_remove_hours_confirm)
+    await dialog_manager.switch_to(AdminMenuSG.room_add_hours_confirm)
 
 
 async def get_room_hours_confirm_data(
@@ -411,60 +385,6 @@ async def on_room_add_hours_confirm(
     await dialog_manager.switch_to(AdminMenuSG.room_info)
 
 
-@inject
-async def on_room_remove_hours_confirm(
-    callback: CallbackQuery,
-    button: Button,
-    dialog_manager: DialogManager,
-    transactions_repository: FromDishka[HoursTransactionsRepository],
-    rooms_repository: FromDishka[RoomsRepository],
-    residents_repository: FromDishka[ResidentsRepository],
-    users_repository: FromDishka[UsersRepository],
-    **kwargs,
-):
-    bot: Bot = dialog_manager.middleware_data.get("bot")
-    
-    room_id = dialog_manager.dialog_data.get("selected_room_id")
-    hours = dialog_manager.dialog_data.get("selected_hours")
-    remark = dialog_manager.dialog_data.get("remark")
-    admin_id = callback.from_user.id
-    
-    if room_id and hours:
-        residents = await residents_repository.get_residents_by_room(room_id)
-        
-        for resident in residents:
-            if resident.active_hours < hours:
-                await callback.answer(
-                    f"⚠️ Недостаточно часов! У резидента {resident.real_name or 'без имени'} только {resident.active_hours} неотработанных ч, а вы пытаетесь отнять {hours} ч",
-                    show_alert=True
-                )
-                await dialog_manager.switch_to(AdminMenuSG.room_info)
-                return
-        
-        results = await transactions_repository.remove_hours_from_room(
-            room_id=room_id,
-            amount=hours,
-            admin_id=admin_id,
-            remark=remark,
-        )
-        
-        for transaction, resident in results:
-            if resident and resident.user_entity:
-                user = await users_repository.get_user_by_id(resident.user_entity)
-                if user:
-                    try:
-                        remark_text = f"\n💬 <i>{remark}</i>" if remark else ""
-                        await bot.send_message(
-                            user.id,
-                            f"<blockquote>📢 <b>Уведомление</b></blockquote>\n\n"
-                            f"С вашей комнаты списано <b>-{hours}</b> ч{remark_text}"
-                        )
-                    except (TelegramBadRequest, TelegramForbiddenError, TelegramRetryAfter):
-                        pass
-    
-    await dialog_manager.switch_to(AdminMenuSG.room_info)
-
-
 async def on_room_hours_cancel(
     callback: CallbackQuery,
     button: Button,
@@ -603,17 +523,10 @@ rooms_list_window = Window(
 
 room_info_window = Window(
     Format("{info_content}"),
-    Row(
-        Button(
-            Const("Добавить часы"),
-            id="room_add_hours_btn",
-            on_click=on_room_add_hours_click,
-        ),
-        Button(
-            Const("Отнять часы"),
-            id="room_remove_hours_btn",
-            on_click=on_room_remove_hours_click,
-        ),
+    Button(
+        Const("Добавить часы"),
+        id="room_add_hours_btn",
+        on_click=on_room_add_hours_click,
     ),
     Button(
         Const("📜 История"),
@@ -678,32 +591,6 @@ room_add_hours_select_window = Window(
     getter=get_room_hours_select_data,
 )
 
-room_remove_hours_select_window = Window(
-    Const("<blockquote>➖ <b>Отнять часы у комнаты</b></blockquote>\n\nВыберите количество часов:"),
-    Group(
-        Select(
-            Format("{item[1]} ч"),
-            id="room_hours_select_remove",
-            item_id_getter=lambda x: x[0],
-            items="hours_options",
-            on_click=on_room_hours_selected,
-        ),
-        width=4,
-    ),
-    Button(
-        Const("✏️ Ввести свое количество"),
-        id="room_custom_hours_remove_btn",
-        on_click=on_room_custom_hours_click,
-    ),
-    SwitchTo(
-        Const("◀️ Отмена"),
-        id="cancel_room_remove_hours",
-        state=AdminMenuSG.room_info,
-    ),
-    state=AdminMenuSG.room_remove_hours_select,
-    getter=get_room_hours_select_data,
-)
-
 room_add_hours_custom_window = Window(
     Const("<blockquote>✏️ <b>Добавить часы</b></blockquote>\n\nВведите количество часов:"),
     MessageInput(on_room_custom_hours_input),
@@ -715,17 +602,6 @@ room_add_hours_custom_window = Window(
     state=AdminMenuSG.room_add_hours_custom,
 )
 
-room_remove_hours_custom_window = Window(
-    Const("<blockquote>✏️ <b>Отнять часы</b></blockquote>\n\nВведите количество часов:"),
-    MessageInput(on_room_custom_hours_input),
-    SwitchTo(
-        Const("◀️ Отмена"),
-        id="cancel_room_custom_remove",
-        state=AdminMenuSG.room_remove_hours_select,
-    ),
-    state=AdminMenuSG.room_remove_hours_custom,
-)
-
 room_add_hours_remark_window = Window(
     Const("<blockquote>💬 <b>Примечание</b></blockquote>\n\nВведите примечание к операции (или пропустите):"),
     MessageInput(on_room_remark_input),
@@ -742,22 +618,6 @@ room_add_hours_remark_window = Window(
     state=AdminMenuSG.room_add_hours_remark,
 )
 
-room_remove_hours_remark_window = Window(
-    Const("<blockquote>💬 <b>Примечание</b></blockquote>\n\nВведите примечание к операции (или пропустите):"),
-    MessageInput(on_room_remark_input),
-    Button(
-        Const("⏭ Пропустить"),
-        id="skip_room_remove_remark",
-        on_click=on_room_skip_remark,
-    ),
-    SwitchTo(
-        Const("◀️ Отмена"),
-        id="cancel_room_remove_remark",
-        state=AdminMenuSG.room_info,
-    ),
-    state=AdminMenuSG.room_remove_hours_remark,
-)
-
 room_add_hours_confirm_window = Window(
     Format("<blockquote>➕ <b>Подтверждение</b></blockquote>\n\nВы уверены, что хотите добавить <code>{hours}</code> часов?{remark_text}"),
     Row(
@@ -776,24 +636,6 @@ room_add_hours_confirm_window = Window(
     getter=get_room_hours_confirm_data,
 )
 
-room_remove_hours_confirm_window = Window(
-    Format("<blockquote>➖ <b>Подтверждение</b></blockquote>\n\nВы уверены, что хотите отнять <code>{hours}</code> часов?{remark_text}"),
-    Row(
-        Button(
-            Const("✅ Да"),
-            id="confirm_room_remove_hours",
-            on_click=on_room_remove_hours_confirm,
-        ),
-        Button(
-            Const("❌ Нет"),
-            id="cancel_room_remove_hours_confirm",
-            on_click=on_room_hours_cancel,
-        ),
-    ),
-    state=AdminMenuSG.room_remove_hours_confirm,
-    getter=get_room_hours_confirm_data,
-)
-
 create_room_select_floor_window = Window(
     Format("{content}"),
     Group(

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

@@ -45,13 +45,9 @@ class AdminMenuSG(StatesGroup):
     room_history = State()
     room_delete_confirm = State()
     room_add_hours_select = State()
-    room_remove_hours_select = State()
     room_add_hours_custom = State()
-    room_remove_hours_custom = State()
     room_add_hours_remark = State()
     room_add_hours_confirm = State()
-    room_remove_hours_remark = State()
-    room_remove_hours_confirm = State()
     create_room_select_floor = State()
     create_room_input = State()
     create_room_confirm = State()