unregistering from events

Author: danvesga

app/api/endpoints/registration.py

@@ -21,7 +21,9 @@ async def get_volunteers_by_event(event_id: str) -> list[Registration]:
 @router.get("/events/{volunteer_id}", response_model=list[Event])
 async def get_events_by_volunteer(
     volunteer_id: str,
-    status: Annotated[RegistrationStatus | None, Query(description="Filter by status")] = None,
+    registration_status: Annotated[
+        RegistrationStatus | None, Query(description="Filter by status")
+    ] = None,
     current_user: Annotated[User, Depends(get_current_user)] = None,
 ) -> list[Event]:
     volunteer = await volunteer_model.get_volunteer_by_id(volunteer_id)
@@ -37,7 +39,7 @@ async def get_events_by_volunteer(
     elif current_user.user_type != UserType.ADMIN:
         raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="Access denied")
 
-    return await registration_model.get_events_by_volunteer(volunteer_id, status)
+    return await registration_model.get_events_by_volunteer(volunteer_id, registration_status)
 
 
 @router.post("/new", response_model=Registration)
@@ -57,3 +59,23 @@ async def create_registrion(
         )
 
     return await registration_model.create_registration(registration, current_user.entity_id)
+
+
+@router.put("/unregister/{registration_id}", response_model=Registration)
+async def unregister_registration(
+    registration_id: str,
+    current_user: Annotated[User, Depends(get_current_user)],
+) -> Registration:
+    if current_user.user_type != UserType.VOLUNTEER:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="Only volunteers can unregister from events",
+        )
+
+    if current_user.entity_id is None:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="You must be associated with a volunteer profile to unregister from an event",
+        )
+
+    return await registration_model.unregister_registration(registration_id, current_user.entity_id)

app/models/registration.py

@@ -89,6 +89,29 @@ async def create_registration(
 
         return self._to_registration(inserted_doc)
 
+    async def unregister_registration(
+        self, registration_id: str, volunteer_id: str
+    ) -> Registration:
+        registration = await self.registrations.find_one({"_id": ObjectId(registration_id)})
+
+        if not registration:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND, detail="Registration not found"
+            )
+
+        if str(registration["volunteer_id"]) != volunteer_id:
+            raise HTTPException(
+                status_code=status.HTTP_403_FORBIDDEN,
+                detail="Not authorized to unregister from this event",
+            )
+
+        await self.registrations.update_one(
+            {"_id": ObjectId(registration_id)},
+            {"$set": {"registration_status": RegistrationStatus.UNREGISTERED}},
+        )
+        updated_doc = await self.registrations.find_one({"_id": ObjectId(registration_id)})
+        return self._to_registration(updated_doc)
+
     def _to_registration(self, doc) -> Registration:
         registration_data = doc.copy()
         registration_data["id"] = str(registration_data["_id"])