B 3¢\ôOã@sdZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZd d lmZd d lmZmZdd lmZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdZdZ defdeefdee fefdeefgZ!dS)z®Tornado 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)ÚgenÚweb)ÚFuture)ÚIOLoop)Ú date_default)Ú cast_unicode)Ú url_path_joinÚ url_escapeé)Ú APIHandler)ÚAuthenticatedZMQStreamHandlerÚdeserialize_binary_message)Úprotocol_versionc@s4eZdZejejdd„ƒƒZejejdd„ƒƒZdS)ÚMainKernelHandlerccs.|j}t | ¡¡V}| tj|td¡dS)N)Údefault)Úkernel_managerrÚ maybe_futureZ list_kernelsÚfinishÚjsonÚdumpsr)ÚselfÚkmÚkernels©rúAlib/python3.7/site-packages/notebook/services/kernels/handlers.pyÚgetszMainKernelHandler.getccs”|j}| ¡}|dkr"d|ji}n| d|j¡t |j|dd¡V}| |¡}t|j ddt |ƒƒ}|  d|¡|  d¡|  tj|td¡dS)NÚname)Z kernel_nameZapirZLocationéÉ)r)rZ get_json_bodyZdefault_kernel_nameÚ setdefaultrrZ start_kernelÚ kernel_modelr Zbase_urlr Z set_headerÚ set_statusrrrr)rrÚmodelÚ kernel_idÚlocationrrrÚpost#s    zMainKernelHandler.postN) Ú__name__Ú __module__Ú __qualname__rÚ authenticatedrÚ coroutinerr%rrrrrsrc@s.eZdZejdd„ƒZejejdd„ƒƒZdS)Ú KernelHandlercCs2|j}| |¡| |¡}| tj|td¡dS)N)r)rZ_check_kernel_idr rrrr)rr#rr"rrrr9s  zKernelHandler.getccs.|j}t | |¡¡V| d¡| ¡dS)NéÌ)rrrZshutdown_kernelr!r)rr#rrrrÚdelete@s zKernelHandler.deleteN) r&r'r(rr)rrr*r-rrrrr+7sr+c@s eZdZejejdd„ƒƒZdS)ÚKernelActionHandlerc cs¨|j}|dkr"| |¡| d¡|dkrœyt | |¡¡VWn<tk r|}z|jjddd| d¡Wdd}~XYn X|  |¡}|  t j |t d¡| ¡dS) NZ interruptr,ZrestartzException restarting kernelT)Úexc_infoiô)r)rZinterrupt_kernelr!rrZrestart_kernelÚ ExceptionÚlogÚerrorr Úwriterrrr)rr#ÚactionrÚer"rrrr%Ks   zKernelActionHandler.postN)r&r'r(rr)rr*r%rrrrr.Isr.csúeZdZdZiZedd„ƒZedd„ƒZedd„ƒZedd „ƒZ d d „Z d d „Z dd„Z dd„Z dd„Z‡fdd„Zej‡fdd„ƒZej‡fdd„ƒZejdd„ƒZ‡fdd„Zdd„Z‡fd d!„Z‡fd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Z‡ZS),ÚZMQChannelsHandlerz]There is one ZMQChannelsHandler per running kernel and it oversees all the sessions. cCs|jj}|j d|¡S)NÚkernel_info_timeout)rr7Úsettingsr)rZ km_defaultrrrr7isz&ZMQChannelsHandler.kernel_info_timeoutcCs|j dd¡S)NÚiopub_msg_rate_limitr)r8r)rrrrr9nsz'ZMQChannelsHandler.iopub_msg_rate_limitcCs|j dd¡S)NÚiopub_data_rate_limitr)r8r)rrrrr:rsz(ZMQChannelsHandler.iopub_data_rate_limitcCs|j dd¡S)NÚrate_limit_windowgð?)r8r)rrrrr;vsz$ZMQChannelsHandler.rate_limit_windowcCsd|jjt|ddƒfS)Nz%s(%s)r#Z uninitialized)Ú __class__r&Úgetattr)rrrrÚ__repr__zszZMQChannelsHandler.__repr__cCsL|j}|jj}x8dD]0}t|d|ƒ}||j|d|j|<}||_qWdS)N)ÚshellÚiopubÚstdinZconnect_)Úidentity)rÚsessionZbsessionr=r#ÚchannelsÚchannel)rrrBrEÚmethÚstreamrrrÚ create_stream}s  z ZMQChannelsHandler.create_streamcs¬ˆj}| ˆj¡}y |j}Wnbtk r~ˆj dˆj¡ˆjdkrT| ˆj¡ˆ_ˆj  ˆj ¡ˆj   ˆjd¡ˆj|_Yn(X|  ¡s”ˆj d¡| ‡fdd„¡ˆjS)zsend a request for kernel_infozRequesting kernel info from %sNZkernel_info_requestz'Waiting for pending kernel_info requestcsˆ | ¡¡S)N)Ú_finish_kernel_infoÚresult)Úf)rrrÚ™óz8ZMQChannelsHandler.request_kernel_info..)rÚ get_kernelr#Ú_kernel_info_futureÚAttributeErrorr1ÚdebugÚkernel_info_channelZ connect_shellÚon_recvÚ_handle_kernel_info_replyrCÚsendÚdoneZadd_done_callback)rrÚkernelÚfuturer)rrÚrequest_kernel_info…s    z&ZMQChannelsHandler.request_kernel_infocCs¨|j |¡\}}y|j |¡}Wn&|jjddd|j i¡dS|d}|j d|¡|ddksrd |kr„|j d |¡i}| |¡|j rž|j   ¡d|_ dS) zbprocess the kernel_info_reply enabling msg spec adaptation, if necessary zBad kernel_info replyT)r/NÚcontentzReceived kernel info: %sÚmsg_typeZkernel_info_replyrz/Kernel info request failed, assuming current %s) rCÚfeed_identitiesÚ deserializer1r2rOÚ set_resultrQrIrRÚclose)rÚmsgÚidentsÚinforrrrTœs    z,ZMQChannelsHandler._handle_kernel_info_replycCsV| dt¡}|tkr.give_up)rhr6Úpre_getÚ_register_sessionrrNr#rCÚkeyrYrÚcurrentZ add_timeoutÚtimer7)rrWrrZloop)r<)rXrrrsÖs  zZMQChannelsHandler.pre_getc#s$t|dƒ|_tt|ƒj|dVdS)NÚascii)r#)rr#rhr6r)rr#)r<rrrîs zZMQChannelsHandler.getccsPd|j|jjf|_|j |j¡}|r@|j d|j¡| ¡V||j|j<dS)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. z%s:%szReplacing stale connection: %sN)r#rCrkÚ_open_sessionsrr1rqr_)rZ stale_handlerrrrrtós  z$ZMQChannelsHandler._register_sessionc s^tt|ƒ ¡|j}| |¡| ||j¡}|r |d|jkr |j d|j¡|d|_ |d}|rž|j dt |ƒ¡xš|D]\}}|j |}|  ||¡q|Wnry |  ¡Wndt jk r}zB|j d|¡x&|j  ¡D]\}}| ¡sÜ| ¡qÜW| ¡dSd}~XYnX| |j|j¡| |j|jd¡x$|j  ¡D]\}}| |j ¡q@WdS)NrkzRestoring connection for %srDÚbufferzReplaying %s buffered messageszError opening stream: %sÚdead)rhr6ÚopenrZnotify_connectZ get_bufferrkr1rbrDÚlenÚ _on_zmq_replyrHrZ HTTPErrorr2ÚitemsÚclosedr_Zadd_restart_callbackr#Úon_kernel_restartedÚon_restart_failedZon_recv_stream) rr#rZ buffer_infoZ replay_bufferrEÚmsg_listrGr5)r<rrr|s4     zZMQChannelsHandler.opencCs”|js|j d|¡dSt|tƒr,t|ƒ}n t |¡}| dd¡}|dkr\|j  d|¡d}||jkrx|j  d|¡dS|j|}|j   ||¡dS)Nz'Received message on closed websocket %rrEz(No channel specified, assuming shell: %sr?zNo such channel: %r) rDr1rQÚ isinstanceÚbytesrrÚloadsÚpoprqrCrU)rr`rErGrrrÚ on_message%s      zZMQChannelsHandler.on_messagecsªˆj |¡\}}ˆj |¡}|d‰‡‡fdd„}t|ddƒ}|dd}|dkrŠ|dkrŠ|d  d ¡d krŠgˆ_d ˆ_d ˆ_d ˆ_d ˆ_ |dkr”|dkr”t   ¡  ¡} xTt ˆjƒd krþˆjd } | | d krúˆj| d8_ˆjd8_ˆjd =q¬Pq¬Wˆjd7_|dkr,tdd„|Dƒƒ} nd } ˆj| 7_ˆj | ˆj| f¡tˆjƒˆj} tˆjƒˆj} ˆjd kr´| ˆjkr´ˆjsædˆ_|td ˆjˆj¡ƒƒn2ˆjræ| dˆjkræd ˆ_ˆj sæˆj d¡ˆjd kr&| ˆjkr&ˆj sXdˆ_ |td ˆjˆj¡ƒƒn2ˆj rX| dˆjkrXd ˆ_ ˆjsXˆj d¡ˆjshˆj r”ˆjd8_ˆj| 8_ˆj d¡dSttˆƒ ||¡dS)NZ parent_headercsHˆj |¡ˆjjd|dddœˆd}d|d<ˆ tj|td¡dS) NrGÚ Ústderr)Útextr)rZÚparentr@rE)r)r1rqrCr`Ú write_messagerrr)Z error_messager`)rŒrrrÚ write_stderr<s   z6ZMQChannelsHandler._on_zmq_reply..write_stderrrEÚheaderr[r@ÚstatusrZÚexecution_stateZidlerF>Ú comm_openrÚ execute_inputérGcSsg|] }t|ƒ‘qSr)r})Ú.0Úxrrrú bsz4ZMQChannelsHandler._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={} (msgs/sec) NotebookApp.rate_limit_window={} (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={} (bytes/sec) NotebookApp.rate_limit_window={} (secs) éÿÿÿÿ)rCr\r]r=rrprlrmrnrorrvrwr}ÚsumÚappendr;Úfloatr9rÚformatr1rqr:r‡rhr6r~)rrGrƒraZ fed_msg_listr`rŽrEr[ZnowZqueuedZ byte_countZmsg_rateZ data_rate)r<)rŒrrr~8sj   "          z ZMQChannelsHandler._on_zmq_replycstt|ƒ ¡|jS)N)rhr6r_rj)r)r<rrr_¥szZMQChannelsHandler.closecCsò|j d|j¡|j |j¡|kr0|j |j¡|j}|j|kr¢| |j¡|  |j|j ¡|  |j|j d¡|j |jdkr¢|  |j|j|j¡|j d¡dSx8|j ¡D]*\}}|dk r®| ¡s®| d¡| ¡q®Wi|_|j d¡dS)NzWebsocket closed %sr{r)r1rQrkryrr‡rr#Znotify_disconnectZremove_restart_callbackrr‚Z_kernel_connectionsZstart_bufferingrDrjr^rr€rSr_)rrrErGrrrÚon_close©s(      zZMQChannelsHandler.on_closecCsT|j dd¡}|r"| ¡s"| ¡|j dd|i¡}d|d<| tj|t d¡dS)Nr@rr‘rE)r) rDrr€ÚflushrCr`rrrr)rrr@r`rrrÚ_send_status_messageÊs  z'ZMQChannelsHandler._send_status_messagecCst d|j¡| d¡dS)Nzkernel %s restartedZ restarting)ÚloggingÚwarnr#rŸ)rrrrr×sz&ZMQChannelsHandler.on_kernel_restartedcCst d|j¡| d¡dS)Nzkernel %s restarted failed!r{)r r2r#rŸ)rrrrr‚Ûsz$ZMQChannelsHandler.on_restart_failed)r&r'r(Ú__doc__ryÚpropertyr7r9r:r;r>rHrYrTrIrirr*rsrrtr|rˆr~r_rrŸrr‚Ú __classcell__rr)r<rr6_s.      # m ! r6z"(?P\w+-\w+-\w+-\w+-\w+)z(?Prestart|interrupt)z /api/kernelsz/api/kernels/%sz/api/kernels/%s/%sz/api/kernels/%s/channels)"r¢rr ÚtextwraprZtornadorrZtornado.concurrentrZtornado.iolooprZjupyter_client.jsonutilrZipython_genutils.py3compatrZnotebook.utilsr r Z base.handlersr Zbase.zmqhandlersr rZjupyter_clientrrdrr+r.r6Z_kernel_id_regexZ_kernel_action_regexZdefault_handlersrrrrÚs2