vayuapi 0.2.0__tar.gz → 0.2.2__tar.gz

{vayuapi-0.2.0 → vayuapi-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vayuapi
-Version: 0.2.0
+Version: 0.2.2
 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.0 → vayuapi-0.2.2}/pyproject.toml

@@ -6,7 +6,7 @@ include = ["vayuapi*"]
 
 [project]
 name = "vayuapi"
-version = "0.2.0"
+version = "0.2.2"
 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.0 → vayuapi-0.2.2}/setup.py

@@ -7,7 +7,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="vayuapi",
-    version="0.2.0",
+    version="0.2.2",
     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.0 → vayuapi-0.2.2}/vayuapi/core/application.py

@@ -100,7 +100,7 @@ class VayuAPI:
         # Add default middleware
         if cors_enabled:
             from starlette.middleware.cors import CORSMiddleware
-            self.add_starlette_middleware(
+            self.add_middleware(
                 CORSMiddleware,
                 allow_origins=allowed_origins or ["*"],
                 allow_credentials=True,
@@ -120,7 +120,7 @@ class VayuAPI:
             # Add session middleware for admin authentication
             from starlette.middleware.sessions import SessionMiddleware
             import secrets
-            self.add_starlette_middleware(
+            self.add_middleware(
                 SessionMiddleware,
                 secret_key=secrets.token_urlsafe(32)
             )
@@ -237,16 +237,79 @@ class VayuAPI:
             return func
         return decorator
 
-    def add_middleware(self, middleware: Middleware):
-        """Add custom middleware to the stack."""
-        self.middleware_stack.add(middleware)
+    def add_middleware(self, middleware_class: Type, **options) -> None:
+        """
+        Add middleware to the application.
 
-    def add_starlette_middleware(self, middleware_class: Type, **options):
-        """Add Starlette middleware."""
+        Identical API to FastAPI / Starlette — always pass a **class** and
+        optional keyword arguments.  VayuAPI supports three kinds:
+
+        1. **ASGI / Starlette middleware** (``CORSMiddleware``,
+           ``GZipMiddleware``, ``SessionMiddleware``, …) — any class whose
+           constructor accepts ``(app, **options)``.
+
+        2. **BaseHTTPMiddleware subclasses** — override ``dispatch(request,
+           call_next)``; pass the class without instantiating it.
+
+        3. **VayuAPI Middleware subclasses** — override
+           ``process_request`` / ``process_response``; pass the class without
+           instantiating it (or pass an instance for backwards compatibility).
+
+        Examples::
+
+            from vayuapi.middleware import CORSMiddleware, GZipMiddleware
+            from vayuapi.middleware import SessionMiddleware, HTTPSRedirectMiddleware
+
+            app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"])
+            app.add_middleware(GZipMiddleware, minimum_size=1000)
+            app.add_middleware(SessionMiddleware, secret_key="supersecret")
+            app.add_middleware(HTTPSRedirectMiddleware)
+
+            # BaseHTTPMiddleware subclass
+            from vayuapi.middleware import BaseHTTPMiddleware
+            class TimingMiddleware(BaseHTTPMiddleware):
+                async def dispatch(self, request, call_next):
+                    import time
+                    start = time.time()
+                    response = await call_next(request)
+                    response.headers["X-Process-Time"] = str(time.time() - start)
+                    return response
+            app.add_middleware(TimingMiddleware)
+
+            # VayuAPI custom Middleware subclass
+            from vayuapi.middleware import Middleware
+            class LogMiddleware(Middleware):
+                async def process_request(self, request):
+                    print(request.method, request.url)
+            app.add_middleware(LogMiddleware)
+        """
+        from starlette.middleware.base import BaseHTTPMiddleware as _StarletteBaseHTTPMiddleware
+
+        # Backward-compat: accept a Middleware *instance* as well as a class
+        if isinstance(middleware_class, Middleware):
+            self.middleware_stack.add(middleware_class)
+            return
+
+        if not isinstance(middleware_class, type):
+            raise TypeError(
+                "add_middleware expects a middleware class, "
+                f"got {type(middleware_class)!r}"
+            )
+
+        # VayuAPI Middleware subclass (process_request / process_response style)
+        if issubclass(middleware_class, Middleware):
+            self.middleware_stack.add(middleware_class(**options))
+            return
+
+        # BaseHTTPMiddleware subclass or any ASGI middleware class
         self._starlette_middleware.append(
             StarletteMiddleware(middleware_class, **options)
         )
 
+    def add_starlette_middleware(self, middleware_class: Type, **options):
+        """Alias for add_middleware with a middleware class. Kept for backward compatibility."""
+        self.add_middleware(middleware_class, **options)
+
     def add_exception_handler(self, exc_class: Type[Exception], handler: Callable):
         """
         Add a custom exception handler.

{vayuapi-0.2.0 → vayuapi-0.2.2}/vayuapi/core/middleware.py

@@ -203,7 +203,7 @@ class CORSMiddleware:
         ```python
         from vayuapi.middleware import CORSMiddleware
 
-        app.add_starlette_middleware(
+        app.add_middleware(
             CORSMiddleware,
             allow_origins=["https://example.com"],
             allow_credentials=True,
@@ -260,7 +260,7 @@ class GZipMiddleware:
         ```python
         from vayuapi.middleware import GZipMiddleware
 
-        app.add_starlette_middleware(
+        app.add_middleware(
             GZipMiddleware,
             minimum_size=1000,
             compresslevel=6,
@@ -298,7 +298,7 @@ class HTTPSRedirectMiddleware:
         ```python
         from vayuapi.middleware import HTTPSRedirectMiddleware
 
-        app.add_starlette_middleware(HTTPSRedirectMiddleware)
+        app.add_middleware(HTTPSRedirectMiddleware)
         ```
 
     Automatically redirects HTTP requests to HTTPS.
@@ -321,7 +321,7 @@ class TrustedHostMiddleware:
         ```python
         from vayuapi.middleware import TrustedHostMiddleware
 
-        app.add_starlette_middleware(
+        app.add_middleware(
             TrustedHostMiddleware,
             allowed_hosts=["example.com", "*.example.com"],
             www_redirect=True,
@@ -359,7 +359,7 @@ class SessionMiddleware:
         ```python
         from vayuapi.middleware import SessionMiddleware
 
-        app.add_starlette_middleware(
+        app.add_middleware(
             SessionMiddleware,
             secret_key="your-secret-key",
             session_cookie="session",