123 lines
4.1 KiB
Python
123 lines
4.1 KiB
Python
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.
|