
    oi                    8   U d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddlZddlmZ ddlmZmZmZ ddlmZ ddlmZ ddlmZmZmZmZmZ d	d
lmZ ej>                  dk7  rddlm Z  ndZ 	 ddl!m"Z# ddl%Z%ddl&Z&ddl'm(Z( ddl!m)Z) ddl*m+Z+ ddl,m-Z- ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 ddl9m:Z: ddl;m<Z< d	dl=m>Z> d	dl?m@Z@ d	dlAmBZBmCZC dZDdeEd<    ej                  d      ZGd ZH G d de-      ZIy# e$$ r ejF                  Z#Y w xY w) z9Base class for a kernel that talks to frontends over 0MQ.    )annotationsN)Mapping)Context
ContextVarcopy_context)datetime)partial)SIGINTSIGTERMSignalsdefault_int_handlersignal   )CONTROL_THREAD_NAMEwin32)SIGKILLzwindown-SIGKILL-sentinel)utcnowStdinNotImplementedError)Session)ioloop)SingletonConfigurable)
AnyBoolDictFloatInstanceIntegerListUnicodedefaultobserve)	ZMQStream)
json_clean)kernel_protocol_version)	OutStream)LazyDict_async_in_contextaa  For consistency across implementations, it is recommended that `{func_name}` either be a coroutine function (`async def`) or return an awaitable object (like an `asyncio.Future`). It might become a requirement in the future. Coroutine functions and awaitables have been supported since ipykernel 6.0 (2021). {target} does not seem to return an awaitablestr_AWAITABLE_MESSAGETc                :   t        j                  |       j                  }t        j	                  |d      }|D ]`  }|j                  |      }|xr$ |j                  |j                  |j                  fv xs  t        d |j                         D              ||<   b |S )NFc              3  N   K   | ]  }|j                   |j                  k(    y wN)kindVAR_KEYWORD).0ps     \/home/ubuntu/docker-apps/notebooks/venv/lib/python3.12/site-packages/ipykernel/kernelbase.py	<genexpr>z&_accepts_parameters.<locals>.<genexpr>U   s     FQ1==(Fs   #%)inspect	signature
parametersdictfromkeysgetr/   KEYWORD_ONLYPOSITIONAL_OR_KEYWORDanyvalues)methparam_namesr7   acceptsparam
param_specs         r3   _accepts_parametersrD   L   s    ""4(33JmmK/G G^^E*
 _J$;$;Z=]=]#^^G F*2C2C2EFF 	G N    c                      e Zd ZU dZ ed      Zi Zded<    ed      d        Z	 e
ed      Z e
d	d      Z e
ed      Z ed
      Zded<   ded<   ded<   ded<    ed      d        Z ed      d        Z e
ed      Z e       Z e       Z e       Z e       Z e       Z e       Z e
ej:                  d      Zded<    ed      Z  e!       Z" ed      d        Z#i Z$ded<    e       Z%ded<    e&dd      jO                  d      Z( e&dd      jO                  d      Z) e&d       Z* e+i       Z,d!ed"<   d#Z-d$ed%<   d&ed'<   d(ed)<   d*ed+<   d,ed-<   d.ed/<   e.d0        Z/ e0d1      jO                  d      Z1 e0d2      jO                  d      Z2 e0d3dd45      Z3dZ4 e+       Z5d6Z6 e
e7jp                        Z9g d7Z:g e:d8d9d:d;d<Z; fd=Z<d> Z=d? Z>d@ Z?d{d|dAZ@dB ZAdC ZBdD ZCdE ZDdF ZEdG ZFd|dHZGdI ZHdJ ZId{dKZJd{dLZKdM ZLd}dNZMd{dOZNd~dPZO	 	 	 	 	 	 	 ddQZPdR ZQdS ZRdT ZS	 	 	 ddddUdVZTdW ZUdX ZVdY ZWddZZXd[ ZY	 	 	 	 	 	 dd\ZZd] Z[e.d^        Z\d_ Z]d` Z^da Z_db Z`dc Zadd Zbde Zcdf Zddg Zed{dhZfdi Zgdj ZhddkZiddlZjddmZkdn Zl e&d       ZmddoZnd{d|dpZodq Zpdr ZqddsZrddtZsdduZtdv Zudw Zvdx Zwdy Zxe.dz        Zy xZzS )KernelzThe base kernel class.Nzdict[str, psutil.Process]	processes	eventloopc                    t         j                  j                         }|j                  |j	                  | j
                         yy)z&schedule call to eventloop from IOLoopN)r   IOLoopcurrentnewadd_callbackenter_eventloop)selfchangeloops      r3   _update_eventloopzKernel._update_eventloopf   s9     }}$$&::!d223 "rE   T)
allow_nonez"IPython.core.profiledir.ProfileDirzDeprecated shell_streams alias. Use shell_stream

        .. versionchanged:: 6.0
            shell_streams is deprecated. Use shell_stream.
        )helpzList[t.Any]shell_streamsr)   implementationimplementation_versionbannerc                p    t        j                  dt        d       | j                  | j                  gS g S )NLKernel.shell_streams is deprecated in ipykernel 6.0. Use Kernel.shell_stream   
stacklevel)warningswarnDeprecationWarningshell_streamrP   s    r3   _shell_streams_defaultzKernel._shell_streams_default}   s9    Z	

 (%%&&	rE   c                    t        j                  dt        d       t        |j                        dkD  rt        j                  dt
        d       |j                  r|j                  d   | _        y y )Nr[   r\   r]   r   zJKernel only supports one shell stream. Additional streams will be ignored.r   )r_   r`   ra   lenrM   RuntimeWarningrb   )rP   rQ   s     r3   _shell_streams_changedzKernel._shell_streams_changed   s\    Z	

 vzz?QMM\
 :: &

1D rE   zlogging.Loggerlogidentc                <    t        t        j                               S r.   )r)   uuiduuid4rc   s    r3   _default_identzKernel._default_ident   s    4::<  rE   zdict[str, object]language_infozList[dict[str, str]]
help_linkszSSet to False if you want to debug python standard and dependent libraries.
        )configzhWhether to use appnope for compatibility with OS X App Nap.

        Only affects OS X >= 10.9.
        FzDict[str, t.Any]_control_parentrE   bytes_control_parent_identzContextVar[dict[str, Any]]_shell_parentzContextVar[bytes]_shell_parent_identr   _shell_contextzMapping[str, bytes]_parent_identzasyncio.Lock_main_asyncio_lockc                Z    t        j                  dt        d       | j                         S )NzEKernel._parent_header is deprecated in ipykernel 6. Use .get_parent()r\   r]   )r_   r`   ra   
get_parentrc   s    r3   _parent_headerzKernel._parent_header   s&    S	

   rE   gMb@?{Gz?        a  time (in seconds) to wait for messages to arrive
        when aborting queued requests after an error.

        Requests that arrive within this window after an error
        will be cancelled.

        Increase in the event of unusually slow network
        causing significant delays,
        which can manifest as e.g. "Run all" in a notebook
        aborting some, but not all, messages after an error.
        )rr   rU   r   )
execute_requestcomplete_requestinspect_requesthistory_requestcomm_info_requestkernel_info_requestconnect_requestshutdown_requestis_complete_requestinterrupt_requestdebug_requestusage_requestcreate_subshell_requestdelete_subshell_requestlist_subshell_requestc                    t           d
i | t        j                   _        t        j
                   _        i  _         j                  D ]  }t         |       j                  |<    i  _
         j                  D ]  }t         |       j                  |<    t         j                  ddg       _        i  _        d _        t#        d       _         j$                  j'                  i        t#        d       _         j(                  j'                  d       t+                _        t/         fd fdd       _        t3        j4                          _        y	)zInitialize the kernel.	cell_metacell_idrE   shell_parentshell_parent_identc                      j                   S r.   )ru   rc   s   r3   <lambda>z!Kernel.__init__.<locals>.<lambda>J  s    4#=#= rE   c                 :     j                   j                        S r.   )_get_shell_context_varrw   rc   s   r3   r   z!Kernel.__init__.<locals>.<lambda>K  s    !<!<T=U=U!V rE   )controlshellN )super__init__sysstdout_stdoutstderr_stderrshell_handlers	msg_typesgetattrcontrol_handlerscontrol_msg_typesrD   
do_execute_do_exec_accepted_paramsrs   ru   r   rv   setrw   r   rx   r'   ry   asyncioLockrz   )rP   kwargsmsg_type	__class__s   `  r3   r   zKernel.__init__%  s?   "6"
 .1ZZ-0ZZ ! 	DH,3D(,CD)	D !#.. 	FH.5dH.ED!!(+	F )<OOk95)
%  "%("'7r"#-.B#C   $$S)*n &=V
 #*,,.rE   c                   K   | j                   4 d{    | j                  |       d{    ddd      d{    y7 .7 7 	# 1 d{  7  sw Y   yxY ww)zMDispatch a control request, ensuring only one message is processed at a time.N)_control_lockprocess_controlrP   msgs     r3   dispatch_controlzKernel.dispatch_controlQ  sX      %% 	, 	,&&s+++	, 	, 	,+	, 	, 	, 	,sQ   AAAAAAAAAAAAAAAc                  K   | j                   sy| j                   j                  |d      \  }}	 | j                   j                  |dd      }| j                  j                  d|       | j                  ||d	
       | j                  dd	       |d   }|d   }| j                  j                  |d      }|| j                  j                  d|       n4	  || j                  ||      }t        j                  |      r
| d{    t        j                  t        j                  j!                          t        j"                  t        j"                  j!                          | j%                  dd	| j                         y# t        $ r  | j                  j                  dd       Y yw xY w7 # t        $ r  | j                  j                  dd       Y w xY ww)zdispatch control requestsNFcopyTcontentr   zInvalid Control Messageexc_infozControl received: %sr   channelbusyheaderr   z UNKNOWN CONTROL MESSAGE TYPE: %rzException in control handler:idle)sessionfeed_identitiesdeserialize	Exceptionri   errordebug
set_parent_publish_statusr   r:   control_streamr5   isawaitabler   r   flushr   _publish_status_and_flush)rP   r   identsr   r   handlerresults          r3   r   zKernel.process_controlW  s    ||ll223U2C	,,**35*IC
 	-s3 	Y7VY/X*%''++Hd;?HHNN=xHO !4!4fcB&&v. LL ::!JJ::!JJ&&vy$:M:MN9  	HHNN4tND	( ! O>NOs`   .GF BG.F2 F0F2 
A:G&F-*G,F--G0F2 2&GGGGc                     y)a"  Check whether a shell-channel message should be handled

        Allows subclasses to prevent handling of certain messages (e.g. aborted requests).

        .. versionchanged:: 7
            Subclass should_handle _may_ be async.
            Base class implementation is not async.
        Tr   )rP   streamr   r   s       r3   should_handlezKernel.should_handle|  s     rE   c               R	  K   t        |      dk(  rC|d   j                  dk(  r1|d| _        y| j                  j                  j                  |d       y| j                  sy| j                  r.t        j                         | j                  | j                  fvsJ | j                  j                  |d      \  }}	 | j                  j                  |dd      }| j!                  ||d       | j#                  dd       |d   d   }|d   j%                  d      |k(  sJ | j                  r&| j                  j                  j'                  |      }n| j(                  }|dk(  r]|| j                  }n%| j                  j                  j+                  |      }|r'| j-                  |||       | j/                  dd|       y| j                  j1                  d|       | j                  j1                  d|d          | j3                  |||      }t5        j6                  |      r
| d{   }|sC| j/                  dd|       | j                  j1                  d||d   j%                  d             y| j8                  j%                  |d      }|| j                  j;                  d|       ni| j                  j1                  d||       	 | j=                          	  ||||      }	t5        j6                  |	      r
|	 d{    	 | jA                          tB        jD                  tB        jD                  jG                          tB        jH                  tB        jH                  jG                          | j/                  dd|       y# t        $ r  | j                  j                  d	d
       Y yw xY w7 # t        $ r! | j                  j1                  dd
       Y w xY w7 # t        $ r! | j                  j                  dd
       Y t>        $ r | j                  j                  d       Y .w xY w# t        $ r! | j                  j1                  dd
       Y Jw xY w# 	 | jA                          w # t        $ r  | j                  j1                  dd
       Y w w xY wxY ww)zdispatch shell requestsr   r      stop abortingNFr   Tr   zInvalid Messager   r   r   r   r   r   subshell_idr   r   z
*** MESSAGE TYPE:%s***z   Content: %s
   --->
   r   zNot handling %s:%smsg_idzUnknown message type: %rz%s: %sz%Unable to signal in pre_handler_hook:zException in message handler:z#KeyboardInterrupt caught in kernel.z&Unable to signal in post_handler_hook:)%rf   buffer	_abortingshell_channel_threadmanagerset_subshell_abortingr   _supports_kernel_subshells	threadingcurrent_threadcontrol_threadr   r   r   ri   r   r   r   r:   $get_subshell_to_shell_channel_socketrb   get_subshell_aborting_send_abort_replyr   r   r   r5   r   r   warningpre_handler_hookKeyboardInterruptpost_handler_hookr   r   r   r   )
rP   r   r   r   r   r   abortingr   r   r   s
             r3   dispatch_shellzKernel.dispatch_shell  s,    s8q=SV]].>>
 "!&  ))11GGUZ[||**++-##))6   
 ll223U2C	,,**35*IC 	W5VW-x=,8}  /;>>>**..66[[F &&F ((">>44<<RRS^_&&vsF;..vwG
 	18<5s9~F262D2DVSRX2Y}-"//M**67FCHHNN/3x=;L;LX;VW%%))(D9?HH7BHHNN8Xs3W%%'\ 5&&v. LL\**, ::!JJ::!JJ&&vw?O  	HHNN,tN<	J 0  WFQUVW
 ! O>N$ FDEF ! \HHNN#KVZN[\\**,  \HHNN#KVZN[\s   B9R'<N ER')N3*B!R'N6 $O% O#O% P8 A0R'&N0-R'/N00R'6&O R'O  R'#O% %&P5Q% #P51Q% 4P55Q% 8&Q"R'!Q""R'%R$'Q87R$8&R!R$ R!!R$$R'c                6    t        t        t              | _        y)z.Hook to execute before calling message handlerN)r   r
   r   saved_sigint_handlerrc   s    r3   r   zKernel.pre_handler_hook  s     %+63F$G!rE   c                8    t        t        | j                         y)z-Hook to execute after calling message handlerN)r   r
   r   rc   s    r3   r   zKernel.post_handler_hook  s    vt001rE   c                      j                   j                  d j                          j                   j                   j                  d       y fd fd         y)zenter eventloopzEntering eventloop %sNz Exiting as there is no eventloopc                 .  K   j                    urj                  j                  d        y j                  j                  d        	          j                    u r         y y # t        $ r j                  j                  d       Y =w xY ww)Nzexiting eventloop %szAdvancing eventloop %sz"KeyboardInterrupt caught in kernel)rI   ri   infor   r   r   )rI   schedule_nextrP   s   r3   advance_eventloopz1Kernel.enter_eventloop.<locals>.advance_eventloop  s     ~~Y.4i@HHNN3Y?E$ ~~* + % ECDEs*   ABA+ B+$BBBBc                 t    j                   j                  d       j                  j                  d        y)z*Schedule the next advance of the eventloopzScheduling eventloop advancegMbP?N)ri   r   io_loop
call_later)r   rP   s   r3   r   z-Kernel.enter_eventloop.<locals>.schedule_next
  s+     HHNN9:LL##E+<=rE   )ri   r   rI   )rP   r   rI   r   s   `@@@r3   rO   zKernel.enter_eventloop  sN    -t~~>NN	HHMM<=	 	> 	rE   c                >   K   t        j                         | _        y wr.   )r   r   r   rc   s    r3   _create_control_lockzKernel._create_control_lock  s     $\\^s   c                   t         j                  j                         | _        | j                  r'| j                  j                  | j                  d       | j                  rCt        j                  dk  r0| j                  j                  j                  | j                         nt        j                         | _        | j                  r| j                   rW| j                   j"                  j%                  | j&                         | j                  j                  | j(                  d       n:| j                  j                  t+        t-        | j&                  d            d       | j/                  dd       y)z register dispatchers for streamsFr   )   
   Nstartingr   )r   rK   rL   r   r   on_recvr   r   r   version_inforN   r   r   r   r   rb   r   r   set_on_recv_callback
shell_mainshell_channel_thread_mainr(   r	   r   rc   s    r3   startzKernel.start  s   }},,.''(=(=E'J3#3#3g#= ''44T5N5NO!(D(())11FFtW!!))$*H*Hu)U!!))%gdoot&DE *  	Z1rE   c                  K   t        j                         | j                  k(  sJ | j                  j                  4 d{    | j                  	 ddd      d{    y| j                  j                  |d      \  }}	 | j                  j                  |dd      }|d   j                  d      }| j                  j                  }|j                  |      }|J  |j                  |d       ddd      d{    y7 7 # t        $ r  | j                  j                  dd	       Y =w xY w7 5# 1 d{  7  sw Y   yxY ww)
z;Handler for shell messages received on shell_channel_threadNFr   r   r   r   zInvalid messageTr   )r   r   r   asyncio_lockr   r   r   r:   r   $get_shell_channel_to_subshell_socketsend_multipartr   ri   r   )rP   r   _msg2msg3r   subshell_managersockets           r3   r   z Kernel.shell_channel_thread_main7  sM    '')T-F-FFFF,,99 	A 	A||#	A 	A 	A ll223U2CGAt
A||//e%/P"8n00? $(#<#<#D#D )NN{[)))%%%c6	A 	A 	A 	A   A04@A!	A 	A 	A 	As   >E DED8EDE# D8A1D
5E D6EE
&D30D82D33D86E8E
>E?E
Ec                "  K   | j                   r|:t        j                         | j                  j                  k(  sJ | j
                  }nnt        j                         | j                  | j                  j                  fvsJ | j                  j                  j                  |      }n|J | j
                  }|J |4 d{    | j                  ||       d{    ddd      d{    y7 07 7 	# 1 d{  7  sw Y   yxY ww)z/Handler of shell messages for a single subshellN)r   )	r   r   r   r   parent_threadrz   r   get_subshell_asyncio_lockr   )rP   r   r   r  s       r3   r   zKernel.shell_mainN  s    **" //1T5N5N5\5\\\\#66 //1----;;:     $88@@ZZ  &&&22L ''' 	D 	D%%c{%CCC	D 	D 	DC	D 	D 	D 	DsZ   CDC4DC:C6C:#D.C8/D6C:8D:D DDDc                    || _         y)zRecord the ports that this kernel is using.

        The creator of the Kernel instance must call this methods if they
        want the :meth:`connect_request` method to return the port numbers.
        N)_recorded_ports)rP   portss     r3   record_portszKernel.record_portsh  s      %rE   c           	         | j                   sy| j                   j                  | j                  d||d|| j                  d             y)z-Publish the code request on the iopub stream.Nexecute_input)codeexecution_countparentrk   )r   sendiopub_socket_topic)rP   r  r  r  s       r3   _publish_execute_inputzKernel._publish_execute_inputt  sF    ||o>++o. 	 	
rE   c           	         | j                   sy| j                   j                  | j                  dd|i|xs | j                  |      | j	                  d             y)z send status (busy/idle) on IOPubNstatusexecution_stater  r   r  r  r|   r  )rP   r  r   r  s       r3   r   zKernel._publish_status  sT    ||'5T__W5++h' 	 	
rE   c                    | j                  |||       |r:t        |d      r-| j                  s |j                  t        j
                         yyyy)zfsend status on IOPub and flush specified stream to ensure reply is sent before handling the next replyr   N)r   hasattrr   r   zmqPOLLOUT)rP   r  r   r   r  s        r3   r   z Kernel._publish_status_and_flush  sC    VWf5gfg.t7V7VLL% 8W.6rE   c           	         | j                   sy | j                   j                  | j                  d|| j                         | j	                  d             y )Ndebug_eventr  r  )rP   events     r3   _publish_debug_eventzKernel._publish_debug_event  sH    ||??$++m, 	 	
rE   c                    |dk(  r|| _         || _        y| j                  j                  |       | j                  j                  |       t               | _        y)a  Set the current parent request

        Side effects (IOPub messages) and replies are associated with
        the request that caused them via the parent_header.

        The parent identity is used to route input_request messages
        on the stdin channel.
        r   N)ru   rs   rw   r   rv   r   rx   )rP   rk   r  r   s       r3   r   zKernel.set_parent  sN     i).D&#)D $$((/""6*"..DrE   c                    |*t        j                         j                  t        k(  rd}nd}|dk(  r| j                  S | j                  | j                        S )aR  Get the parent request associated with a channel.

        .. versionadded:: 6

        Parameters
        ----------
        channel : str
            the name of the channel ('shell' or 'control')

        Returns
        -------
        message : dict
            the parent message for the most recent request on the channel.
        r   r   )r   r   namer   rs   r   rv   )rP   r   s     r3   r|   zKernel.get_parent  sV      ?'')..2EE#!i'''**4+=+=>>rE   c                `    	 |j                         S # t        $ r | j                  |   cY S w xY w)zLookup a ContextVar, falling back on the shell context

        Allows for user-launched Threads to still resolve to the shell's main context

        necessary for e.g. display from threads.
        )r:   LookupErrorrx   )rP   vars     r3   r   zKernel._get_shell_context_var  s3    	,779 	,&&s++	,s    --c
                    | j                   sy| j                   j                  |||| j                  |	      |||||	      S )a  Send a response to the message we're currently processing.

        This accepts all the parameters of :meth:`jupyter_client.session.Session.send`
        except ``parent``.

        This relies on :meth:`set_parent` having been called for the current
        message.
        N)r   r  r|   )
rP   r   msg_or_typer   rk   bufferstrackr   metadatar   s
             r3   send_responsezKernel.send_response  sI    ( ||||  OOG$

 
	
rE   c                    dt               iS )zRInitialize metadata.

        Run at the beginning of execution requests.
        started)now)rP   r  s     r3   init_metadatazKernel.init_metadata  s     su
 	
rE   c                    |S )zXFinish populating metadata.

        Run after completing an execution request.
        r   )rP   r  r0  reply_contents       r3   finish_metadatazKernel.finish_metadata  s	    
 rE   c                  K   | j                   sy	 |d   }|d   }|j                  dd      }|j                  d|       }|j                  di       }|j                  dd      }	|j                  d	i       }
|
j                  d
      }|j                  dd      }| j                  |      }|s2| xj                  dz  c_        | j                  ||| j                         |||||	d}| j                  d   r|
|d<   | j                  d   r||d<   |d   j                  d      } | j                  di |}t        j                  |      r| d{   }n;t        j                  t        j                  d| j                        t         d       t"        j$                  t"        j$                  j'                          t"        j(                  t"        j(                  j'                          | j*                  rt-        j.                  | j*                         t1        |      }| j3                  |||      }| j                   j5                  |d||||      }| j                  j7                  d|       |s5|d   d   dk(  r)|r&|d   j                  d      }| j9                  |       yyyy# t        $ r: | j                  j	                  d       | j                  j	                  d|       Y yw xY w7 w)zhandle an execute_requestNr   r  silentFstore_historyuser_expressionsallow_stdinr0  cellIdzGot bad msg: %sstop_on_errorTr   )r  r:  r;  r<  r=  r   r   r   r   r   	func_nametargetr]   execute_reply)r0  rk   r  r   r   )r   r:   r   ri   r   r5  r  r  r   r   r5   r   r_   r`   r*   formatPendingDeprecationWarningr   r   r   r   _execute_sleeptimesleepr$   r8  r  r   _abort_queues)rP   r   rk   r  r   r  r:  r;  r<  r=  r   r   r@  r0  do_execute_argsr   r7  	reply_msgs                     r3   r   zKernel.execute_request  s    ||	Y'G6?D[[51F#KKVDM&{{+=rB!++mU;K

:r2ImmH-G  OT:%%f-   A% ''fd6J6JK * 0&
 ((5+4OK(((3)0OI&X&**=9 (:/:}-"//MMM"))L)Y) ::!JJ::!JJ JJt**+ #=1''-H&*ll&7&7 '8 '
	 	tY')I.x8GC *..}=K{+ IVCvK  	HHNN?+HHNN4(	F 0s8   K2A6J) CK2K/EK2)A K,)K2+K,,K2)r   r   c                   t         )z4Execute user code. Must be overridden by subclasses.NotImplementedError)rP   r  r:  r;  r<  r=  r   r   s           r3   r   zKernel.do_executea  s
     "!rE   c                v  K   | j                   sy|d   }|d   }|d   }| j                  ||      }t        j                  |      r| d{   }n;t	        j
                  t        j                  d| j                        t        d       t        |      }| j                   j                  |d	|||       y7 kw)
zHandle a completion request.Nr   r  
cursor_posdo_completerA  r   r]   complete_reply)r   rR  r5   r   r_   r`   r*   rE  rF  r$   r  )rP   r   rk   r  r   r  rQ  matchess           r3   r   zKernel.complete_requesto  s     ||#v\*
""44w'#mGMM"))M$JZJZ)[) W%&"2GVUK $s   A	B9B7A,B9c                    g ||i ddS )+Override in subclasses to find completions.ok)rT  
cursor_endcursor_startr0  r  r   )rP   r  rQ  s      r3   rR  zKernel.do_complete  s     $&
 	
rE   c                  K   | j                   sy|d   }| j                  |d   |d   |j                  dd      t        |j                  dg                   }t	        j
                  |      r| d{   }n;t        j                  t        j                  d| j                  	      t        d
       t        |      }| j                   j                  |d|||      }| j                  j                  d|       y7 w)zHandle an inspect request.Nr   r  rQ  detail_levelr   omit_sections
do_inspectrA  r   r]   inspect_replyr?  )r   r]  r:   r   r5   r   r_   r`   r*   rE  rF  r$   r  ri   r   rP   r   rk   r  r   r7  r   s          r3   r   zKernel.inspect_request  s     ||#FOL!KK*OR01	
 }-"//MMM"))L)Y) #=1llPUVtS! 0s   A0C<2C:3BC<c                    di i ddS )z.Override in subclasses to allow introspection.rW  F)r  datar0  foundr   )rP   r  rQ  r[  r\  s        r3   r]  zKernel.do_inspect  s    UKKrE   c                  K   | j                   sy|d   } | j                  d	i |}t        j                  |      r| d{   }n;t	        j
                  t        j                  d| j                        t        d       t        |      }| j                   j                  |d|||      }| j                  j                  d|       y7 w)
zHandle a history request.Nr   
do_historyrA  r   r]   history_replyr?  r   )r   rd  r5   r   r_   r`   r*   rE  rF  r$   r  ri   r   r_  s          r3   r   zKernel.history_request  s     ||#'2'2}-"//MMM"))L)Y) #=1llPUVtS! 0s   ?CC	BCc
                    dg dS )z)Override in subclasses to access history.rW  )r  historyr   )
rP   hist_access_typeoutputrawr   r   stopnpatternuniques
             r3   rd  zKernel.do_history  s     2..rE   c                   K   | j                   sy| j                  r| j                  j                         ni }d|d<   | j                   j                  |d|||      }| j                  j                  d|       yw)zHandle a connect request.NrW  r  connect_replyr?  )r   r  r   r  ri   r   rP   r   rk   r  r   r   s         r3   r   zKernel.connect_request  sf     ||151E1E$&&++-2 ll&%PtS!s   A7A9c                    ddl m} g }| j                  r|j                  d       |r|j                  d       t        | j
                  | j                  | j                  | j                  | j                  |dS )Nr   )_is_debugpy_availablezkernel subshellsdebugger)protocol_versionrW   rX   rp   rY   rq   supported_features)
rt  rs  r   appendr%   rW   rX   rp   rY   rq   )rP   rs  rv  s      r3   kernel_infozKernel.kernel_info  so    3(***%%&89 %%j1 !8"11&*&A&A!//kk//"4
 	
rE   c                   K   | j                   syddi}|j                  | j                         | j                   j                  |d|||      }| j                  j                  d|       yw)zHandle a kernel info request.Nr  rW  kernel_info_replyr?  )r   updaterx  r  ri   r   rq  s         r3   r   zKernel.kernel_info_request  s[     ||T"t''(ll(;WfeTtS!s   A)A+c                  K   | j                   sy|d   }|j                  dd      }t        | d      r]| j                  j                  j                         D ci c]-  \  }}|j                  |k(  s||t        |j                        / }}}ni }t        |d      }	| j                   j                  |d|	||      }
| j                  j                  d	|
       yc c}}w w)
zHandle a comm info request.Nr   target_namecomm_manager)r}  rW  )commsr  comm_info_replyr?  )r   r:   r  r~  r  itemsr}  r8   r  ri   r   )rP   r   rk   r  r   r}  kvr  r7  r   s              r3   r   zKernel.comm_info_request  s     ||#kk-6 4( #//55;;=Q==K/;3F 4AMM22E  E56ll(9=&RWXtS!s   AC!2CAC!c                   t         j                  dk(  r| j                  j                  d       y t        j                         }t        j
                  |      }|r1||k(  r,t        t         d      r	 t        j                  |t               y t        j                  |t               y # t        $ r t        j                  |t                w xY w)Nntz*Interrupt message not supported on Windowskillpg)osr(  ri   r   getpidgetpgidr  r  r
   OSErrorkill)rP   pidpgids      r3   _send_interrupt_childrenzKernel._send_interrupt_children	  s    77d?HHNNGH))+C::c?D H(=IIdF+
 V$	  GGC(s   1B' '%Cc                8  K   | j                   syddi}	 | j                          | j                   j                  |d|||       y# t        $ rH}ddl}d|j	                         t        t        |      j                        t        |      d}Y d}~md}~ww xY ww)	zHandle an interrupt request.Nr  rW  r   r   )r  	tracebackenameevalueinterrupt_replyrk   )	r   r  r  r  format_stackr)   type__name__r  )rP   r   rk   r  r   errr  s          r3   r   zKernel.interrupt_request  s     ||%-t$4
	))+ 	&"3WfER  	 "&335T#Y//0c(	G	s-   BA !B	B>BBBBc                  K   | j                   sy| j                  |d   d         }t        j                  |      r| d{   }n;t	        j
                  t        j                  d| j                        t        d       | j                   j                  |d|||	       | j                   j                  d||      | _        | j                          d{    | j                  j                  d
       | j                  r1| j                  j                   }|j#                  |j$                         | j                  j                  d       | j                   j#                  | j                   j$                         | j&                  rV| j&                  j                   | j                   k7  r2| j&                  j                   }|j#                  |j$                         yyy7 7 w)zHandle a shutdown request.Nr   restartdo_shutdownrA  r   r]   shutdown_replyr  zStopping control ioloopzStopping shell ioloop)r   r  r5   r   r_   r`   r*   rE  rF  r  r   _shutdown_message_at_shutdownri   r   r   r   rN   rk  rb   )rP   r   rk   r  r   control_io_loopshell_io_loops          r3   r   zKernel.shutdown_request.  s}    ||""6)#4Y#?@w'#mGMM"))M$JZJZ)[)
 	&"2GV5Q!%!1!12BGV!T!!!01"1199O(()=)=>./!!$,,"3"34!2!2!:!:dll!J --55M&&}'9'9: "K) $ 	"s%   ?G'G!BG'G$D
G'$G'c                    d|dS )z]Override in subclasses to do things when the frontend shuts down the
        kernel.
        rW  )r  r  r   )rP   r  s     r3   r  zKernel.do_shutdownL  s     733rE   c                  K   | j                   sy|d   }|d   }| j                  |      }t        j                  |      r| d{   }n;t	        j
                  t        j                  d| j                        t        d       t        |      }| j                   j                  |d|||      }| j                  j                  d	|       y7 w)
zHandle an is_complete request.Nr   r  do_is_completerA  r   r]   is_complete_replyr?  )r   r  r5   r   r_   r`   r*   rE  rF  r$   r  ri   r   )rP   r   rk   r  r   r  r7  rL  s           r3   r   zKernel.is_complete_requestR  s     ||#v++D1}-"//MMM"))4DTM`M`)a)
 #=1LL%%f.A=RXZ_`	tY' 0s   ACCBCc                
    ddiS )rV  r  unknownr   )rP   r  s     r3   r  zKernel.do_is_completef  s    )$$rE   c                  K   | j                   sy|d   }| j                  |      }t        j                  |      r| d{   }n;t	        j
                  t        j                  d| j                        t        d       t        |      }| j                   j                  |d|||      }| j                  j                  d|       y7 w)	zHandle a debug request.Nr   do_debug_requestrA  r   r]   debug_replyr?  )r   r  r5   r   r_   r`   r*   rE  rF  r$   r  ri   r   )rP   r   rk   r  r   r7  rL  s          r3   r   zKernel.debug_requestj  s     ||#--g6}-"//MMM"))09N9N *  * #=1LL%%fm]FTYZ	tY' 0s   >C
 CBC
c                d    	  t        ||             }|t        ||      S |S # t        $ r Y yw xY w)zGet the process metric value.r   )r   BaseException)rP   processr(  	attributemetric_values        r3   get_process_metric_valuezKernel.get_process_metric_value~  sE    		177D13L$|Y77  		s   # # 	//c           
       K   | j                   syt        j                         t        j                         d}t        j                         }|g|j                  d      }|D ci c]3  }|j                  | j                  j                  |j                  |      5 c}| _	        t        | j                  j                         D cg c]  }| j                  |dd       c}      |d<   t        |j                         d      rdnd}t        | j                  j                         D cg c]  }| j                  |d	|       c}      |d
<   t        j                          }	|	
|	dk7  r|	|d<   t        j"                  d      |d<   t%        t        j&                         j)                               |d<   | j                   j+                  |d|||      }
| j,                  j/                  d|
       yc c}w c c}w c c}w w)zHandle a usage request.N)hostnamer  T	recursivecpu_percent
kernel_cpupssrssmemory_full_infokernel_memoryr   host_cpu_percent)logical	cpu_counthost_virtual_memoryusage_replyr?  )r   r  gethostnamer  r  psutilProcesschildrenr  rH   r:   sumr>   r  r  r  r  r  r8   virtual_memory_asdictr  ri   r   )rP   r   rk   r  r7  current_processall_processesr  mem_info_typer  rL  s              r3   r   zKernel.usage_request  s    ||%+%7%7%9"))+N ..*(T?+C+Cd+C+ST
 )
 KK++GKKAA
 '*  $~~446 --g}dK'
l# "))I)I)KU!SY^),  $~~446 --g7I=Y*
o& ((* "{c'90;M,-%+%5%5d%Ck"/3F4I4I4K4S4S4U/V+,LL%%fm]FTYZ	tY'3

s,   A$H &8G1)H G6!A	H *G;B<H c                   K   t         wr.   rN  r   s     r3   r  zKernel.do_debug_request  s     !!s   	c                  K   | j                   sy| j                  s| j                  j                  d       yt	        j
                         j                  t        k(  sJ | j                  j                  j                  j                  }|j                  ddi       |j                         }| j                   j                  |d|||       yw)zGHandle a create subshell request.

        .. versionadded:: 7
        N*Subshells are not supported by this kernelr  createcreate_subshell_replyr   r   ri   r   r   r   r(  r   r   r   control_to_shell_channelfrom_socket	send_json	recv_jsonr  rP   r  rk   r  control_socketreplys         r3   r   zKernel.create_subshell_request  s     
 ||..HHNNGH'')..2EEEE 22::SS__  &(!34((*&"95&%P   C
Cc                
  K   | j                   sy| j                  s| j                  j                  d       yt	        j
                         j                  t        k(  sJ 	 |d   }|d   }| j                  j                  j                  j                  }|j                  d|d       |j                         }| j                   j                  |d|||       y# t        $ r | j                  j                  d|       Y yw xY ww)	zGHandle a delete subshell request.

        .. versionadded:: 7
        NzKERNEL SUBSHELLS NOT SUPPORTEDr   r   zGot bad msg from parent: %sdelete)r  r   delete_subshell_reply)r   r   ri   r   r   r   r(  r   r   r   r   r  r  r  r  r  )rP   r  rk   r  r   r   r  r  s           r3   r   zKernel.delete_subshell_request  s     
 ||..HHNN;<'')..2EEEE	Y'G!-0K 22::SS__  (;!OP((*&"95&%P  	HHNN8&A	s+   AD 
C *A.D%D =D?D  Dc                  K   | j                   sy| j                  s| j                  j                  d       yt	        j
                         j                  t        k(  sJ | j                  j                  j                  j                  }|j                  ddi       |j                         }| j                   j                  |d|||       yw)zEHandle a list subshell request.

        .. versionadded:: 7
        Nr  r  listlist_subshell_replyr  r  s         r3   r   zKernel.list_subshell_request  s     
 ||..HHNNGH'')..2EEEE 22::SS__  &&!12((*&"7Nr  c                J    d| j                   z  }| d| j                         S )z!prefixed topic for IOPub messagesz	kernel.%s.)rk   encode)rP   topicbases      r3   r  zKernel._topic  s*    TZZ'&%!))++rE   c                    | j                   j                  }|j                  |      }|J d} |j                  |d       y)znPost a dummy message to the correct subshell that when handled will unset
        the _aborting flag.
        Nr   Fr   )r   r   r  r  )rP   r   r  r  r   s        r3   !_post_dummy_stop_aborting_messagez(Kernel._post_dummy_stop_aborting_message  sG      44<<!FF{S!!!Ce$rE   c                    |d _         n& j                  j                  j                  |d        j                  j                  d        j                  r1 j                  j                  j                   j                  |       y  j                  r& j                  s j                  j                           fd} j                  rs j                  j                   j                  |        j                  @t         j                  d      r) j                  j!                   j                  dz          y y y  j                  j                  |       y )NTzAborting queuec                 R   K    j                   j                  d       d _        y w)NzFinishing abortF)ri   r   r   rc   s   r3   stop_abortingz+Kernel._abort_queues.<locals>.stop_aborting&  s     HHMM+,"DNs   $'_schedule_exitr~   )r   r   r   r   ri   r   r   rN   r  rb   r   r   stop_on_error_timeoutr   rI   r  r  )rP   r   r  s   `  r3   rJ  zKernel._abort_queues  s    !DN%%--CCKQUV&'$$%%--::66  T%D%D##%	# %% LL##D$>$>N
 ~~)gdnnFV.W--d.H.H4.OP /X) LL%%m4rE   c                `   | j                   sy| j                  j                  d|d   d   |d   d          |d   d   j                  dd      d   d	z   }d
di}| j	                  |      }| j                  |||      }|j                  |       | j                   j                  ||||||       y)z"Send a reply to an aborted requestNzAborting %s: %sr   r   r   r  r   r   _replyr  aborted)r0  r   r  rk   )r   ri   r   rsplitr5  r8  r{  r  )rP   r   r   r   
reply_typer  mds          r3   r   zKernel._send_abort_reply7  s    ||'Xx)@#h-PZB[\]:.55c1=a@8K
I&$!!#r62
		& 	 	
rE   c                    d}t        |      )zPRaise StdinNotImplementedError if active frontend doesn't support
        stdin.z?raw_input was called, but this frontend does not support stdin.r   r   s     r3   _no_raw_inputzKernel._no_raw_inputK  s     P&s++rE   c                    | j                   sd}t        |      |ddl} |j                  dt        d       | j                  || j                  | j                        | j                  d      d	      S )
zForward getpass to frontends

        Raises
        ------
        StdinNotImplementedError if active frontend doesn't support stdin.
        zFgetpass was called, but this frontend does not support input requests.Nr   zTThe `stream` parameter of `getpass.getpass` will have no effect when using ipykernelr\   r]   r   Tpassword)	_allow_stdinr   r_   r`   UserWarning_input_requestr   rw   r|   )rP   promptr   r   r_   s        r3   getpasszKernel.getpassQ  s}       ZC*3//HMMf
 ""''(@(@AOOG$	 # 
 	
rE   c                    | j                   sd}t        |      | j                  t        |      | j	                  | j
                        | j                  d      d      S )zForward raw_input to frontends

        Raises
        ------
        StdinNotImplementedError if active frontend doesn't support stdin.
        zHraw_input was called, but this frontend does not support input requests.r   Fr  )r  r   r  r)   r   rw   r|   )rP   r  r   s      r3   	raw_inputzKernel.raw_inputj  s_       \C*3//""K''(@(@AOOG$	 # 
 	
rE   c                
   t         j                  t         j                  j                          t         j                  t         j                  j                          	 	 | j                  j                  t        j                         +# t        j                  $ r(}|j                  t        j                  k(  rY d }~n	 d }~ww xY w| j                  J t        t        ||            }| j                  j                  | j                  d|||       	 	 t        j                  | j                  gg | j                  gd      \  }}}	|s|	r0| j                  j!                  | j                        \  }}
||
fdk7  rnDnB# t"        $ r d}t#        |      d t$        $ r  | j&                  j)                  dd	       Y nw xY w	 |
d
   d   }n-# t$        $ r! | j&                  j+                  d|       d}Y nw xY w|dk(  rt,        |S )NT)r  r  input_requestr  r~   )NNzInterrupted by userzInvalid Message:r   r   valuezBad input_reply: %s )r   r   r   r   stdin_socketrecv_multipartr   NOBLOCKZMQErrorerrnoEAGAINr   r$   r8   r  selectrecvr   r   ri   r   r   EOFError)rP   r  rk   r  r  er   rlistr  xlistr  r   r  s                r3   r  zKernel._input_request{  s   ::!JJ::!JJ !!00=  << 77cjj( ||'''T(CD$++_gvUZ[ D #&**d.?.?-@"tGXGXFY[_"`q%E#'<<#4#4T5F5F#GLE5u~5$ 7+',$6 D  !3d CD# (	)$W-E 	HHNN0&9E	 F?NsC   )B	 	CB?>B??CA%E> ><F=<F=G 'G54G5c                >   t        |      }| j                         D ]e  }| j                  j                  d||       	 |t        k(  r|j                          n+|t        k(  r|j                          n|j                  |       g y# t        j                  $ r Y ~w xY w)z
        Send a signal to all our children

        Like `killpg`, but does not include the current process
        (or possible parents).
        zSending %s to subprocess %sN)r   _process_childrenri   r   r   	terminater   r  send_signalr  NoSuchProcess)rP   signumsig_repr2   s       r3   _signal_childrenzKernel._signal_children  s     V_''') 
	AHHNN8'1EW$KKMw&FFHMM&)
	 '' s   ABBBc                B   t        j                         }|j                  d      }t        j                  dk(  r|S t        j
                         }g }|D ]9  }	 t        j                  |j                        }||k(  s)|j                  |       ; |S # t        $ r Y Iw xY w)zRetrieve child processes in the kernel's process group

        Avoids:
        - including parents and self with killpg
        - including all children that may have forked-off a new group
        Tr  r  )
r  r  r  r  r(  getpgrpr  r  rw  r  )rP   kernel_processall_childrenkernel_pgidprocess_group_childrenchild
child_pgids          r3   r  zKernel._process_children  s      )%...>77d?jjl!#! 	9E9ZZ		2
 ,*11%8	9 &%  s   B	BBc                  K   d}| j                         s| j                  j                  d       y t        t        fD ]  }|D ]}  }| j                         }|s| j                  j                  d         y | j                  |       | j                  j                  d||       t        j                  |       d {      y 7 	w)N)r~   gQ?g?g333333?r   r   r   zKernel has no children.z.No more children, continuing shutdown routine.z?Will sleep %s sec before checking for children and retrying. %s)r  ri   r   r   r   r  r   rI  )rP   sleepsr  delayr  s        r3   %_progressively_terminate_all_childrenz,Kernel._progressively_terminate_all_children  s     1%%'HHNN45( 	+F +113HHNN#ST%%f-U
 mmE***+	+ +s   B9C;C<
Cc           	       K   	 | j                          d{    | j                  s| j
                  rg| j
                  j                  | j                  | j                  | j                  d             | j                  j                  d| j                         | j                  r*| j                  j                  t        j                         yy7 # t        $ r&}| j                  j                  d|       Y d}~d}~ww xY w# | j                  s| j
                  rg| j
                  j                  | j                  | j                  | j                  d             | j                  j                  d| j                         | j                  r*| j                  j                  t        j                         w w xY ww)zCActions taken at shutdown by the kernel, called by python's atexit.Nz,Exception during subprocesses termination %sshutdownr  r?  )r  r   ri   	exceptionr  r   r  r  r  r   r   r   r   r!  )rP   r	  s     r3   r  zKernel._at_shutdown  s^    	7<<>>>
 %%1dll!!%%**++j1 " 
 tT%;%;<""##))#++6 # ? 	RHHMqQQ	R %%1dll!!%%**++j1 " 
 tT%;%;<""##))#++6 #sK   G C CC B6G C 	DC>9D >DD B7F==G c                    | j                   d uS r.   )r   rc   s    r3   r   z!Kernel._supports_kernel_subshells  s    ((44rE   r.   )r   
str | None)r   )r+  zContextVar[T]returnr+   )NNNFNNN)TNF)r   r   )NNNNNF)r%  None)r   r$  r%  r&  )r  N)r  )F){r  
__module____qualname____doc__r   rI   rH   __annotations__r"   rS   r   r   r   profile_dirr#   rb   r   rV   r!   rd   rh   r   debug_shell_socketr   r   r  iopub_threadr   loggingLoggerri   r   int_idr    rk   ro   rp   rq   r   tagdebug_just_my_code_darwin_app_napr  r   rs   ru   propertyr}   r   rG  _poll_intervalr  r  r  r  r   r   r   r   r   r   r   r   r   r   r   r   rO   r   r   r   r   r  r  r   r   r%  r   r|   r   r1  r5  r8  r   r   r   rR  r   r]  r   rd  r   rx  r   r   r  r   r   r  r   r  r   r  r   r  r   r   r   r  r   r  rJ  r   r  r  r  r  r  r  r  r  r   __classcell__)r   s   @r3   rG   rG   Z   sP     D	I+-I(-[4 4 w40G?DQKI$7L!%"M;  K_  _. . iD9NUN55L5L5L"7>>dCCC R[FIEW! !
 (*M$) (,vJ$-
  
cc	   
cc  ;L(,RO%0#&5&--** '& %$! ! 6]&&d&3N
 4[__D_1N!
*  fO O W\\*MI	 	 	"	
 	" 	 *1X,#OJ	`@DH
2#J,28A.D4%



&	
1$?8
,   
D	
T,t " "L*
"4L"2 
/" 
 
&""(%"(;<4((%(("(H"Q(Q8O2, UI	%'5R
(,
2
"2h(&.+*7$ 5 5rE   rG   )Jr)  
__future__r   r   r5   r.  r  r  r   r   rH  typingtrm   r_   collections.abcr   contextvarsr   r   r   r   	functoolsr	   r   r
   r   r   r   threadr   platformr   jupyter_client.sessionr   r4  ImportErrorr  r   IPython.core.errorr   r   tornador   traitlets.config.configurabler   traitlets.traitletsr   r   r   r   r   r   r   r    r!   r"   zmq.eventloop.zmqstreamr#   ipykernel.jsonutilr$   _versionr%   iostreamr&   utilsr'   r(   r*   r*  TypeVarr+   rD   rG   r   rE   r3   <module>rK     s    ? #    	  
      # 9 9   H H '<<7(G4
  
 7 *  ?   . ) -  .K C  AIIcNi5" i5o  
,,Cs   ?D DD