mirror of
https://github.com/Portabase/portabase.git
synced 2026-07-14 11:16:13 +02:00
21 lines
578 B
Python
21 lines
578 B
Python
|
from fastapi import Request, Response
|
||
|
|
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
|