:vj ndZddlmZddlmZddlmZddlmZddl m Z ddl m Z dZ Gd d e ee Zy ) z/Tornado handlers for WebSocket <-> ZMQ sockets.) ensure_async)web)WebSocketHandler)ws_authenticated)JupyterHandler)WebSocketMixinkernelscneZdZdZeZedZdZdZ dZ e fdZ fdZ dZd Zd ZxZS) KernelWebsocketHandlerz$The kernels websocket should connectc8|jjdS)z&The kernel websocket connection class.!kernel_websocket_connection_classsettingsgetselfs z/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_server/services/kernels/websocket.pyr z8KernelWebsocketHandler.kernel_websocket_connection_classs}}  !DEEcy)zhUndo the set_default_headers in JupyterHandler which doesn't make sense for websockets Nrs rset_default_headersz*KernelWebsocketHandler.set_default_headerssrc:|jjddS)z"Get the socket connection options.websocket_compression_optionsNrrs rget_compression_optionsz.KernelWebsocketHandler.get_compression_options s}}  !@$GGrcjK|j}t|jj||ddd{}|st j d|j j|j}|j|||j|_ |jddr+|jd|jj_ n|jjdt!|jdr#|jj#d{yy77w) zHandle a pre_get.executer Ni)parentwebsocket_handlerconfig session_idzNo session ID specifiedprepare) current_userr authorizer is_authorizedr HTTPErrorkernel_manager get_kernel kernel_idr r connection get_argumentsessionlogwarninghasattrr!)ruser authorizedkernels rpre_getzKernelWebsocketHandler.pre_get$s  ( OO ) )$i K  --$ $$$//?@@T$++A    \4 0.2.?.? .MDOO # # + HH  6 7 4??I .//))+ + + /# $ ,s"7D3D/C.D3(D1)D31D3cK||_|jd{t| |d{y77w)z"Handle a get request for a kernel.N)r()r(r2superrrr( __class__s rrzKernelWebsocketHandler.get=s<#llngkIk... .sA<A>AAcKt||jjd|jd|j j d{y7w)zOpen a kernel websocket.zConnecting to kernel .N)r4openr,infor(r)connectr5s rr9zKernelWebsocketHandler.openDsG    -dnn-=Q?@oo%%'''sAA AA c:|jj|y)zGGet a kernel message from the websocket and turn it into a ZMQ message.N)r)handle_incoming_message)r ws_messages r on_messagez!KernelWebsocketHandler.on_messageMs // ;rcF|jjd|_y)zHandle a socket closure.N)r) disconnectrs ron_closezKernelWebsocketHandler.on_closeQs ""$rc^|jj}|d}n|dk(rd}||vr|}|Sd}|S)z'Select the sub protocol for the socket.Nzv1.kernel.websocket.jupyter.org)r)kernel_ws_protocol)r subprotocolspreferred_protocolselected_subprotocols rselect_subprotocolz)KernelWebsocketHandler.select_subprotocolVsP!__??  %!B  2 %!% 5G<5W1##^b##r)__name__ __module__ __qualname____doc__ AUTH_RESOURCE auth_resourcepropertyr rrr2rrr9r?rBrI __classcell__)r6s@rr r sW.!M FF H,2// (< $rr N)rMjupyter_core.utilsrtornadortornado.websocketrjupyter_server.auth.decoratorrjupyter_server.base.handlersrjupyter_server.base.websocketrrNr rrrrXs45,.:78 O$^-=~O$r