Felipe Cardoso
85eed0eac0
This commit introduces a new RSVP module with endpoints for creating, reading, updating, and deleting RSVPs, along with a custom status update endpoint. The router is integrated into the events API, providing full RSVP management capabilities. Validation and error handling have been implemented to ensure data consistency.
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from typing import List
|
|
from uuid import UUID
|
|
|
|
from app.core.database import get_db
|
|
from app.schemas.rsvp import RSVPSchema, RSVPSchemaCreate, RSVPSchemaUpdate
|
|
from app.crud.rsvp import crud_rsvp
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/", response_model=RSVPSchema, operation_id="create_rsvp")
|
|
def create_rsvp(
|
|
rsvp_in: RSVPSchemaCreate,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
# First, check if RSVP already exists to prevent duplicates
|
|
existing_rsvp = crud_rsvp.get_rsvp_by_event_and_guest(
|
|
db, event_id=rsvp_in.event_id, guest_id=rsvp_in.guest_id
|
|
)
|
|
if existing_rsvp:
|
|
raise HTTPException(
|
|
status_code=400, detail="RSVP already exists for this event and guest"
|
|
)
|
|
|
|
rsvp = crud_rsvp.create(db=db, obj_in=rsvp_in)
|
|
return rsvp
|
|
|
|
|
|
@router.get("/{rsvp_id}", response_model=RSVPSchema, operation_id="get_rsvp")
|
|
def read_rsvp(rsvp_id: UUID, db: Session = Depends(get_db)):
|
|
rsvp = crud_rsvp.get(db=db, id=str(rsvp_id))
|
|
if not rsvp:
|
|
raise HTTPException(status_code=404, detail="RSVP not found")
|
|
return rsvp
|
|
|
|
|
|
@router.get("/", response_model=List[RSVPSchema], operation_id="get_rsvps")
|
|
def read_rsvps(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
|
|
rsvps = crud_rsvp.get_multi(db=db, skip=skip, limit=limit)
|
|
return rsvps
|
|
|
|
|
|
@router.put("/{rsvp_id}", response_model=RSVPSchema, operation_id="update_rsvp")
|
|
def update_rsvp(
|
|
rsvp_id: UUID, rsvp_in: RSVPSchemaUpdate, db: Session = Depends(get_db)
|
|
):
|
|
rsvp = crud_rsvp.get(db, id=str(rsvp_id))
|
|
if not rsvp:
|
|
raise HTTPException(status_code=404, detail="RSVP not found")
|
|
|
|
updated_rsvp = crud_rsvp.update(db, db_obj=rsvp, obj_in=rsvp_in)
|
|
return updated_rsvp
|
|
|
|
|
|
@router.delete("/{rsvp_id}", response_model=RSVPSchema, operation_id="delete_rsvp")
|
|
def delete_rsvp(rsvp_id: UUID, db: Session = Depends(get_db)):
|
|
rsvp = crud_rsvp.get(db, id=str(rsvp_id))
|
|
if not rsvp:
|
|
raise HTTPException(status_code=404, detail="RSVP not found")
|
|
|
|
rsvp = crud_rsvp.remove(db, id=str(rsvp_id))
|
|
return rsvp
|
|
|
|
|
|
@router.patch("/{rsvp_id}/status", response_model=RSVPSchema, operation_id="update_rsvp_status")
|
|
def update_rsvp_status(
|
|
rsvp_id: UUID,
|
|
status: RSVPSchemaUpdate,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
rsvp = crud_rsvp.get(db, id=str(rsvp_id))
|
|
if not rsvp:
|
|
raise HTTPException(status_code=404, detail="RSVP not found")
|
|
|
|
# Validation of provided fields specifically for status update
|
|
if not status.status:
|
|
raise HTTPException(status_code=400, detail="Status field required for update.")
|
|
|
|
updated_rsvp = crud_rsvp.update_rsvp_status(
|
|
db,
|
|
db_obj=rsvp,
|
|
status=status.status,
|
|
number_of_guests=status.number_of_guests,
|
|
response_message=status.response_message,
|
|
dietary_requirements=status.dietary_requirements,
|
|
additional_info=status.additional_info
|
|
)
|
|
return updated_rsvp |