From 37adbf1437471a48d743e6f8c8b84c60bea557f2 Mon Sep 17 00:00:00 2001 From: Joe Banks Date: Sun, 25 Oct 2020 19:50:14 +0000 Subject: Add database middleware --- backend/middleware.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 backend/middleware.py (limited to 'backend') diff --git a/backend/middleware.py b/backend/middleware.py new file mode 100644 index 0000000..c1aa731 --- /dev/null +++ b/backend/middleware.py @@ -0,0 +1,17 @@ +from starlette.middleware.base import BaseHTTPMiddleware +import pymongo +import ssl + +from backend.constants import DATABASE_URL, MONGO_DATABASE + + +class DatabaseMiddleware(BaseHTTPMiddleware): + async def dispatch(self, request, call_next): + client = pymongo.MongoClient( + DATABASE_URL, + ssl_cert_reqs=ssl.CERT_NONE + ) + db = client[MONGO_DATABASE] + request.state.db = db + response = await call_next(request) + return response -- cgit v1.2.3