refactor: create async context manager to replace fastapi event deprecated. Pass context manager lifespan to fastapi instance instead and configure taskiq broker setup to lifespan context manager.

Author: Akin01

src/labs/api.py

@@ -10,10 +10,10 @@
   API endpoints are built and served using the FastAPI micro-framework.
 
 """
+from contextlib import asynccontextmanager
 from . import __title__, __version__
 
 from fastapi import FastAPI, Request, status, WebSocket
-from fastapi.responses import JSONResponse
 from fastapi.routing import APIRoute
 
 from .settings import settings
@@ -39,19 +39,33 @@ def generate_operation_id(route: APIRoute) -> str:
     return route.name
 
 
+@asynccontextmanager
+async def lifespan(_fastapi: FastAPI):
+    if broker.is_worker_process:
+        # TaskIQ configurartion so we can share FastAPI dependencies in tasks
+        await broker.startup()
+    
+    yield
+
+    if broker.is_worker_process:
+        # On shutdown, we need to shutdown the broker
+        await broker.shutdown()
+
+
 """A FastAPI application that serves handlers
 """
 app = FastAPI(
     title=__title__,
     version=__version__,
-    description=settings.api_router.__doc__,
+    description=str(settings.api_router.__doc__),
     docs_url=settings.api_router.path_docs,
     root_path=settings.api_router.path_root,
     terms_of_service=settings.api_router.terms_of_service,
     contact=settings.api_router.contact,
     license_info=settings.api_router.license_info,
     openapi_tags=settings.api_router.open_api_tags,
     generate_unique_id_function=generate_operation_id,
+    lifespan=lifespan
 )
 
 
@@ -67,23 +81,7 @@ async def websocket_endpoint(websocket: WebSocket):
 app.include_router(router_root)
 
 
-# TaskIQ configurartion so we can share FastAPI dependencies in tasks
-@app.on_event("startup")
-async def app_startup():
-    if not broker.is_worker_process:
-        await broker.startup()
-
-# On shutdown, we need to shutdown the broker
-
-
-@app.on_event("shutdown")
-async def app_shutdown():
-    if not broker.is_worker_process:
-        await broker.shutdown()
-
 # Default handler
-
-
 @app.get(
     "/",
     status_code=status.HTTP_200_OK,
@@ -93,5 +91,5 @@ async def root(request: Request) -> RootResponse:
     """
     return RootResponse(
         message="Welcome to the {} API".format(__name__),
-        root_path=request.scope.get("root_path")
+        root_path=str(request.scope.get("root_path"))
     )