Update app.py

app.py

@@ -34,7 +34,12 @@ from logger import log_error, log_info, log_warning
 
 # Exception imports
 from exceptions import (
-    FlareException, RaceConditionError, format_error_response, 
+    DuplicateResourceError, 
+    ValidationError, 
+    ResourceNotFoundError,    
+    FlareException, 
+    RaceConditionError, 
+    format_error_response, 
     get_http_status_code
 )
 from dotenv import load_dotenv
@@ -189,37 +194,49 @@ async def global_exception_handler(request: Request, exc: Exception):
         })
     )
 
-# Validation error handler
+@app.exception_handler(DuplicateResourceError)
+async def duplicate_resource_handler(request: Request, exc: DuplicateResourceError):
+    """Handle duplicate resource errors"""
+    return JSONResponse(
+        status_code=409,
+        content={
+            "detail": str(exc),
+            "error_type": "duplicate_resource",
+            "resource_type": exc.details.get("resource_type"),
+            "identifier": exc.details.get("identifier")
+        }
+    )
+
+@app.exception_handler(RaceConditionError)
+async def race_condition_handler(request: Request, exc: RaceConditionError):
+    """Handle race condition errors"""
+    return JSONResponse(
+        status_code=409,
+        content=exc.to_http_detail()
+    )
 
 @app.exception_handler(ValidationError)
-async def validation_exception_handler(request: Request, exc: ValidationError):
-    """Handle Pydantic validation errors"""
-    request_id = getattr(request.state, 'request_id', 'unknown')
-    
-    errors = []
-    for error in exc.errors():
-        field = " -> ".join(str(x) for x in error['loc'])
-        errors.append({
-            'field': field,
-            'message': error['msg'],
-            'type': error['type'],
-            'input': error.get('input')
-        })
-    
-    log_warning(
-        "Validation error",
-        request_id=request_id,
-        errors=errors
+async def validation_error_handler(request: Request, exc: ValidationError):
+    """Handle validation errors"""
+    return JSONResponse(
+        status_code=422,
+        content={
+            "detail": str(exc),
+            "error_type": "validation_error",
+            "details": exc.details
+        }
     )
-    
+
+@app.exception_handler(ResourceNotFoundError)
+async def resource_not_found_handler(request: Request, exc: ResourceNotFoundError):
+    """Handle resource not found errors"""
     return JSONResponse(
-        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
+        status_code=404,
         content={
-            "error": "ValidationError",
-            "message": "Invalid request data. Please check the fields and try again.",
-            "details": errors,
-            "request_id": request_id,
-            "timestamp": datetime.utcnow().isoformat()
+            "detail": str(exc),
+            "error_type": "resource_not_found",
+            "resource_type": exc.details.get("resource_type"),
+            "identifier": exc.details.get("identifier")
         }
     )