oi:ddlZddlmZddlmZGddeZy)N)Dict)WebSocketHandlercNeZdZdZiZiZdeddfdZd dZe ddde ddfd Z y) RequestInfoSocketHandlera]A websocket handler used to provide the request info associated with kernel ids in preheat kernel mode. Class variables --------------- - _waiters : A dictionary which holds the `websocket` connection associated with the kernel id. - cache : A dictionary which holds the request info associated with the kernel id. kernel_idreturnNc|tj|<||jvr|j|j|yy)zCreate a new websocket connection, this connection is identified by the kernel id. Args: kernel_id (str): Kernel id used by the notebook when it opens the websocket connection. N)r_waiters_cache write_message)selfrs b/home/ubuntu/docker-apps/notebooks/venv/lib/python3.12/site-packages/voila/request_info_handler.pyopenzRequestInfoSocketHandler.opens=8< )))4  #   t{{95 6 $ctjjD] \}}||k(s ntj=y)N)rr items)r k_idwaiters ron_closez!RequestInfoSocketHandler.on_close#s>4==CCE LD&~  % - -d 3rclsmsgc|d}|d}|jj|d}| |j|y||j |<y#t$rt j ddYywxYw)a*Class method used to dispatch the request info to the waiting notebook. This method is called in `VoilaHandler` when the request info becomes available. If this method is called before the opening of websocket connection, `msg` is stored in `_cache` and the message will be dispatched when a notebook with corresponding kernel id is connected. Args: - msg (Dict): this dictionary contains the `kernel_id` to identify the waiting notebook and `payload` is the request info. rpayloadNzError sending messageT)exc_info)r getr Exceptionloggingerrorr )rrrrrs r send_updatesz%RequestInfoSocketHandler.send_updates)sy $ i.!!)T2   F$$W-%,CJJy ! F 5E FsA A/.A/)rN) __name__ __module__ __qualname____doc__r r strrr classmethodrrrrrrsU H F 7c 7d 74 ,4,4,D,,rr)rtypingrtornado.websocketrrr&rrr)s.8,/8,r