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