mirror of
http://88.130.71.182:3000/BlitTech/badoHair_be.git
synced 2026-06-12 23:23:22 +00:00
66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
from datetime import date
|
|
from typing import Annotated
|
|
from fastapi import APIRouter, Depends, Query
|
|
import asyncpg
|
|
|
|
from app.core.pagination import pagination_params
|
|
from app.core.responses import ok, paginated
|
|
from app.dependencies import get_db, get_current_user, get_current_user_optional
|
|
from app.models.bookings import BookingCreate
|
|
from app.services import booking_service, slot_service
|
|
|
|
router = APIRouter(prefix="/bookings", tags=["Bookings"])
|
|
|
|
|
|
@router.get("/slots")
|
|
async def get_available_slots(
|
|
from_date: date = Query(...),
|
|
to_date: date = Query(...),
|
|
db: asyncpg.Connection = Depends(get_db),
|
|
):
|
|
slots = await slot_service.list_slots(db, from_date, to_date, available_only=True)
|
|
return ok(slots)
|
|
|
|
|
|
@router.post("", status_code=201)
|
|
async def create_booking(
|
|
body: BookingCreate,
|
|
user: dict | None = Depends(get_current_user_optional),
|
|
db: asyncpg.Connection = Depends(get_db),
|
|
):
|
|
result = await booking_service.create_booking(db, body, user=user)
|
|
return ok(result)
|
|
|
|
|
|
@router.get("")
|
|
async def list_my_bookings(
|
|
pagination: Annotated[tuple, Depends(pagination_params)],
|
|
status: str | None = Query(None),
|
|
user: dict = Depends(get_current_user),
|
|
db: asyncpg.Connection = Depends(get_db),
|
|
):
|
|
page, per_page, offset = pagination
|
|
bookings, total = await booking_service.list_bookings(
|
|
db, page, per_page, offset, user_id=str(user["id"]), status=status
|
|
)
|
|
return paginated(bookings, total, page, per_page)
|
|
|
|
|
|
@router.get("/{booking_id}")
|
|
async def get_booking(
|
|
booking_id: str,
|
|
user: dict = Depends(get_current_user),
|
|
db: asyncpg.Connection = Depends(get_db),
|
|
):
|
|
booking = await booking_service.get_booking(db, booking_id, user_id=str(user["id"]))
|
|
return ok(booking)
|
|
|
|
|
|
@router.delete("/{booking_id}", status_code=204)
|
|
async def cancel_booking(
|
|
booking_id: str,
|
|
user: dict = Depends(get_current_user),
|
|
db: asyncpg.Connection = Depends(get_db),
|
|
):
|
|
await booking_service.cancel_booking(db, booking_id, str(user["id"]))
|