vayuapi 0.2.3__tar.gz → 0.2.4__tar.gz

{vayuapi-0.2.3 → vayuapi-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vayuapi
-Version: 0.2.3
+Version: 0.2.4
 Summary: The fastest Python async API framework for rapid development with AI/ML support
 Home-page: https://github.com/vayuapi/vayuapi
 Author: VayuAPI Team

{vayuapi-0.2.3 → vayuapi-0.2.4}/pyproject.toml

@@ -6,7 +6,7 @@ include = ["vayuapi*"]
 
 [project]
 name = "vayuapi"
-version = "0.2.3"
+version = "0.2.4"
 description = "The fastest Python async API framework for rapid development with AI/ML support"
 readme = "README.md"
 requires-python = ">=3.11,<3.16"

{vayuapi-0.2.3 → vayuapi-0.2.4}/setup.py

@@ -7,7 +7,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="vayuapi",
-    version="0.2.3",
+    version="0.2.4",
     author="VayuAPI Team",
     author_email="codeswithalok@gmail.com",
     description="The fastest Python async API framework for rapid development with AI/ML support",

{vayuapi-0.2.3 → vayuapi-0.2.4}/vayuapi/__init__.py

@@ -58,6 +58,7 @@ from vayuapi.core.concurrency import (
 
 # Exception handling
 from vayuapi.core.exceptions import (
+    ExceptionHandler,
     HTTPException,
     RequestValidationError,
     WebSocketException,

{vayuapi-0.2.3 → vayuapi-0.2.4}/vayuapi/core/application.py

@@ -61,6 +61,7 @@ class VayuAPI:
         allowed_origins: List[str] = None,
         middleware: List[Middleware] = None,
         lifespan: Optional[Callable] = None,
+        exception_handlers: Optional[Dict[Type[Exception], Callable]] = None,
     ):
         """
         Initialize VayuAPI application.
@@ -81,6 +82,7 @@ class VayuAPI:
             allowed_origins: List of allowed origins for CORS
             middleware: List of middleware to add
             lifespan: Lifespan context manager
+            exception_handlers: Dict mapping exception classes to handler callables
         """
         self.title = title
         self.version = version
@@ -154,7 +156,10 @@ class VayuAPI:
 
         # Exception handlers
         from vayuapi.core.exceptions import ExceptionHandler
-        self.exception_handler = ExceptionHandler()
+        self._exception_handler = ExceptionHandler()
+        if exception_handlers:
+            for exc_class, handler in exception_handlers.items():
+                self._exception_handler.add_handler(exc_class, handler)
 #---------------------Routes------------------------------Added-------#
     @property
     def routes(self):
@@ -343,7 +348,27 @@ class VayuAPI:
             exc_class: Exception class to handle
             handler: Async function that handles the exception
         """
-        self.exception_handler.add_handler(exc_class, handler)
+        self._exception_handler.add_handler(exc_class, handler)
+
+    def exception_handler(self, exc_class: Type[Exception]):
+        """
+        Decorator to register a custom exception handler.
+
+        Example:
+            ```python
+            @app.exception_handler(HTTPException)
+            async def http_error_handler(request, exc):
+                return JSONResponse(status_code=exc.status_code, content={"error": exc.detail})
+
+            @app.exception_handler(RedirectToLogin)
+            async def redirect_handler(request, exc):
+                return RedirectResponse(url="/login")
+            ```
+        """
+        def decorator(handler: Callable):
+            self._exception_handler.add_handler(exc_class, handler)
+            return handler
+        return decorator
 
     def route(
         self,
@@ -649,7 +674,7 @@ class VayuAPI:
                             return JSONResponse({"message": "Success"})
                         return JSONResponse({"data": result})
                     except Exception as e:
-                        return await self.exception_handler.handle(request, e)
+                        return await self._exception_handler.handle(request, e)
                 return simple_wrapped
             else:
                 async def simple_wrapped_sync(request: Request) -> Response:
@@ -668,7 +693,7 @@ class VayuAPI:
                             return JSONResponse({"message": "Success"})
                         return JSONResponse({"data": result})
                     except Exception as e:
-                        return await self.exception_handler.handle(request, e)
+                        return await self._exception_handler.handle(request, e)
                 return simple_wrapped_sync
 
 #------------------------------------------------------------------------------------------------------------#
@@ -867,10 +892,10 @@ class VayuAPI:
             except ValidationError as e:
                 from vayuapi.core.exceptions import RequestValidationError
                 validation_error = RequestValidationError(e.errors())
-                return await self.exception_handler.handle(request, validation_error)
+                return await self._exception_handler.handle(request, validation_error)
             except Exception as e:
                 # Use exception handler
-                return await self.exception_handler.handle(request, e)
+                return await self._exception_handler.handle(request, e)
 
         return wrapped
 
@@ -952,7 +977,7 @@ class VayuAPI:
             from starlette.exceptions import HTTPException as StarletteHTTPException
 
             mw_stack = self.middleware_stack  # capture reference
-            exc_handler = self.exception_handler  # capture reference
+            exc_handler = self._exception_handler  # capture reference
 
             class _VayuMiddlewareWrapper(BaseHTTPMiddleware):
                 async def dispatch(self, request, call_next):