From 3b5d3e4424c23c8ab27e2c469e3d10af860bbf2e Mon Sep 17 00:00:00 2001 From: Chris Lovering Date: Mon, 8 Jul 2024 21:35:06 +0100 Subject: Update middleware to use SQLA to create db sessions --- backend/middleware.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'backend/middleware.py') diff --git a/backend/middleware.py b/backend/middleware.py index 0b08859..5b36473 100644 --- a/backend/middleware.py +++ b/backend/middleware.py @@ -3,7 +3,7 @@ from starlette.requests import Request from starlette.responses import JSONResponse from starlette.types import ASGIApp, Receive, Scope, Send -from backend.constants import DATABASE_URL, DOCS_PASSWORD, MONGO_DATABASE +from backend.constants import DB_SESSION_MAKER, DOCS_PASSWORD, MONGO_DATABASE, MONGO_DATABASE_URL class DatabaseMiddleware: @@ -12,12 +12,14 @@ class DatabaseMiddleware: async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None: client: AsyncIOMotorClient = AsyncIOMotorClient( - DATABASE_URL, + MONGO_DATABASE_URL, tlsAllowInvalidCertificates=True, ) db = client[MONGO_DATABASE] Request(scope).state.db = db - await self._app(scope, receive, send) + async with DB_SESSION_MAKER() as session, session.begin(): + Request(scope).state.psql_db = session + await self._app(scope, receive, send) class ProtectedDocsMiddleware: -- cgit v1.2.3