Format: application/x-python-code
1 o
2
3 �h�� � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddl Z ddl
4 Z
5 ddl Z ddl Z ddl
6 Z
7 ddlZddlZddlZddlmZ ddlmZ ddlmZmZmZmZmZ ddlmZ e j d krhdd
8 lm Z nd Z zdd l m Z W n e!y~ ej Z Y nw ddl"Z"ddl#Z#dd
9 l$m%Z% ddl m&Z& ddl'm(Z( ddl)m*Z*m+Z+ ddl,m-Z- ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9 ddl:m;Z; ddl<m=Z= ddl>m?Z? ddl@mAZA dd� ZBG dd� de-�ZCdS )z9Base class for a kernel that talks to frontends over 0MQ.� )� annotationsN)�datetime)�partial)�SIGINT�SIGTERM�Signals�default_int_handler�signal� )�CONTROL_THREAD_NAME�win32)�SIGKILLzwindown-SIGKILL-sentinel)�utcnow��StdinNotImplementedError)�Session)�ioloop)�Queue�
10 QueueEmpty)�SingletonConfigurable) �Any�Bool�Dict�Float�Instance�Integer�List�Set�Unicode�default�observe)� ZMQStream)�
11 json_clean)�kernel_protocol_version)� OutStreamc C sb t �| �j}dd� |D �}|D ] }|�|�}|o |j|j|jfv p+tdd� |�� D ��||< q|S )Nc S s i | ]}|d �qS �F� )�.0�paramr& r& �Jc:\Users\shubh\Videos\agent\nova\Lib\site-packages\ipykernel/kernelbase.py�
12 <dictcomp>E s z'_accepts_parameters.<locals>.<dictcomp>c s s � | ] }|j |jkV qd S �N)�kind� VAR_KEYWORD)r' �pr& r& r) � <genexpr>L s � z&_accepts_parameters.<locals>.<genexpr>) �inspect� signature�
13 parameters�getr, � KEYWORD_ONLY�POSITIONAL_OR_KEYWORD�any�values)�methZ param_namesr2 Zacceptsr( Z
14 param_specr& r& r) �_accepts_parametersC s
15 ��r9 c s< e Zd ZU dZed�Zi Zded< ed�dd� �Z e
16 e dd �Z e
17 d
18 dd �Z
19 e
20 edd �Zed d �Zd
21 ed< ded< ded< ded< ed�dd� �Zed�dd� �Ze
22 edd �Ze� Ze� Ze� Ze� Ze� Ze
23 ej dd �Z ded< e d�Z e � Z!ed�dd � �Z"i Z#d ed < e� Z$d ed < e%dd!d �j&dd"�Z'e%dd#d �j&dd"�Z(e%d$�Z)e*i i d%��Z+d&ed'< e*d(d(d%��Z,e-d)d*� �Z.e/d+�j&dd"�Z0e/d,�j&dd"�Z1e/d-dd.d/�Z2dZ3e*� Z4e5� Z6d0Z7g d1�Z8g e8�d2�d3�d4�d5�Z9� fd6d7�Z:d8d9� Z;d:d;� Z<d<d=� Z=d>d?� Z>d@dA� Z?dBdC� Z@dDdE� ZAdFdG� ZBdHdI� ZCdJdK� ZDd�dLdM�ZEdNdO� ZFedPd �ZGedQ�dRdS� �ZHdTdU� ZIdVdW� ZJdXdY� ZKdZd[� ZLd�d\d]�ZMd^d_� ZNd�dadb�ZOd�dcdd�ZP $ d�dedf�ZQdgdh� ZRdidj� ZSdkdl� ZT $d�dddm�dndo�ZUdpdq� ZVdrds� ZWdtdu� ZXd�dwdx�ZYdydz� ZZ $d�d{d|�Z[d}d~� Z\e-dd�� �Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�� Zbd�d�� Zcd�d�� Zdd�d�� Zed�d�� Zfd�d�d��Zgd�d�� Zhd�d�� Zid�d�� Zjd�d�� Zkd�d�� Zld�d�� Zmd�d�� Znd�d�� Zoe%d$�Zpd�d�� Zqd�d�� Zrd�d�� Zsd�d�d��Ztd�d�d��Zud�d�d��Zvd�d�� Zwd�d�� Zxd�d�� Zyd�d�� Zz� Z{S )��KernelzThe base kernel class.Nzdict[str, psutil.Process]� processes� eventloopc C s( t j�� }|jdur|�| j� dS dS )z&schedule call to eventloop from IOLoopN)r �IOLoop�current�new� add_callback�enter_eventloop)�self�change�loopr& r& r) �_update_eventloop] s
24
25 �zKernel._update_eventloopT)�
26 allow_nonez"IPython.core.profiledir.ProfileDirz�Deprecated shell_streams alias. Use shell_stream
27
28 .. versionchanged:: 6.0
29 shell_streams is deprecated. Use shell_stream.
30 )�helpz List[t.Any]�
31 shell_streams�str�implementation�implementation_version�bannerc C s& t jdtdd� | jd ur| jgS g S )N�LKernel.shell_streams is deprecated in ipykernel 6.0. Use Kernel.shell_stream� ��
32 stacklevel)�warnings�warn�DeprecationWarning� shell_stream�rB r& r& r) �_shell_streams_defaultt s �
33 z Kernel._shell_streams_defaultc C sH t jdtdd� t|j�dkrt jdtdd� |jr"|jd | _d S d S )NrM rN rO r
34 zJKernel only supports one shell stream. Additional streams will be ignored.r )rQ rR rS �lenr? �RuntimeWarningrT )rB rC r& r& r) �_shell_streams_changed s ���z Kernel._shell_streams_changedzlogging.Logger�log������identc C s t t�� �S r+ )rI �uuid�uuid4rU r& r& r) �_default_ident� s zKernel._default_identzdict[str, object]�
35 language_infozList[dict[str, str]]�
36 help_linkszSSet to False if you want to debug python standard and dependent libraries.
37 )�configzhWhether to use appnope for compatibility with OS X App Nap.
38
39 Only affects OS X >= 10.9.
40 F)�shell�controlzDict[str, t.Any]�_parents� c C s t jdtdd� | �� S )NzEKernel._parent_header is deprecated in ipykernel 6. Use .get_parent()rN rO )rQ rR rS �
41 get_parentrU r& r& r) �_parent_header� s �zKernel._parent_headerg����Mb@?�{�G�z�?� a� time (in seconds) to wait for messages to arrive
42 when aborting queued requests after an error.
43
44 Requests that arrive within this window after an error
45 will be cancelled.
46
47 Increase in the event of unusually slow network
48 causing significant delays,
49 which can manifest as e.g. "Run all" in a notebook
50 aborting some, but not all, messages after an error.
51 )rb rG r ) �execute_request�complete_request�inspect_request�history_request�comm_info_request�kernel_info_request�connect_request�shutdown_request�is_complete_request�interrupt_request�
52 apply_request�
53 clear_request�
54 abort_request�
55 debug_request�
56 usage_requestc s� t � jdi |�� tj| _tj| _i | _| jD ]
57 }t | |�| j|< qi | _
58 | j D ]
59 }t | |�| j
60 |< q(t � | _
61 t| jddg�| _dS )zInitialize the kernel.� cell_meta�cell_idNr& )�super�__init__�sys�stdoutZ_stdout�stderr�_stderr�shell_handlers� msg_types�getattr�control_handlers�control_msg_typesr �
62 control_queuer9 �
63 do_execute�_do_exec_accepted_params)rB �kwargs�msg_type�� __class__r& r) r} s
64
65
66
67 �zKernel.__init__c C s | j �|� d S r+ )r� �
68 put_nowait�rB �msgr& r& r) �dispatch_control+ s zKernel.dispatch_controlc � sF � | j �� I d H }t|tjjtjf�r|�d � q| �|�I d H qr+ ) r� r3 �
69 isinstance�
70 concurrent�futures�Future�asyncio�
71 set_result�process_controlr� r& r& r) �poll_control_queue. s �
72 �zKernel.poll_control_queuec � sT �� j r� j j}tj�� �t���}n � j}t�� �}� �fdd�}|�|� |S )zDFlush the control queue, wait for processing of any pending messagesc s � j r� j �� � j��� d S r+ )�control_stream�flushr� �putr& �rB Z
73 tracer_futurer& r) �_flushD s
74 z+Kernel._flush_control_queue.<locals>._flush)�control_thread�io_loopr� r� r� r� � wrap_futurer@ )rB � control_loopZawaitable_futurer� r& r� r) �_flush_control_queue7 s �
75
76 zKernel._flush_control_queuec � sB �| j sdS | j j|dd�\}}z | j j|ddd�}W n ty- | jjddd� Y dS w | j�d|� | j||d d
77 � | �d d � |d }|d
78 }| j �
79 |d�}|du r^| j�d|� n%z|| j ||�}t �
80 |�rp|I dH W n ty� | jjddd� Y nw tj�� tj�� | �dd � | j r�| j �tj� dS dS )zdispatch control requestsNF��copyT��contentr� zInvalid Control Message��exc_infozControl received: %srd ��channel�busy�headerr� z UNKNOWN CONTROL MESSAGE TYPE: %rz Exception in control handler:�idle)�session�feed_identities� deserialize� ExceptionrZ �error�debug�
81 set_parent�_publish_statusr� r3 r� r0 � isawaitabler~ r r� r� �zmq�POLLOUT)rB r� �identsr� r� �handler�resultr& r& r) r� O s@ � �
82
83 � �
84
85 �zKernel.process_controlc C s8 |d d }|| j v r| j �|� | �|||� dS dS )z�Check whether a shell-channel message should be handled
86
87 Allows subclasses to prevent handling of certain messages (e.g. aborted requests).
88 r� �msg_idFT)�aborted�remove�_send_abort_reply)rB �streamr� r� r� r& r& r) �
89 should_handleu s
90 zKernel.should_handlec � sz �| j sdS | �� I dH | j j|dd�\}}z | j j|ddd�}W n ty4 | jjddd� Y dS w | j||dd � | �d
91 d� |d d }| j rj|d
92 krj| �
93 | j ||� | �dd� | j rh| j � t
94 j� dS | j�d|� | j�d|d � | �| j ||�s�dS | j�|d�}|du r�| j�d|� n�| j�d||� z| �� W n ty� | jjddd� Y nw zMz|| j ||�}t�|�r�|I dH W n ty� | jjddd� Y n ty� | j�d� Y nw W z| �� W n+ t�y | jjddd� Y nw z| �� W w t�y | jjddd� Y w w tj� � tj� � | �dd� | j �r;| j � t
95 j� dS dS )zdispatch shell requestsNFr� Tr� zInvalid Messager� rc r� r� r� r� rk r� z
96 *** MESSAGE TYPE:%s***z Content: %s
97 --->
98 r� zUnknown 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:) r� r� r� r� r� rZ r� r� r� � _abortingr� rT r� r� r� r� r� r� r3 �warning�pre_handler_hookr0 r� �KeyboardInterrupt�post_handler_hookr~ r r� )rB r� r� r� r� r� r& r& r) �dispatch_shell� sx � � �
99
100 � �� �� �
101
102 �zKernel.dispatch_shellc C s t tt�| _dS )z.Hook to execute before calling message handlerN)r r r �saved_sigint_handlerrU r& r& r) r� � s zKernel.pre_handler_hookc C s t t| j� dS )z-Hook to execute after calling message handlerN)r r r� rU r& r& r) r� � s zKernel.post_handler_hookc sV �j �d�j� �j��du r�j �d� dS ���fdd�� � �fdd���� dS )zenter eventloopzEntering eventloop %sNz Exiting as there is no eventloopc � s� ��j � ur�j�d� � d S �j�� r �j�d� �� d S �j�d� � z� �� W n ty; �j�d� Y nw �j � u rF�� d S d S )Nzexiting eventloop %sz*Delaying eventloop due to waiting messageszAdvancing eventloop %sz"KeyboardInterrupt caught in kernel)r< rZ �info� msg_queue�qsizer� r� r� r& )r< �
103 schedule_nextrB r& r) �advance_eventloop� s" �
104
105 �
106
107 �z1Kernel.enter_eventloop.<locals>.advance_eventloopc s& �j �d� �j�dt�j� �� dS )z*Schedule the next advance of the eventloopz Scheduling eventloop advanceg����MbP?N)rZ r� r� �
108 call_laterr �schedule_dispatchr& )r� rB r& r) r� � s z-Kernel.enter_eventloop.<locals>.schedule_next)rZ r� r< rU r& )r� r< r� rB r) rA � s
109 zKernel.enter_eventloopc � s( �| j r | j �� | jdd�I dH dS )z�Process a single shell message
110
111 Any pending control messages will be flushed as well
112
113 .. versionchanged:: 5
114 This is now a coroutine
115 F)�waitN)rT r� � process_onerU r& r& r) �do_one_iteration� s �
116 zKernel.do_one_iterationc � s` �|r| j �� I dH \}}}nz
117 | j �� \}}}W n
118 tjtfy& Y dS w ||� I dH dS )zMProcess one request
119
120 Returns None if no message was handled.
121 N)r� r3 �
122 get_nowaitr� r )rB r� �t�dispatch�argsr& r& r) r�
123 s ��zKernel.process_onec � s8 � z | � � I dH W n ty | j�d� Y nw q)z�Coroutine to preserve order of message handling
124
125 Ensures that only one message is processing at a time,
126 even when the handler is async
127 TNzError in message handler)r� r� rZ � exceptionrU r& r& r) �dispatch_queue s � ��zKernel.dispatch_queuez&Monotonic counter of messages
128 �_message_counterc C s t �� S r+ )� itertools�countrU r& r& r) �_message_counter_default* � z Kernel._message_counter_defaultc G s0 t | j�}| j�|||f� | j�dd� � dS )z schedule a message for dispatchc S s d S r+ r& r& r& r& r) �<lambda>: s z*Kernel.schedule_dispatch.<locals>.<lambda>N)�nextr� r� r� r� r@ )rB r� r� �idxr& r& r) r� . s
129 ��zKernel.schedule_dispatchc C s� t j�� | _t� | _| j�| j� | jr | jj | j
130 dd� | j r$| j jn| j}t �
131 | �� |j� | jr@| jj t| j| j�dd� | �dd� dS )z register dispatchers for streamsFr� �startingrc N)r r= r> r� r r� r@ r� r� Zon_recvr� r� r� �run_coroutine_threadsafer� Z asyncio_looprT r r� r� r� )rB r� r& r& r) �start< s �� z Kernel.startc C s
132 || _ dS )z�Record the ports that this kernel is using.
133
134 The creator of the Kernel instance must call this methods if they
135 want the :meth:`connect_request` method to return the port numbers.
136 N)�_recorded_ports)rB Zportsr& r& r) � record_portsT s
137 zKernel.record_portsc C s2 | j sdS | j j| jd||d�|| �d�d� dS )z-Publish the code request on the iopub stream.NZ
138 execute_input)�code�execution_count��parentr\ )r� �send� iopub_socket�_topic)rB r� r� r� r& r& r) �_publish_execute_input` s
139 �z Kernel._publish_execute_inputc C s: | j sdS | j j| jdd|i|p| �|�| �d�d� dS )z send status (busy/idle) on IOPubN�statusZexecution_stater� �r� r� r� rg r� )rB r� r� r� r& r& r) r� l s
140 �zKernel._publish_statusc C s0 | j sd S | j j| jd|| �� | �d�d� d S )NZ debug_eventr� r� )rB �eventr& r& r) �_publish_debug_eventx s
141 �zKernel._publish_debug_eventrc c C s || j |< || j|< dS )a
142 Set the current parent request
143
144 Side effects (IOPub messages) and replies are associated with
145 the request that caused them via the parent_header.
146
147 The parent identity is used to route input_request messages
148 on the stdin channel.
149 N)�
150 _parent_identre )rB r\ r� r� r& r& r) r� � s
151 zKernel.set_parentc C s. |du rt �� jtkrd}nd}| j�|i �S )aR Get the parent request associated with a channel.
152
153 .. versionadded:: 6
154
155 Parameters
156 ----------
157 channel : str
158 the name of the channel ('shell' or 'control')
159
160 Returns
161 -------
162 message : dict
163 the parent message for the most recent request on the channel.
164 Nrd rc )� threading�current_thread�namer re r3 )rB r� r& r& r) rg � s
165 zKernel.get_parentc
166
167 C s, | j sdS | j �|||| �| �|||||� S )a Send a response to the message we're currently processing.
168
169 This accepts all the parameters of :meth:`jupyter_client.session.Session.send`
170 except ``parent``.
171
172 This relies on :meth:`set_parent` having been called for the current
173 message.
174 N)r� r� rg )
175 rB r� Z msg_or_typer� r\ �buffers�trackr� �metadatar� r& r& r) �
176 send_response� s �zKernel.send_responsec C s
177 dt � iS )zRInitialize metadata.
178
179 Run at the beginning of execution requests.
180 �started)�now)rB r� r& r& r) �
181 init_metadata� s �zKernel.init_metadatac C s |S )zXFinish populating metadata.
182
183 Run after completing an execution request.
184 r& )rB r� r� �
185 reply_contentr& r& r) �finish_metadata� s zKernel.finish_metadatac � s� �| j sdS z.|d }|d }|�dd�}|�d| �}|�di �}|�dd�} |�d i �}
186 |
187 �d
188 �} W n tyK | j�d � | j�d |� Y dS w |�d
189 d�} | �|�}
190 |sh| jd7 _| �||| j� ||||| d�}| jd ry|
191 |d< | jd r�| |d< | j di |��}t
192 � |�r�|I dH }t j
193 �� t j�� | jr�t�| j� t|�}| �||
194 |�}
195 | j j|d|||
196 |d�}| j�d |� |s�|d d dkr�| r�| �� dS dS dS dS )zhandle an execute_requestNr� r� �silentF�
197 store_history�user_expressions� allow_stdinr� ZcellIdz
198 Got bad msg: �%s�
199 stop_on_errorTr
200 )r� r r r r rz r{ Z
201 execute_reply)r� r\ r� r� r& )r� r3 r� rZ r� r� r� r� r� r� r0 r� r~ r r� r� �_execute_sleep�time�sleepr" r� r� r� �
202 _abort_queues)rB r� r\ r� r� r� r r r r rz r{ r r� Zdo_execute_argsr� � reply_msgr& r& r) rk � sj � �
203 �
204
205
206
207
208
209 � �zKernel.execute_request)rz r{ c C � t �)z4Execute user code. Must be overridden by subclasses.��NotImplementedError)rB r� r r r r rz r{ r& r& r) r� ' s zKernel.do_executec � sd �| j sdS |d }|d }|d }| �||�}t�|�r"|I dH }t|�}| j �|d|||� dS )z Handle a completion request.Nr� r� �
210 cursor_posZcomplete_reply)r� � do_completer0 r� r" r� )rB r� r\ r� r� r� r �matchesr& r& r) rl 5 s �
211
212 zKernel.complete_requestc C s g ||i dd�S )�+Override in subclasses to find completions.�ok)r Z
213 cursor_endZ cursor_startr� r� r& )rB r� r r& r& r) r D s �zKernel.do_completec
214 � s� �| j sdS |d }| �|d |d |�dd�t|�dg ���}t�|�r*|I dH }t|�}| j �|d|||�}| j� d |� dS )
215 zHandle an inspect request.Nr� r� r � detail_levelr �
216 omit_sectionsZ
217 inspect_replyr )
218 r� �
219 do_inspectr3 �setr0 r� r" r� rZ r� �rB r� r\ r� r� r� r� r& r& r) rm N s �
220 �
221
222 zKernel.inspect_requestr& c C s di i dd�S )z.Override in subclasses to allow introspection.r F)r� �datar� �foundr& )rB r� r r r r& r& r) r b s zKernel.do_inspectc � sf �| j sdS |d }| jdi |��}t�|�r |I dH }t|�}| j �|d|||�}| j�d|� dS )zHandle a history request.Nr� Z
223 history_replyr r& )r� �
224 do_historyr0 r� r" r� rZ r� r r& r& r) rn f s �
225
226 zKernel.history_requestc
227
228 C s
229 dg d�S )z)Override in subclasses to access history.r )r� �historyr& )
230 rB Zhist_access_type�output�rawr� r� �stop�n�pattern�uniquer& r& r) r t s
231
232 zKernel.do_historyc � sN �| j sdS | jr| j�� ni }d|d< | j �|d|||�}| j�d|� dS )zHandle a connect request.Nr r� Z
233 connect_replyr )r� r� r� r� rZ r� �rB r� r\ r� r� r� r& r& r) rq � s �zKernel.connect_requestc C s t | j| j| j| j| jd�S )N)Zprotocol_versionrJ rK r` rL ra )r# rJ rK r` rL ra rU r& r& r) � kernel_info� s �zKernel.kernel_infoc � sF �| j sdS ddi}|�| j� | j �|d|||�}| j�d|� dS )z Handle a kernel info request.Nr� r Zkernel_info_replyr )r� �updater# r� rZ r� r" r& r& r) rp � s � zKernel.kernel_info_requestc � s| �| j sdS |d }|�dd�� t| d�r#� fdd�| jj�� D �}ni }t|dd�}| j �|d |||�}| j� d
234 |� dS ) zHandle a comm info request.Nr� � target_name� comm_managerc s0 i | ]\}}|j � ks� d u r|t|j d��qS )N�r% )r% �dict)r' �k�vr' r& r) r* � s
235 �z,Kernel.comm_info_request.<locals>.<dictcomp>r )�commsr� Zcomm_info_replyr )
236 r� r3 �hasattrr& r+ �itemsr( r� rZ r� )rB r� r\ r� r� r+ r� r� r& r' r) ro � s �
237
238
239 � zKernel.comm_info_requestc C s� t jdkr
240 | j�d� d S t �� }t �|�}|r9||kr9tt d�r9z t �|t� W d S t y8 t �
241 |t� � w t �
242 |t� d S )N�ntz*Interrupt message not supported on Windows�killpg) �osr� rZ r� �getpid�getpgidr, r/ r �OSError�kill)rB �pidZpgidr& r& r) �_send_interrupt_children� s
243
244 �z Kernel._send_interrupt_childrenc
245 � s� �| j sdS ddi}z| �� W n& ty6 } zddl}d|�� tt|�j�t|�d�}W Y d}~nd}~ww | j j|d|||d� dS ) z Handle an interrupt request.Nr� r r r� )r� � tracebackZenameZevalueZinterrupt_reply�r\ ) r� r6 r3 r7 � format_stackrI �type�__name__r� )rB r� r\ r� r� �errr7 r& r& r) rt � s" � ���
246 zKernel.interrupt_requestc � s� �| j sdS | �|d d �}t�|�r|I dH }| j j|d|||d� | j �d||�| _| �� I dH | j� d� | j
247 rG| j
248 j }|� |j
249 � | j� d� | jr\| jj }|� |j
250 � dS dS )zHandle a shutdown request.Nr� �restartZshutdown_replyr8 zStopping control ioloopzStopping shell ioloop)r� � do_shutdownr0 r� r� r� �_shutdown_message� _at_shutdownrZ r� r� r� r@ r rT )rB r� r\ r� r� Zcontrol_io_loopZ
251 shell_io_loopr& r& r) rr � s$ �
252
253 �zKernel.shutdown_requestc C s
254 d|d�S )z]Override in subclasses to do things when the frontend shuts down the
255 kernel.
256 r )r� r= r& )rB r= r& r& r) r> � s
257 zKernel.do_shutdownc � sh �| j sdS |d }|d }| �|�}t�|�r |I dH }t|�}| j �|d|||�}| j�d|� dS )z Handle an is_complete request.Nr� r� Zis_complete_replyr )r� �do_is_completer0 r� r" r� rZ r� )rB r� r\ r� r� r� r� r
258 r& r& r) rs � s �
259
260
261 zKernel.is_complete_requestc C s ddiS )r r� �unknownr& )rB r� r& r& r) rA r� zKernel.do_is_completec � s` �| j sdS |d }| �|�}t�|�r|I dH }t|�}| j �|d|||�}| j�d|� dS )zHandle a debug request.Nr� Z debug_replyr )r� �do_debug_requestr0 r� r" r� rZ r� )rB r� r\ r� r� r� r
262 r& r& r) rx s �
263
264
265 zKernel.debug_requestc C s< zt ||�� }|durt ||�W S |W S ty Y dS w )z Get the process metric value.Nr )r� �
266 BaseException)rB �processr� � attributeZ metric_valuer& r& r) �get_process_metric_value s �z Kernel.get_process_metric_valuec � s ��j sdS t�� t�� d�}t�� }|g|jdd��}�fdd�|D ��_t �fdd��j�
267 � D ��|d < t |� � d
268 �r?d
269 nd � t � �fd d��j�
270 � D ��|d
271 < t�
272 � }|durb|dkrb||d< tjdd�|d< tt�� �� �|d< �j �|d|||�}�j�d|� dS )zHandle a usage request.N)�hostnamer5 T�� recursivec s i | ] }|j � j�|j |��qS r& )r5 r; r3 �r' rE rU r& r) r* * s ��z(Kernel.usage_request.<locals>.<dictcomp>c s g | ] }� � |d d��qS )� cpu_percentN�rG rK rU r& r) �
273 <listcomp>/ � ��z(Kernel.usage_request.<locals>.<listcomp>Z
274 kernel_cpuZpssZrssc s g | ] }�� |d � ��qS )�memory_full_inforM rK �Z
275 mem_info_typerB r& r) rN 6 rO Z
276 kernel_memoryrj Zhost_cpu_percent)Zlogical� cpu_countZhost_virtual_memoryZ usage_replyr )r� �socket� gethostnamer0 r1 �psutil�Process�childrenr; �sumr7 r, rP rL rR r( Zvirtual_memory�_asdictr� rZ r� ) rB r� r\ r� r� �current_processZ
277 all_processesrL r
278 r& rQ r) ry ! s6 �
279 �
280 �� ��zKernel.usage_requestc � s �t �r+ r r� r& r& r) rC E s �zKernel.do_debug_requestc
281 � s� �| j �d� z|d }|d }|d d }W n ty* | j jd|dd� Y d S w | �|�}| �||||�\}} tj�� tj �� | �
282 |||�}| j sPd S | j j |d
283 |||| |d � d S ) zHandle an apply request.zBapply_request is deprecated in kernel_base, moving to ipyparallel.r� r� r� r� zGot bad msg: %sTr� NZ apply_reply)r� r\ r� r� )
284 rZ r� r� r� r� �do_applyr~ r r� r� r� r� r� )
285 rB r� r\ r� r� �bufsr� �mdr� Z
286 result_bufr& r& r) ru L s4 � �
287
288
289
290 �zKernel.apply_requestc C r )Z
291 DEPRECATEDr )rB r� r\ r� Zreply_metadatar& r& r) r[ l � zKernel.do_applyc � s� �| j �d� |d �dd�}t|t�r|g}|s | �� |D ]
292 }| j�t|�� q tdd�}| j s4dS | j j
293 |d|||d�}| j � d |� dS )
294 zabort a specific msg by idzOabort_request is deprecated in kernel_base. It is only part of IPython parallelr� �msg_idsNr )r� Z abort_reply)r� r� r\ r ) rZ r� r3 r� rI r r� �addr( r� r� r� )rB r� r\ r� r_ �midr� r
295 r& r& r) rw t s$ ��
296
297
298 �zKernel.abort_requestc � s: �| j �d� | �� }| jr| jj|d|||d� dS dS )zClear our namespace.zOclear_request is deprecated in kernel_base. It is only part of IPython parallelZ clear_reply)r\ r� r� N)rZ r� �do_clearr� r� )rB r� r� r� r� r& r& r) rv � s ���zKernel.clear_requestc C r )zDEPRECATED since 4.0.3r rU r& r& r) rb � r^ zKernel.do_clearc C s d| j }|� d|� ��� S )z!prefixed topic for IOPub messagesz kernel.%s�.)r\ �encode)rB Ztopic�baser& r& r) r� � s
299 z
300 Kernel._topicc s� d� _ � j�d� � jr� j�� � fdd�}t� j|�}� jrB� j� � j|� � j
301 d ur>t � j
302 d�r@� j
303 � � jd � d S d S d S |� d S )NTzAborting queuec � s �� j �d� d� _d S )NzFinishing abortF)rZ r� r� r& rU r& r) �
304 stop_aborting� s �
305 z+Kernel._abort_queues.<locals>.stop_aborting�_schedule_exitri )
306 r� rZ r� rT r� r r� �stop_on_error_timeoutr� r� r< r, rg )rB rf Zschedule_stop_abortingr& rU r) r � s
307 �
308 zKernel._abort_queuesc C s� | j sdS | j�d|d d |d d � |d d �dd�d d }d
309 d i}| �|�}| �|||�}|�|� | j j||||||d � dS )
310 z"Send a reply to an aborted requestNzAborting %s: %sr� r� r� �_r
311 r Z_replyr� r� )r� r� r� r\ )r� rZ r� �rsplitr� r� r$ r� )rB r� r� r� Z
312 reply_typer� r] r& r& r) r� � s
313
314
315 �zKernel._send_abort_replyc C s d}t |��)zPRaise StdinNotImplementedError if active frontend doesn't support
316 stdin.z?raw_input was called, but this frontend does not support stdin.r r� r& r& r) �
317 _no_raw_input� s zKernel._no_raw_input� c C sP | j s d}t|��|durddl}|jdtdd� | j|| jd | �d�dd �S )
318 z�Forward getpass to frontends
319
320 Raises
321 ------
322 StdinNotImplementedError if active frontend doesn't support stdin.
323 zFgetpass was called, but this frontend does not support input requests.Nr zTThe `stream` parameter of `getpass.getpass` will have no effect when using ipykernelrN rO rc T��password)� _allow_stdinr rQ rR � UserWarning�_input_requestr� rg )rB �promptr� r� rQ r& r& r) �getpass� s ��zKernel.getpassc C s4 | j s d}t|��| jt|�| jd | �d�dd�S )z�Forward raw_input to frontends
324
325 Raises
326 ------
327 StdinNotImplementedError if active frontend doesn't support stdin.
328 zHraw_input was called, but this frontend does not support input requests.rc Frm )ro r rq rI r� rg )rB rr r� r& r& r) � raw_input� s �zKernel.raw_inputc
329
330 C sj t j�� t j�� z | j�tj� W n tjy. } z
331 |j tj
332 kr)W Y d }~n� d }~ww q | j d us7J �t t
333 ||d��}| j j| jd|||d� z&t�| jgg | jgd�\}}} |s`| rq| j �| j�\}}
334 ||
335 fdkrqW n!W n ty� d} t| �d � ty� | jjddd � Y nw qLz|
336 d
337 d } W n ty� | j�d |� d
338 } Y nw | dkr�t�| S )NT)rr rn Z
339 input_requestr8 ri )NNzInterrupted by userzInvalid Message:r� r� �valuezBad input_reply: %srl �)r~ r� r� r � stdin_socketZrecv_multipartr� ZNOBLOCKZZMQError�errno�EAGAINr� r" r( r� �select�recvr� r� rZ r� r� �EOFError)
340 rB rr r\ r� rn �er� Zrlistri ZxlistZreplyr� ru r& r& r) rq sN
341
342 ��� �
343 �� �zKernel._input_requestc C sr t |��}| �� D ]-}| j�d||� z|tkr |�� n|tkr&|�� n|�|� W q t j
344 y6 Y q w dS )z�
345 Send a signal to all our children
346
347 Like `killpg`, but does not include the current process
348 (or possible parents).
349 zSending %s to subprocess %sN) r �_process_childrenrZ r� r � terminater
350 r4 � send_signalrU Z
351 NoSuchProcess)rB �signumZsig_repr. r& r& r) �_signal_children; s
352
353
354
355 ���zKernel._signal_childrenc C sr t �� }|jdd�}tjdkr|S t�� }g }|D ] }zt�|j�}W n ty, Y qw ||kr6|� |� q|S )z�Retrieve child processes in the kernel's process group
356
357 Avoids:
358 - including parents and self with killpg
359 - including all children that may have forked-off a new group
360 TrI r. )
361 rU rV rW r0 r� Zgetpgrpr2 r5 r3 �append)rB Zkernel_processZ all_childrenZ kernel_pgidZprocess_group_children�childZ
362 child_pgidr& r& r) r~ O s
363 �
364 �zKernel._process_childrenc � s� �d}| � � s| j�d� d S ttfD ],}|D ]'}| � � }|s)| j�d� d S | �|� | j�d||� t�|�I d H qqd S )N)ri g� ��Q��?g�������?g333333�?r
365 � �
366 zKernel has no children.z.No more children, continuing shutdown routine.z?Will sleep %s sec before checking for children and retrying. %s)r~ rZ r� r r
367 r� r� r )rB Zsleepsr� �delayrW r& r& r) �%_progressively_terminate_all_childrenf s( �
368 ���z,Kernel._progressively_terminate_all_childrenc � s� �zPz | � � I dH W n ty$ } z
369 | j�d|� W Y d}~nd}~ww W | jdurD| jrD| jj| j| j| �d�d� | j� d| j� | j
370 rP| j
371 � t j
372 � dS dS | jdurp| jrp| jj| j| j| �d�d� | j� d| j� | j
373 r{| j
374 � t j
375 � w w )zCActions taken at shutdown by the kernel, called by python's atexit.Nz,Exception during subprocesses termination %s�shutdownr8 r )r� r� rZ r� r? r� r� r� r� r� r� r� r� r� )rB r} r& r& r) r@ { s8 ���������zKernel._at_shutdown)Tr+ )rc )NNNFNNN)TNF)r r& )NNNNNF)rl N)rl r% )|r; �
376 __module__� __qualname__�__doc__r r< r; �__annotations__r rE r r r� Z profile_dirr! rT r rH r rV rY r� Zdebug_shell_socketr� r� Z iopub_threadrw �logging�LoggerrZ r Zint_idr r\ r_ r` ra r �tagZdebug_just_my_codeZ_darwin_app_napro r re r� �propertyrh r r Z_poll_intervalrh r? r� r r� r� r� r� r} r� r� r� r� r� r� r� r� rA r� r� r� r� r� r� r� r� r� r� r� r� rg r� r� r� rk r� rl r rm r rn r rq r# rp ro r6 rt rr r> rs rA rx rG ry rC ru r[ rw rv rb r� r� r r� rk rs rt rq r� r~ r� r@ �
377 __classcell__r& r& r� r) r: Q s@
378
379 �
380
381
382
383 �� ��
384 ������ &
385 G*
386
387 �
388
389
390
391
392 �" O� �
393
394
395 �
396
397
398
399 $ #
400
401
402 2r: )Dr� �
403 __future__r r� �concurrent.futuresr� r0 r� r� r0 rS r~ r� r �typingr� r] rQ r � functoolsr r r r r r rd r �platformr
404 Zjupyter_client.sessionr r� � ImportErrorrU r� ZIPython.core.errorr r �tornador Ztornado.queuesr r Z traitlets.config.configurabler Ztraitlets.traitletsr r r r r r r r r r r Zzmq.eventloop.zmqstreamr! Zipykernel.jsonutilr" �_versionr# Ziostreamr$ r9 r: r& r& r& r) �<module> sR
405
406 � 4
407