84 lines
3.6 KiB
Python
84 lines
3.6 KiB
Python
"""My bookings handler - view, edit, cancel bookings."""
|
|
from aiogram import Router, F
|
|
from aiogram.types import Message, CallbackQuery
|
|
from aiogram.fsm.context import FSMContext
|
|
|
|
from app.config import USE_POSTGRES
|
|
from app.services.booking_service import get_user_bookings, cancel_booking
|
|
from app.keyboards.my_booking import get_booking_actions
|
|
from app.keyboards.services import get_services_keyboard
|
|
from app.states.booking import BookingState
|
|
from app.logger import logger
|
|
|
|
router = Router()
|
|
|
|
|
|
@router.message(F.text == "\U0001f4c4 \u041c\u043e\u0438 \u0437\u0430\u043f\u0438\u0441\u0438")
|
|
async def my_bookings(message: Message):
|
|
logger.info("User {uid} opened My Bookings", uid=message.from_user.id)
|
|
try:
|
|
bookings = await get_user_bookings(message.from_user.id)
|
|
logger.debug("Found {n} bookings", n=len(bookings))
|
|
|
|
if not bookings:
|
|
await message.answer("\u0423 \u0432\u0430\u0441 \u043d\u0435\u0442 \u0430\u043a\u0442\u0438\u0432\u043d\u044b\u0445 \u0437\u0430\u043f\u0438\u0441\u0435\u0439")
|
|
return
|
|
|
|
for booking in bookings:
|
|
svc = booking["service"]
|
|
mst = booking["master"]
|
|
dat = booking["date"]
|
|
tim = booking["time"]
|
|
text = (
|
|
"\U0001f486 " + svc + "\n"
|
|
"\U0001f464 " + mst + "\n"
|
|
"\U0001f4c5 " + dat + "\n"
|
|
"\U0001f552 " + tim
|
|
)
|
|
await message.answer(text, reply_markup=get_booking_actions(booking["id"]))
|
|
except Exception as e:
|
|
logger.exception("Error in my_bookings: {error}", error=e)
|
|
await message.answer("\u041f\u0440\u043e\u0438\u0437\u043e\u0448\u043b\u0430 \u043e\u0448\u0438\u0431\u043a\u0430. \u041f\u043e\u043f\u0440\u043e\u0431\u0443\u0439\u0442\u0435 \u043f\u043e\u0437\u0436\u0435.")
|
|
|
|
|
|
@router.callback_query(F.data.startswith("cancel_booking_"))
|
|
async def cancel_booking_handler(callback: CallbackQuery):
|
|
booking_id = int(callback.data.split("_")[2])
|
|
logger.info("User {uid} cancelled booking #{bid}", uid=callback.from_user.id, bid=booking_id)
|
|
await cancel_booking(booking_id)
|
|
await callback.message.edit_text("\u2705 \u0417\u0430\u043f\u0438\u0441\u044c \u043e\u0442\u043c\u0435\u043d\u0435\u043d\u0430")
|
|
await callback.answer()
|
|
|
|
|
|
@router.callback_query(F.data.startswith("edit_booking_"))
|
|
async def edit_booking_handler(callback: CallbackQuery, state: FSMContext):
|
|
booking_id = int(callback.data.split("_")[2])
|
|
|
|
if USE_POSTGRES:
|
|
from app.db.postgres import PostgresDB
|
|
booking = await PostgresDB.get_booking_by_id(booking_id)
|
|
else:
|
|
from app.db.memory import load_bookings
|
|
bookings = load_bookings()
|
|
booking = next((b for b in bookings if b["id"] == booking_id), None)
|
|
|
|
if not booking:
|
|
await callback.message.edit_text("\u0417\u0430\u043f\u0438\u0441\u044c \u043d\u0435 \u043d\u0430\u0439\u0434\u0435\u043d\u0430")
|
|
await callback.answer()
|
|
return
|
|
|
|
await state.set_data({
|
|
"editing_booking_id": booking_id,
|
|
"service": booking.get("service", ""),
|
|
"master": booking.get("master", ""),
|
|
"date": booking.get("date", ""),
|
|
"time": booking.get("time", ""),
|
|
})
|
|
|
|
await state.set_state(BookingState.service)
|
|
await callback.message.edit_text(
|
|
"\u270f\ufe0f \u0420\u0435\u0434\u0430\u043a\u0442\u0438\u0440\u043e\u0432\u0430\u043d\u0438\u0435 \u0437\u0430\u043f\u0438\u0441\u0438 #" + str(booking_id) + "\n\n\u0412\u044b\u0431\u0435\u0440\u0438\u0442\u0435 \u043d\u043e\u0432\u0443\u044e \u0443\u0441\u043b\u0443\u0433\u0443:",
|
|
reply_markup=get_services_keyboard()
|
|
)
|
|
await callback.answer()
|