o "i@s4ddlmZmZddlmZGdddZeZdS))DictList) WebSocketc@sJeZdZddZdedefddZdedefddZded efd d Z d S) WebSocketManagercCs i|_dSN) connections)selfr L/lsinfo/ai/hellotax_ai/training_center/backend/app/core/websocket_manager.py__init__s zWebSocketManager.__init__task_id websocketcs8|IdH||jvrg|j|<|j||dSr)acceptrappendrr r r r r connect s   zWebSocketManager.connectcs:||jvr|j|||j|s|j|=dSdSdSr)rremoverr r r disconnects   zWebSocketManager.disconnectmessagecsB||jvr|j|D]}z ||IdHWq Yq dSdSr)r send_json)rr r connectionr r r broadcasts zWebSocketManager.broadcastN) __name__ __module__ __qualname__r strrrrdictrr r r r rs rN)typingrrfastapirrmanagerr r r r s