Web socket connection manager.
| 27 | |
| 28 | |
| 29 | class ConnectionManager: |
| 30 | """Web socket connection manager.""" |
| 31 | |
| 32 | def __init__(self): |
| 33 | self.active_connections: List[WebSocket] = [] |
| 34 | |
| 35 | async def connect(self, websocket: WebSocket): |
| 36 | await websocket.accept() |
| 37 | self.active_connections.append(websocket) |
| 38 | |
| 39 | def disconnect(self, websocket: WebSocket): |
| 40 | self.active_connections.remove(websocket) |
| 41 | |
| 42 | async def send_message(self, message: str, websocket: WebSocket): |
| 43 | await websocket.send_text(message) |
| 44 | |
| 45 | async def broadcast(self, message: str): |
| 46 | for connection in self.active_connections: |
| 47 | await connection.send_text(message) |
| 48 | |
| 49 | |
| 50 | conn_mgr = ConnectionManager() |