separated clients handlers, cart->checkout ppline

handlers/client/checkout.py

@@ -0,0 +1,122 @@
+from aiogram import Router, F
+from aiogram.types import CallbackQuery, Message
+from aiogram.fsm.context import FSMContext
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from dto.checkout import CheckoutContext
+from misc.kb.client import (
+    order_confirmation,
+    order_specs_confirmation,
+    return_menu,
+)
+from misc.states import CheckoutStorage
+from misc.texts import get_order_item_list
+from misc.utils import is_valid_phone
+from repositories.orders import OrderRepository
+from services.orders import OrderService
+
+router = Router()
+
+
+@router.callback_query(F.data == "checkout")
+async def checkout(
+    cb: CallbackQuery,
+    state: FSMContext,
+    session: AsyncSession,
+    order_service: OrderService,
+):
+    await state.clear()
+
+    cart = await order_service.build_full_cart_dto(session, cb.from_user.id)
+    await cb.message.edit_text(
+        text=get_order_item_list(cart), reply_markup=order_confirmation(cart.order_id)
+    )
+
+
+@router.callback_query(F.data.startswith("confirm_order:"))
+async def order_confirm(
+    cb: CallbackQuery,
+    state: FSMContext,
+    session: AsyncSession,
+    order_repo: OrderRepository,
+):
+    await state.clear()
+
+    order_id = cb.data.split(":")[1]
+    if not order_id.isdigit():
+        await cb.message.edit_text("🍃 Что-то пошло не так.", reply_markup=return_menu)
+        return
+
+    order = await order_repo.get_order_by_id(session, int(order_id))
+    if not order or order.customer != cb.from_user.id:
+        await cb.message.edit_text("🍃 Что-то пошло не так.", reply_markup=return_menu)
+        return
+
+    await cb.message.edit_text("<b>👤 Укажите ваше имя:</b>", reply_markup=return_menu)
+
+    ctx = CheckoutContext(orig_msg=cb.message, order_id=order.id)
+    await state.set_state(CheckoutStorage.name)
+    await state.set_data({"ctx": ctx})
+
+
+@router.message(CheckoutStorage.name)
+async def checkout_name(msg: Message, state: FSMContext):
+    data = await state.get_data()
+    ctx: CheckoutContext = data.get("ctx", CheckoutContext)
+    await state.clear()
+
+    ctx.name = msg.text
+    await msg.delete()
+
+    await ctx.orig_msg.edit_text(
+        f"<i>👤 Имя: {ctx.name}</i>\n\n<b>📱 Укажите ваш номер телефона:</b>",
+        reply_markup=return_menu,
+    )
+
+    await state.set_state(CheckoutStorage.phone)
+    await state.set_data({"ctx": ctx})
+
+
+@router.message(CheckoutStorage.phone)
+async def checkout_phone(msg: Message, state: FSMContext):
+    data = await state.get_data()
+    ctx: CheckoutContext = data.get("ctx", CheckoutContext)
+    await state.clear()
+
+    await msg.delete()
+    if not is_valid_phone(msg.text):
+        await ctx.orig_msg.edit_text(
+            f"❌ <i>{msg.text}</i> не является действительным номером, попробуйте ещё раз.",
+            reply_markup=return_menu,
+        )
+        await state.set_state(CheckoutStorage.phone)
+        await state.set_data({"ctx": ctx})
+        return
+
+    ctx.phone = msg.text
+    await ctx.orig_msg.edit_text(
+        f"<i>👤 Имя: {ctx.name}</i>\n<i>📱 Номер телефона: {ctx.phone}</i>\n\n<b>📍 Укажите ваш адрес (город, улица):</b>",
+        reply_markup=return_menu,
+    )
+    await state.set_state(CheckoutStorage.address)
+    await state.set_data({"ctx": ctx})
+
+
+@router.message(CheckoutStorage.address)
+async def checkout_address(msg: Message, state: FSMContext):
+    data = await state.get_data()
+    ctx: CheckoutContext = data.get("ctx", CheckoutContext)
+    await state.clear()
+
+    await msg.delete()
+
+    ctx.address = msg.text
+    await ctx.orig_msg.edit_text(
+        "<b>‼️ Проверьте правильность ваших данных.</b>\n\n"
+        "━━━━━━━━━━━━━━\n\n"
+        f"<i>👤 Имя: {ctx.name}</i>\n"
+        f"<i>📱 Номер телефона: {ctx.phone}</i>\n"
+        f"<i>📍 Ваш адрес: {ctx.address}</i>\n\n"
+        "━━━━━━━━━━━━━━",
+        reply_markup=order_specs_confirmation(ctx.order_id),
+    )  # TODO: State-driven Payment System.