Files
portabase/backend/middleware/db_connection.py
T

27 lines
692 B
Python

from fastapi import Request, Response
from sqlalchemy.ext.asyncio import AsyncSession
from database import get_async_session
# from starlette.middleware.base import BaseHTTPMiddleware
# from database import SessionLocal
#
# class DatabaseSessionMiddleware(BaseHTTPMiddleware):
# async def dispatch(self, request: Request, call_next):
# response = Response("Internal server error", status_code=500)
# try:
# request.state.db = SessionLocal()
# response = await call_next(request)
# finally:
# request.state.db.close()
# return response
def get_db(request: Request):
print(request)
return request.state.db