def commit_transaction(response):
"""Commit any open transaction if the request was successful."""
if 200 <= response.status_code < 400:
Session.commit()
else:
Session.rollback()
return response
async def db_middleware(request: Request, call_next):
try:
response: Response = await call_next(request)
if 200 <= response.status_code < 400:
Session.commit()
else:
Session.rollback()
finally:
Session.remove()
return response