o i_g@sBdZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m ZzddlmZWneyAdd lmZYnwdd lmZdd lmZmZmZd d lmZd dlmZmZGdddeZGdddeZGdddeZ GdddeZ!dZ"dZ#defde"efde"de#e fde"de!fgZ$dS)zTornado handlers for kernels. Preliminary documentation at https://github.com/ipython/ipython/wiki/IPEP-16%3A-Notebook-multi-directory-dashboard-and-URL-mapping#kernels-api N)dedent)genweb)Future)IOLoop)protocol_version) json_default) date_default) cast_unicode) maybe_future url_path_join url_escape) APIHandler)AuthenticatedZMQStreamHandlerdeserialize_binary_messagec@4eZdZejejddZejejddZdS)MainKernelHandlerccs.|j}t|V}|tj|tddSNdefault)kernel_managerr list_kernelsfinishjsondumpsr)selfkmkernelsr]/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/services/kernels/handlers.pyget szMainKernelHandler.getccs|j}|}|durd|ji}n|d|jt|j|ddV}t||V}t|jddt |}| d|| d| t j|tddS)Nname) kernel_nameapirLocationr)r get_json_bodydefault_kernel_name setdefaultr start_kernel kernel_modelr base_urlr set_header set_statusrrrr)rrmodel kernel_idlocationrrr post's  zMainKernelHandler.postN) __name__ __module__ __qualname__r authenticatedr coroutiner!r2rrrr r rc@r) KernelHandlerccs0|j}t||V}|tj|tddSr)rr r+rrrr)rr0rr/rrr r!=szKernelHandler.getccs.|j}t||V|d|dS)N)rr shutdown_kernelr.r)rr0rrrr deleteDs   zKernelHandler.deleteN) r3r4r5rr6rr7r!r<rrrr r9;r8r9c@s eZdZejejddZdS)KernelActionHandlerc cs|j}|dkrt||V|d|dkrUz t||VWntyB}z|jjddd|dWYd}~nd}~wwt||V}| t j |t d| dS) N interruptr:restartzException restarting kernelTexc_infoir)rr interrupt_kernelr.restart_kernel Exceptionlogerrorr+writerrrr)rr0actionrer/rrr r2Os   zKernelActionHandler.postN)r3r4r5rr6rr7r2rrrr r=Msr=cseZdZdZiZeddZeddZeddZedd Z d d Z d d Z ddZ ddZ ddZddZfddZejfddZejfddZejddZfddZd d!Zfd"d#Zfd$d%Zd&d'Zd(d)Zd*d+Zd,d-ZZS).ZMQChannelsHandlerz]There is one ZMQChannelsHandler per running kernel and it oversees all the sessions. cCs|jj}|jd|S)Nkernel_info_timeout)rrKsettingsr!)r km_defaultrrr rKmsz&ZMQChannelsHandler.kernel_info_timeoutcC|jddS)Niopub_msg_rate_limitrrLr!rrrr rOrz'ZMQChannelsHandler.iopub_msg_rate_limitcCrN)Niopub_data_rate_limitrrPrQrrr rSvrRz(ZMQChannelsHandler.iopub_data_rate_limitcCrN)Nrate_limit_windowg?rPrQrrr rTzrRz$ZMQChannelsHandler.rate_limit_windowcCs|jjdt|dddS)N(r0 uninitialized)) __class__r3getattrrQrrr __repr__~szZMQChannelsHandler.__repr__cCsH|j}|jj}dD]}t|d|}||j|d|j|<}||_q dS)N)iopubshellcontrolstdinconnect_)identity)rsessionbsessionrYr0channelschannel)rrr`rdmethstreamrrr create_streamsz ZMQChannelsHandler.create_streamcsD j j}t|ddkr jd jt}|d|S| |  j dttt gdfdd d fdd } | fd d } fd d } fdd}| ||t fddjdddt  j}| |S)aNudge the zmq connections with kernel_info_requests Returns a Future that will resolve when we have received a shell or control reply and at least one iopub message, ensuring that zmq subscriptions are established, sockets are fully connected, and kernel is responsive. Keeps retrying kernel_info_request until these are both received. execution_statebusyz!Nudge: not nudging busy kernel %sNr[cs$fD] }|s|dqdS)zXEnsure all futures are resolved which in turn triggers cleanup N)done set_resultf) info_future iopub_futurerr rs  z(ZMQChannelsHandler.nudge..finishcs:ssdSdS)zCommon cleanupN)remove_timeout stop_on_recvclosedcloserl)control_channel iopub_channelloop nudge_handle shell_channelrr cleanups  z)ZMQChannelsHandler.nudge..cleanupc:jdjsjdjddSdS)Nz$Nudge: shell info reply received: %sz!Nudge: resolving shell future: %srEdebugr0rjrkmsgrnrrr on_shell_reply z0ZMQChannelsHandler.nudge..on_shell_replycrz)Nz&Nudge: control info reply received: %sz#Nudge: resolving control future: %sr{r}rrr on_control_replyrz2ZMQChannelsHandler.nudge..on_control_replycsBjdjsjdjddSdS)NzNudge: IOPub received: %sz!Nudge: resolving iopub future: %s)rEr|r0rjrqrkr})rurorrr on_iopubs z*ZMQChannelsHandler.nudge..on_iopubcs |d7}jdusjrjdjdSjjvr.jdjdSr?jdjdSrPjdjdSs|ddkr^jjnjj}|d|djj d j d  d |dSdS) Nz)Nudge: cancelling on closed websocket: %sz'Nudge: cancelling on stopped kernel: %sz*Nudge: cancelling on closed zmq socket: %s rzNudge: attempt z on kernel kernel_info_requestg?) ws_connection is_closingrEr|r0rrrrjwarningrasend call_later)countrE) both_donertrrvnudgerwrrxrr rs< z'ZMQChannelsHandler.nudge..nudger)rN)r get_kernelr0rYrEr|rrk connect_shellconnect_controlrcrmultiadd_done_callbackon_recvrcurrentr with_timeouttimerK)rkernelrmryrrrfuturer) rrtrrnrurorvrrwrrxr rs6       * zZMQChannelsHandler.nudgecsj}|j}z|j}Wn2ty@jdjjdur)|j_j j j jdj|_YjSw| sKjd|fddjS)zsend a request for kernel_infozRequesting kernel info from %sNrz'Waiting for pending kernel_info requestcs|Sr)_finish_kernel_inforesultrlrQrr !sz8ZMQChannelsHandler.request_kernel_info..)rrr0_kernel_info_futureAttributeErrorrEr|kernel_info_channelrr_handle_kernel_info_replyrarrjr)rrrrrrQr request_kernel_info s"      z&ZMQChannelsHandler.request_kernel_infocCs|j|\}}z|j|}Wn|jjddd|jiYdS|d}|jd||ddks:d |vrC|jd |i}|||j rP|j d|_ dS) zZprocess the kernel_info_reply enabling msg spec adaptation, if necessary zBad kernel_info replyTr@NcontentzReceived kernel info: %smsg_typekernel_info_replyrz/Kernel info request failed, assuming current %s) rafeed_identities deserializerErFrrkr|rrrs)rr~identsinforrr r$s     z,ZMQChannelsHandler._handle_kernel_info_replyc Csj|dt}|tkr&t|dd|j_|jd|d|jdtd|j s3|j |dSdS) zFinish handling kernel_info reply Set up protocol adaptation, if needed, and signal that connection can continue. r.rzAdapting from protocol version z (kernel z) to z (client).N) r!client_protocol_versionintsplitra adapt_versionrErr0rrjrk)rrrrrr r=s " z&ZMQChannelsHandler._finish_kernel_infocsZtd|_i|_d|_d|_t|_t|_d|_ d|_ d|_ d|_ d|_ g|_dS)NrF)super initialize zmq_streamrcr0rrr _close_future session_key_iopub_window_msg_count_iopub_window_byte_count_iopub_msgs_exceeded_iopub_data_exceeded_iopub_window_byte_queuerQrXrr rJs  zZMQChannelsHandler.initializec#sntVjj}|jjj_fdd}t }| | j |VdS)Ncs*rdSjdjidS)z*Don't wait forever for the kernel to replyNz-Timeout waiting for kernel_info reply from %s)rjrErr0rkrrrrr give_upksz+ZMQChannelsHandler.pre_get..give_up)rpre_get_register_sessionrrr0rakeyrrr add_timeoutrrK)rrrrvrrr r^s    zZMQChannelsHandler.pre_getc#s"t|d|_tj|dVdS)Nascii)r0)r r0rr!)rr0rrr r!vs zZMQChannelsHandler.getccsd|jd|jj|_|j|j}|r"|jd|j|V|j|jvr0||j|j<dSdS)a#Ensure we aren't creating a duplicate session. If a previous identical session is still open, close it to avoid collisions. This is likely due to a client reconnecting from a lost network connection, where the socket on our side has not been cleaned up yet. :zReplacing stale connection: %sN) r0rar_open_sessionsr!rErrsr)r stale_handlerrrr r{s  z$ZMQChannelsHandler._register_sessionc s*tj}||||jr;djkr;jdjd_ }fdd}| |n.replayzError opening stream: %sdeadcs$jD] \}}|jqdSr)rcitemson_recv_streamr)rrdrfrQrr subscribesz*ZMQChannelsHandler.open..subscribe)ropenrnotify_connect get_bufferrrErrcrrrgr HTTPErrorrFrrrrsadd_restart_callbackr0on_kernel_restartedon_restart_failed) rr0r connectedrrIrdrfrrrr rs8        zZMQChannelsHandler.opencCs|js |jd|dSt|trt|}nt|}|dd}|dur.|j d|d}||jvr<|j d|dS|j j }|dd}|rX||vrX|j d|d dS|j|}|j ||dS) Nz'Received message on closed websocket %rrdz(No channel specified, assuming shell: %sr\zNo such channel: %rheaderrzReceived message of type "z"", which is not allowed. Ignoring.)rcrEr| isinstancebytesrrloadspoprrallowed_message_typesrar)rr~rdammtrfrrr on_messages&        zZMQChannelsHandler.on_messagecsj|\}}j|}|dfdd}t|dd}|dd}|dkrE|dkrE|d d d krEg_d _d _d _d _ |dkrM|dvrMt } t jd krjd } | | d kr|j| d8_jd8_jd =nnt jd ks\jd7_|dkrtdd|D} nd } j| 7_j| j| ftjj} tjj} jd kr| jkr݈jsd_|tdjdjdnjr| djkrd _j sjdjd kr| jkrj sd_ |tdjdjdnj r/| djkr/d _ js/jdjs7j rMjd8_j| 8_jddSt||dS)N parent_headercsHj|jjd|dddd}d|d<tj|tddS) Nrf stderr)textr")rparentr[rdr)rErrar~ write_messagerrr) error_messager~rrrr write_stderrs  z6ZMQChannelsHandler._on_zmq_reply..write_stderrrdrrr[statusrrhidlerF>r comm_open execute_inputrrfcss|]}t|VqdSr)r).0xrrr sz3ZMQChannelsHandler._on_zmq_reply..Ta IOPub message rate exceeded. The notebook server will temporarily stop sending output to the client in order to avoid crashing it. To change this limit, set the config variable `--NotebookApp.iopub_msg_rate_limit`. Current values: NotebookApp.iopub_msg_rate_limit=z> (msgs/sec) NotebookApp.rate_limit_window=z (secs) g?ziopub messages resumeda IOPub data rate exceeded. The notebook server will temporarily stop sending output to the client in order to avoid crashing it. To change this limit, set the config variable `--NotebookApp.iopub_data_rate_limit`. Current values: NotebookApp.iopub_data_rate_limit=z? (bytes/sec) NotebookApp.rate_limit_window=)rarrrYr!rrrrrrrrrsumappendrTfloatrOrrErrSrrr)rrfrr fed_msg_listr~rrdrnowqueued byte_countmsg_rate data_raterrr rs|   "            z ZMQChannelsHandler._on_zmq_replycst|jSr)rrsrrQrrr rs>s zZMQChannelsHandler.closecCs|jd|j|j|j|ur|j|j|j}|j|vrQ||j| |j|j | |j|j d|j |jdkrQ| |j|j|j|jddS|jD]\}}|durk|sk|d|qVi|_|jddS)NzWebsocket closed %srr)rEr|rrr!rrr0notify_disconnectremove_restart_callbackrr_kernel_connectionsstart_bufferingrcrrkrrrrrs)rrrdrfrrr on_closeBs.     zZMQChannelsHandler.on_closecCsT|jdd}|r|s||jdd|i}d|d<|tj|t ddS)Nr[rrhrdr) rcr!rrflushrar~rrrr)rrr[r~rrr _send_status_messagecs z'ZMQChannelsHandler._send_status_messagecCtd|j|ddS)Nzkernel %s restarted restarting)loggingwarnr0rrQrrr rpz&ZMQChannelsHandler.on_kernel_restartedcCr)Nzkernel %s restarted failed!r)rrFr0rrQrrr rtrz$ZMQChannelsHandler.on_restart_failed)r3r4r5__doc__rpropertyrKrOrSrTrZrgrrrrrrr7rr!rrrrrsrrrr __classcell__rrrr rJcsB       .  m! rJz"(?P\w+-\w+-\w+-\w+-\w+)z(?Prestart|interrupt)z /api/kernelsz /api/kernels//z /channels)%r rrtextwraprtornadorrtornado.concurrentrtornado.iolooprjupyter_clientrrjupyter_client.jsonutilr ImportErrorr ipython_genutils.py3compatr notebook.utilsr r r base.handlersrbase.zmqhandlersrrrr9r=rJ_kernel_id_regex_kernel_action_regexdefault_handlersrrrr s@