easier middleware installation

Author: krylosov-aa

README.md

@@ -179,12 +179,12 @@ async def fastapi_db_session_middleware(
 
 You can use ready-made FastAPI middleware:
 ```python
-from context_async_sqlalchemy import fastapi_db_session_middleware
-from starlette.middleware.base import BaseHTTPMiddleware
+from fastapi import FastAPI
+from context_async_sqlalchemy import add_fastapi_db_session_middleware
 
-app.add_middleware(
-    BaseHTTPMiddleware, dispatch=fastapi_db_session_middleware
-)
+app = FastAPI()
+
+add_fastapi_db_session_middleware(app)
 ```
 
 

context_async_sqlalchemy/__init__.py

@@ -23,7 +23,10 @@
     rollback_all_sessions,
     close_all_sessions,
 )
-from .fastapi_utils.middleware import fastapi_db_session_middleware
+from .fastapi_utils import (
+    fastapi_db_session_middleware,
+    add_fastapi_db_session_middleware,
+)
 
 __all__ = [
     "init_db_session_ctx",
@@ -46,4 +49,5 @@
     "rollback_all_sessions",
     "close_all_sessions",
     "fastapi_db_session_middleware",
+    "add_fastapi_db_session_middleware",
 ]

context_async_sqlalchemy/fastapi_utils/__init__.py

@@ -0,0 +1,9 @@
+from .middleware import (
+    fastapi_db_session_middleware,
+    add_fastapi_db_session_middleware,
+)
+
+__all__ = [
+    "fastapi_db_session_middleware",
+    "add_fastapi_db_session_middleware",
+]

context_async_sqlalchemy/fastapi_utils/middleware.py

@@ -1,7 +1,8 @@
-from fastapi import Request
+from fastapi import FastAPI, Request
 from starlette.middleware.base import (  # type: ignore[attr-defined]
     Response,
     RequestResponseEndpoint,
+    BaseHTTPMiddleware,
 )
 
 from context_async_sqlalchemy import (
@@ -13,6 +14,13 @@
 )
 
 
+def add_fastapi_db_session_middleware(app: FastAPI) -> None:
+    """Adds middleware to the application"""
+    app.add_middleware(
+        BaseHTTPMiddleware, dispatch=fastapi_db_session_middleware
+    )
+
+
 async def fastapi_db_session_middleware(
     request: Request, call_next: RequestResponseEndpoint
 ) -> Response:

exmaples/fastapi_example/setup_app.py

@@ -3,9 +3,8 @@
 from contextlib import asynccontextmanager
 from typing import Any, AsyncGenerator
 from fastapi import FastAPI
-from starlette.middleware.base import BaseHTTPMiddleware
 
-from context_async_sqlalchemy import fastapi_db_session_middleware
+from context_async_sqlalchemy import add_fastapi_db_session_middleware
 
 from .database import master
 from .routes.atomic_usage import handler_with_db_session_and_atomic
@@ -29,7 +28,7 @@ def setup_app() -> FastAPI:
     app = FastAPI(
         lifespan=lifespan,
     )
-    setup_middlewares(app)
+    add_fastapi_db_session_middleware(app)
     setup_routes(app)
     return app
 
@@ -41,16 +40,6 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, Any]:
     await master.close()  # Close the engine if it was open
 
 
-def setup_middlewares(app: FastAPI) -> None:
-    """
-    The middleware will be responsible for initializing the context.
-    And will also be responsible for autocommit or autorollback.
-    """
-    app.add_middleware(
-        BaseHTTPMiddleware, dispatch=fastapi_db_session_middleware
-    )
-
-
 def setup_routes(app: FastAPI) -> None:
     """
     It's just a single point where I collected all the APIs.

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "context-async-sqlalchemy"
-version = "1.2.2"
+version = "1.2.3"
 description = "A convenient way to configure and interact with a async sqlalchemy session through context in asynchronous applications"
 readme = "README.md"
 authors = [