refactored org code

Author: nishika26

backend/app/api/routes/organization.py

@@ -17,11 +17,12 @@
 )
 from app.crud.organization import create_organization, get_organization_by_id
 from app.utils import APIResponse
+from app.core.exception_handlers import NotFoundException
+
 
 router = APIRouter(prefix="/organizations", tags=["organizations"])
 
 
-# Retrieve organizations
 @router.get(
     "/",
     dependencies=[Depends(get_current_active_superuser)],
@@ -37,7 +38,6 @@ def read_organizations(session: SessionDep, skip: int = 0, limit: int = 100):
     return APIResponse.success_response(organizations)
 
 
-# Create a new organization
 @router.post(
     "/",
     dependencies=[Depends(get_current_active_superuser)],
@@ -54,16 +54,12 @@ def create_new_organization(*, session: SessionDep, org_in: OrganizationCreate):
     response_model=APIResponse[OrganizationPublic],
 )
 def read_organization(*, session: SessionDep, org_id: int):
-    """
-    Retrieve an organization by ID.
-    """
     org = get_organization_by_id(session=session, org_id=org_id)
     if org is None:
-        raise HTTPException(status_code=404, detail="Organization not found")
+        raise NotFoundException("Organization not found")
     return APIResponse.success_response(org)
 
 
-# Update an organization
 @router.patch(
     "/{org_id}",
     dependencies=[Depends(get_current_active_superuser)],
@@ -74,7 +70,7 @@ def update_organization(
 ):
     org = get_organization_by_id(session=session, org_id=org_id)
     if org is None:
-        raise HTTPException(status_code=404, detail="Organization not found")
+        raise NotFoundException("Organization not found")
 
     org_data = org_in.model_dump(exclude_unset=True)
     org = org.model_copy(update=org_data)
@@ -86,7 +82,6 @@ def update_organization(
     return APIResponse.success_response(org)
 
 
-# Delete an organization
 @router.delete(
     "/{org_id}",
     dependencies=[Depends(get_current_active_superuser)],
@@ -96,7 +91,7 @@ def update_organization(
 def delete_organization(session: SessionDep, org_id: int):
     org = get_organization_by_id(session=session, org_id=org_id)
     if org is None:
-        raise HTTPException(status_code=404, detail="Organization not found")
+        raise NotFoundException("Organization not found")
 
     session.delete(org)
     session.commit()