product management, unification of kb and so on.

handlers/admins/product_mgmt.py

@@ -0,0 +1,95 @@
+import logging
+from aiogram import Router, F
+from aiogram.types import Message, CallbackQuery
+from aiogram.fsm.context import FSMContext
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from dto.control import EditProductContext
+from misc.filters import IsVerified
+from misc.kb.admins import back_to_product_kb
+from misc.states import AdminControlStorage
+from misc.texts import product_editing_mapping
+from repositories.products import ProductRepository
+
+logger = logging.getLogger(__name__)
+router = Router()
+
+
+@router.callback_query(IsVerified(), F.data.startswith("edit_product:"))
+async def edit_product(cb: CallbackQuery, state: FSMContext):
+    await state.clear()
+
+    data = cb.data.split(":")
+    product_id = data[1]
+    mode = data[2]
+
+    if cb.message.photo:
+        await cb.message.delete()
+        msg = await cb.message.answer(
+            f"<b>✍️ Введите {product_editing_mapping.get(mode)}</b>",
+            reply_markup=back_to_product_kb(
+                product_id, cb_factory=lambda p: f"product:{p}"
+            ),
+        )
+
+    else:
+        msg = cb.message
+        await cb.message.edit_text(
+            f"<b>✍️ Введите {product_editing_mapping.get(mode)}</b>",
+            reply_markup=back_to_product_kb(
+                product_id, cb_factory=lambda p: f"product:{p}"
+            ),
+        )
+
+    ctx = EditProductContext(int(product_id), mode, msg)
+    await state.set_state(AdminControlStorage.edit_product)
+    await state.set_data({"ctx": ctx})
+
+
+@router.message(AdminControlStorage.edit_product)
+async def edit_product_query(
+    msg: Message,
+    state: FSMContext,
+    session: AsyncSession,
+    products_repo: ProductRepository,
+):
+    data = await state.get_data()
+    await state.clear()
+
+    await msg.delete()
+
+    ctx: EditProductContext = data.get("ctx", EditProductContext)
+    kb = back_to_product_kb(ctx.product_id, cb_factory=lambda p: f"product:{p}")
+
+    try:
+        await ctx.msg.edit_text("⏳")
+        if ctx.mode == "name":
+            await products_repo.update_product_name_by_id(
+                session, ctx.product_id, msg.text
+            )
+        elif ctx.mode == "description":
+            await products_repo.update_product_description_by_id(
+                session, ctx.product_id, msg.text
+            )
+        elif ctx.mode == "price":
+            if msg.text.isdigit():
+                await products_repo.update_product_price_by_id(
+                    session, ctx.product_id, int(msg.text)
+                )
+            else:
+                await ctx.msg.edit_text(
+                    "❌ Вы ввели не число, попробуйте ещё раз...",
+                    reply_markup=kb,
+                )
+                await state.set_state(AdminControlStorage.edit_product)
+                await state.set_data({"ctx": ctx})
+                return
+        else:
+            await ctx.msg.edit_text("🍃", reply_markup=kb)
+            return
+
+        await ctx.msg.edit_text("✅ Изменения применены.", reply_markup=kb)
+
+    except Exception as e:
+        logger.exception(e)
+        await ctx.msg.edit_text(text="‼️ Что-то пошло не так.", reply_markup=kb)