fix: removes references to session_context

session_context is meant to be able to provide us a context which
commit the session and closes the connection, this is not understood and
documented for the async world, until such point that we do  understand this
properly, this should not be used

Author: devraj

src/labs/db.py

@@ -43,15 +43,15 @@ async def init_models():
 
 # TODO: Revise implementation in async context
 # Donot use until we have a better understanding of how to use async context
-@asynccontextmanager
-async def session_context():
-    """Provide a transactional scope around a series of operations.
-    """
-    try:
-        yield async_session
-        await async_session.commit()
-    except:  # noqa: E722
-        await async_session.rollback()
-        raise
-    finally:
-        await async_session.close()
+# @asynccontextmanager
+# async def session_context():
+#     """Provide a transactional scope around a series of operations.
+#     """
+#     try:
+#         yield async_session
+#         await async_session.commit()
+#     except:  # noqa: E722
+#         await async_session.rollback()
+#         raise
+#     finally:
+#         await async_session.close()

src/labs/routers/auth/__init__.py

@@ -11,7 +11,7 @@
 from sqlalchemy.ext.asyncio import AsyncSession
 
 
-from ...db import session_context, get_async_session
+from ...db import get_async_session
 from ...models import User
 from ...schema import UserRequest,\
   PasswordLoginRequest, AuthResponse
@@ -56,7 +56,7 @@ async def login_user(request: PasswordLoginRequest,
   operation_id="refresh_token",
 )
 async def refresh_jwt_token(request: Request,
-  session: AsyncSession = Depends(session_context)):
+  session: AsyncSession = Depends(get_async_session)):
   """ Provides a refresh token for the JWT session.
   """
   return {}
@@ -66,15 +66,15 @@ async def refresh_jwt_token(request: Request,
   summary=""" Provides an endpoint for logging out the user""",
   operation_id="logout_user"
 )
-async def logout_user(session: AsyncSession = Depends(session_context)):
+async def logout_user(session: AsyncSession = Depends(get_async_session)):
   """ Ends a users session
   
   """
   return {}
 
 @router.get("/me", response_model=UserRequest)
 async def get_me(request: Request,
-  session: AsyncSession = Depends(session_context)
+  session: AsyncSession = Depends(get_async_session)
 ):
   """Get the currently logged in user or myself
 

src/labs/routers/ext/__init__.py

@@ -4,7 +4,7 @@
 from fastapi import APIRouter, Request, Depends
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from ...db import session_context, session_context
+from ...db import get_async_session
 from ...config import config
 
 router = APIRouter(tags=["ext"])
@@ -25,13 +25,12 @@ async def get_health(request: Request):
     Purpose of this endpoint is to check the health of the server.
     We check for connection to the database, queue and logger
     """
-    async with session_context() as session:
-        return {"message": "ok"}
+    return {"message": "ok"}
 
 
 @router.get("/log")
 async def test_logger(request: Request,
-    session: AsyncSession = Depends(session_context)
+    session: AsyncSession = Depends(get_async_session)
 ):
     """Log a message.