Working broadcasting library

pkgs/clan-cli/clan_cli/webui/app.py

@@ -4,11 +4,13 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.routing import APIRoute
 from fastapi.staticfiles import StaticFiles
+from contextlib import asynccontextmanager
+from typing import Any
 
 from ..errors import ClanError
 from .assets import asset_path
 from .error_handlers import clan_error_handler
-from .routers import health, root, sql_connect, socket_manager # sql router hinzufügen
+from .routers import health, root, sql_connect, socket_manager2 # sql router hinzufügen
 
 #import for sql
 from fastapi import Depends, FastAPI, HTTPException
@@ -24,12 +26,19 @@ origins = [
 log = logging.getLogger(__name__)
 
 
+@asynccontextmanager
+async def lifespan(app: FastAPI) -> Any:
+    await socket_manager2.brd.connect()
+    yield
+    await socket_manager2.brd.disconnect()
+
+
 def setup_app() -> FastAPI:
     # bind sql engine
     sql_models.Base.metadata.drop_all(engine)
     sql_models.Base.metadata.create_all(bind=engine)
-    
-    app = FastAPI()
+
+    app = FastAPI(lifespan=lifespan)
     app.add_middleware(
         CORSMiddleware,
         allow_origins=origins,
@@ -43,6 +52,7 @@ def setup_app() -> FastAPI:
     app.include_router(sql_connect.router)
 
     app.include_router(socket_manager.router)
+    app.include_router(socket_manager2.router)
 
     # Needs to be last in register. Because of wildcard route
     app.include_router(root.router)