104 lines
3.9 KiB
Python
104 lines
3.9 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from app.schemas.guests import GuestCreate, GuestUpdate, GuestRead, GuestWithRSVPResponse, map_rsvp_status_to_guest_status
|
|
from app.crud.guest import guest_crud
|
|
from typing import List
|
|
import uuid
|
|
|
|
from app.core.database import get_db
|
|
from app.models import GuestStatus
|
|
from app.crud.rsvp import crud_rsvp
|
|
from app.schemas.rsvp import RSVPSchemaCreate
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/", response_model=GuestRead, operation_id="create_guest")
|
|
def create_guest(guest_in: GuestCreate, db: Session = Depends(get_db)):
|
|
if guest_in.invitation_code:
|
|
existing_guest = guest_crud.get_by_invitation_code(db, guest_in.invitation_code)
|
|
if existing_guest:
|
|
raise HTTPException(
|
|
status_code=400, detail="Guest with this invitation code already exists"
|
|
)
|
|
else:
|
|
guest_in.invitation_code = str(uuid.uuid4())[:8]
|
|
guest = guest_crud.create(db, obj_in=guest_in)
|
|
return guest
|
|
|
|
|
|
@router.get("/{guest_id}", response_model=GuestRead, operation_id="get_guest")
|
|
def read_guest(guest_id: uuid.UUID, db: Session = Depends(get_db)):
|
|
guest = guest_crud.get(db, guest_id)
|
|
if not guest:
|
|
raise HTTPException(status_code=404, detail="Guest not found")
|
|
return guest
|
|
|
|
|
|
@router.get("/", response_model=List[GuestRead], operation_id="get_guests")
|
|
def read_guests(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
|
|
guests = guest_crud.get_multi(db, skip=skip, limit=limit)
|
|
return guests
|
|
|
|
|
|
@router.put("/{guest_id}", response_model=GuestRead, operation_id="update_guest")
|
|
def update_guest(guest_id: uuid.UUID, guest_in: GuestUpdate, db: Session = Depends(get_db)):
|
|
guest = guest_crud.get(db, guest_id)
|
|
if not guest:
|
|
raise HTTPException(status_code=404, detail="Guest not found")
|
|
guest = guest_crud.update(db, db_obj=guest, obj_in=guest_in)
|
|
return guest
|
|
|
|
|
|
@router.delete("/{guest_id}", response_model=GuestRead, operation_id="delete_guest")
|
|
def delete_guest(guest_id: uuid.UUID, db: Session = Depends(get_db)):
|
|
guest = guest_crud.get(db, guest_id)
|
|
if not guest:
|
|
raise HTTPException(status_code=404, detail="Guest not found")
|
|
guest = guest_crud.remove(db, guest_id=str(guest_id))
|
|
return guest
|
|
|
|
|
|
@router.patch("/{guest_id}/status", response_model=GuestRead, operation_id='set_guest_status')
|
|
def set_guest_status(guest_id: uuid.UUID, status: GuestStatus, db: Session = Depends(get_db)):
|
|
guest = guest_crud.update_status(db, guest_id=guest_id, status=status)
|
|
if not guest:
|
|
raise HTTPException(status_code=404, detail="Guest not found")
|
|
return guest
|
|
|
|
@router.post("/{guest_id}/rsvp", response_model=GuestWithRSVPResponse, operation_id='create_guest_rsvp')
|
|
def process_guest_rsvp(
|
|
guest_id: uuid.UUID,
|
|
rsvp_data: RSVPSchemaCreate,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
# Start a transaction
|
|
try:
|
|
# Update or create RSVP
|
|
existing_rsvp = crud_rsvp.get_rsvp_by_event_and_guest(
|
|
db, event_id=rsvp_data.event_id, guest_id=guest_id
|
|
)
|
|
|
|
if existing_rsvp:
|
|
rsvp = crud_rsvp.update(db, db_obj=existing_rsvp, obj_in=rsvp_data)
|
|
else:
|
|
rsvp_data.guest_id = guest_id # Ensure guest_id matches URL
|
|
rsvp = crud_rsvp.create(db, obj_in=rsvp_data)
|
|
|
|
# Update guest status to match RSVP
|
|
guest_status = map_rsvp_status_to_guest_status(rsvp_data.status)
|
|
additional_guests = max(0, rsvp_data.number_of_guests - 1)
|
|
guest = guest_crud.update_status(db, guest_id=guest_id, status=guest_status, number_of_guests=additional_guests)
|
|
|
|
# Both operations succeeded, commit the transaction
|
|
db.commit()
|
|
|
|
# Return combined response
|
|
return {
|
|
"guest": guest,
|
|
"rsvp": rsvp
|
|
}
|
|
|
|
except Exception as e:
|
|
db.rollback()
|
|
raise HTTPException(status_code=500, detail=f"Failed to process RSVP: {str(e)}") |