53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
import uuid
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from backend.app.core.database import get_session
|
|
from backend.app.models.rider import Rider
|
|
from backend.app.schemas.rider import RiderCreate, RiderUpdate, RiderResponse
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/profile", response_model=RiderResponse)
|
|
async def create_rider(
|
|
data: RiderCreate,
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
rider = Rider(**data.model_dump())
|
|
session.add(rider)
|
|
await session.commit()
|
|
await session.refresh(rider)
|
|
return rider
|
|
|
|
|
|
@router.get("/profile/{rider_id}", response_model=RiderResponse)
|
|
async def get_rider(
|
|
rider_id: uuid.UUID,
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
rider = await session.get(Rider, rider_id)
|
|
if not rider:
|
|
raise HTTPException(status_code=404, detail="Rider not found")
|
|
return rider
|
|
|
|
|
|
@router.put("/profile/{rider_id}", response_model=RiderResponse)
|
|
async def update_rider(
|
|
rider_id: uuid.UUID,
|
|
data: RiderUpdate,
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
rider = await session.get(Rider, rider_id)
|
|
if not rider:
|
|
raise HTTPException(status_code=404, detail="Rider not found")
|
|
|
|
update_data = data.model_dump(exclude_unset=True)
|
|
for key, value in update_data.items():
|
|
setattr(rider, key, value)
|
|
|
|
await session.commit()
|
|
await session.refresh(rider)
|
|
return rider
|