Format: application/x-python-code
1 o
2
3 �hcs � @ s� U d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z
4 ddl m Z ddl
5 mZ ddlZddlmZ ddlmZmZ ddlmZ ddlmZmZmZmZm Z m Z m Z m Z dd l m!Z! d
6 d l"m#Z# d
7 d l$m%Z% d
8 d
9 l&m'Z' d
10 dl(m)Z) d
11 dl*m+Z+ d
12 dl,m-Z. d
13 dl,m/Z/ d
14 dl0m1Z1 zddl2m3Z3 W n e4y� dZ3Y nw zddl5m6Z7 ddl5m8Z9 dZ:W n e4y� dZ:Y nw dZ;dd� Z<e�=� Z>da?e
15 j@e% eAd< d d � ZBe<e_CeBe_DG d d � d e.�ZEG d d!� d!eE�Z-dS )"z!The IPython kernel implementation� N)�contextmanager)�partial)�release)�line_at_cursor�token_at_cursor)�extract_header)�Any�Bool� HasTraits�Instance�List�Type�observe�observe_compat)� ZMQStream� ��BaseComm)� CommManager)�XCachingCompiler)� _use_appnope)� OutStream)�Kernel)�_accepts_parameters)�ZMQInteractiveShell)�_asyncio_runner)�provisionalcompleter)�rectify_completionsTFZ_jupyter_types_experimentalc O s t | i |��S )zCreate a new Comm.r ��args�kwargs� r! �Hc:\Users\shubh\Videos\agent\nova\Lib\site-packages\ipykernel/ipkernel.py� _create_comm0 s r# �
16 _comm_managerc O sX t du r*t� t du rt| i |��a W d � t S W d � t S 1 s%w Y t S )zCreate a new CommManager.N)r$ �
17 _comm_lockr r r! r! r" �_get_comm_manager: s
18 ��
19 ��r& c
20 s\ e Zd ZdZeddd�Zee�Ze ddd�j
21 dd�Z ee dd�Z
22 e� Zed�ed d
23 � ��Zeed dd �Zed
24 �edd� ��Ze� Ze� Z� fdd�Zeddejd d� d�ddd�ddd�ddd�d d d�d d d�d d!d�g�j
25 dd�Z d"Z e j Z d#ej �!� d$ d%d"ejd$ d&�d'd( d#d)d*�Z"d+d,� Z#e$d-d.� �Z%d/d0� Z&� fd1d2�Z'de� fd4d5� Z(� fd6d7�Z)d8d9� Z*dfd;d<�Z+d=d>� Z,e$d?d@� �Z-e-j.dAd@� �Z-e/dBdC� �Z0� fdDdE�Z1 :dgd d dF�dGdH�Z2dIdJ� Z3dKdL� Z4dMdN� Z5dhdPdQ�Z6 $ $ :didRdS�Z7dTdU� Z8dVdW� Z9dXdY� Z:dZd[� Z;d\d]� Z<d^d_� Z=d`e>j?da dbe>j@eAe>jf fdcdd�ZB� ZCS )j�
26 IPythonKernelzThe IPython Kernel class.z1IPython.core.interactiveshell.InteractiveShellABCT)�
27 allow_nonezUSet this flag to False to deactivate the use of experimental IPython completion APIs.)�help)�config� user_modulec C s | j d ur
28 |d | j _d S d S �N�new)�shellr+ ��selfZchanger! r! r" �_user_module_changedW s
29 �z"IPythonKernel._user_module_changedN)r r( �user_nsc C s( | j d ur|d | j _| j �� d S d S r, )r. r2 Z init_user_nsr/ r! r! r" �_user_ns_changed_ s
30 �z IPythonKernel._user_ns_changedc sZ t � jd i |�� ddlm}m} |r#|| j| j| j| j| j | j
31 �| _| j j | | j
32 | j| j| td�| _| j | jj_ tj�d�}|rG|| jjd< | j| jj_| �d�| jj_| j | jj_ | j| jj_t� � | _ t | j t �snJ �| jj �!| j � g d�}|D ] }t"| j |�| j#|< q|t$� r�| j%r�dd l&}|�'� i | _(| �)� t*t+d
33 �r�t+j,�!| j-� d S d S ) zInitialize the kernel.r )�Debugger�_is_debugpy_available)�parent� profile_dirr+ r2 �kernelZcompiler_classZJPY_SESSION_NAMEZ __session__Zexecute_result)Z comm_openZcomm_msgZ
34 comm_closer N� callbacksr! ).�super�__init__�debuggerr4 r5 �log�debugpy_streamZ_publish_debug_eventZdebug_shell_socket�sessionZdebug_just_my_code� shell_class�instancer7 r+ r2 r r. � displayhook�os�environ�get� iopub_socketZ
35 pub_socket�_topicZtopicZ display_pub�comm�get_comm_managerZ comm_manager�
36 isinstancer
37 Z
38 configurables�append�getattrZshell_handlersr Z_darwin_app_nap�appnopeZnope�_new_threads_parent_header�_initialize_thread_hooks�hasattr�gcr9 �_clean_thread_parent_frames)r0 r r4 r5 Zjupyter_session_nameZcomm_msg_types�msg_typerM �� __class__r! r" r; k sR �
39 �
40
41 �zIPythonKernel.__init__zPython Referencez https://docs.python.org/%i.%i� )�text�urlzIPython Referencez&https://ipython.org/documentation.htmlzNumPy Referencez+https://docs.scipy.org/doc/numpy/reference/zSciPy Referencez+https://docs.scipy.org/doc/scipy/reference/zMatplotlib Referencez$https://matplotlib.org/contents.htmlzSymPy Referencez'http://docs.sympy.org/latest/index.htmlzpandas Referencez-https://pandas.pydata.org/pandas-docs/stable/�ipython�pythonr z
42 text/x-python)�name�versionz ipython%d� z.py)r[ r\ �mimetypeZcodemirror_modeZpygments_lexerZnbconvert_exporterZfile_extensionc C sD ddl m} |r |d j�d�}| j�d|� | j j�|� d S d S )Nr �r5 zutf-8zDebugpy received: %s)r< r5 �bytes�decoder= �debugZ
43 tcp_clientZreceive_dap_frame)r0 �msgr5 �framer! r! r" �dispatch_debugpy� s �z IPythonKernel.dispatch_debugpyc C � | j r| j jS d S �N)r. �banner�r0 r! r! r" rh � � zIPythonKernel.bannerc � s � | j �� I dH q)zPoll the stopped queue.TN)r< Zhandle_stopped_eventri r! r! r" �poll_stopped_queue� s ��z IPythonKernel.poll_stopped_queuec sf | j rd| j _| jdu r| j�d� n | jj| jdd� t� �� | jr1t �
44 | � � | j j
45 j� dS dS )zStart the kernel.FNz7debugpy_stream undefined, debugging will not be enabled��copy)r. �exit_nowr> r= �warningZon_recvre r: �start�asyncio�run_coroutine_threadsaferk Zcontrol_thread�io_loopZ asyncio_loopri rT r! r" rp � s
46
47 ��zIPythonKernel.startr. c s6 t � �|||� |dkr| jr| j�|� dS dS dS )zmOverridden from parent to tell the display hook and output streams
48 about the parent message.
49 r. N)r: �
50 set_parentr. )r0 �identr6 ZchannelrT r! r" rt � s �zIPythonKernel.set_parentc s" t � �|�}|�d| jd�� |S )zVInitialize metadata.
51
52 Run at the beginning of each execution request.
53 T)�dependencies_metZengine)r: �
54 init_metadata�updateru )r0 r6 �mdrT r! r" rw � s ��zIPythonKernel.init_metadatac C s0 |d |d< |d dkr|d dkrd|d< |S )zXFinish populating metadata.
55
56 Run after completing an execution request.
57 �status�error�enameZUnmetDependencyFrv r! )r0 r6 �metadata�
58 reply_contentr! r! r" �finish_metadata s z IPythonKernel.finish_metadataFc C s* || _ tj| _| jt_tj| _| jt_dS )zZForward raw_input and getpass to the current frontend.
59
60 via input_request
61 N)Z _allow_stdin�builtins�input�_sys_raw_input� raw_input�getpass�
62 _save_getpass)r0 � allow_stdinr! r! r" �_forward_input s
63 z IPythonKernel._forward_inputc C s | j t_| jt_dS )zRestore raw_input, getpassN)r� r� r� r� r� ri r! r! r" �_restore_input( s z IPythonKernel._restore_inputc C rf rg )r. �execution_countri r! r! r" r� . rj z IPythonKernel.execution_countc C s d S rg r! )r0 �valuer! r! r" r� 4 s c
64 # sz � t �� �dd� }��t||�� |�t|��� � �fdd�}t�tj|�}z
65 dV W t�tj|� dS t�tj|� w )aP ContextManager for capturing SIGINT and cancelling a future
66
67 SIGINT raises in the event loop when running async code,
68 but we want it to halt a coroutine.
69
70 Ideally, it would raise KeyboardInterrupt,
71 but this turns it into a CancelledError.
72 At least it gets a decent traceback to the user.
73 c S s | � � s| �� r
74 d S | �� d S rg )� cancelled�done�cancel)�fZ_ignoredr! r! r" �cancel_unless_doneI s z;IPythonKernel._cancel_on_sigint.<locals>.cancel_unless_donec s �fdd�}� j �|� d S )Nc s" � � � s� �� r
75 d S � �d� d S )Nr )r� r� �
76 set_resultr! )�
77 sigint_futurer! r" �set_sigint_resultV s zQIPythonKernel._cancel_on_sigint.<locals>.handle_sigint.<locals>.set_sigint_result)rs Z add_callback)r r� �r0 r� r! r" �
78 handle_sigintU s z6IPythonKernel._cancel_on_sigint.<locals>.handle_sigintN)rq �Future�add_done_callbackr �signal�SIGINT)r0 �futurer� r� Z save_sigintr! r� r" �_cancel_on_sigint: s �
79 "z IPythonKernel._cancel_on_sigintc � s. �t |�}| �|� t� �|||�I dH dS )z/Override for cell output - cell reconciliation.N)r �%_associate_new_top_level_threads_withr: �execute_request)r0 �streamru r6 �
80 parent_headerrT r! r" r� f s �
81 z IPythonKernel.execute_request)� cell_meta�cell_idc � s� �| j � � dus
82 J �| �|� i }t� d�r(t� d�r(� j} � j}
83 t| dg�} ndd� }
84 � fdd�} t� jdg�} z�d} z� �|�}
85 W n tyR |}
86 t �
87 � } Y nw t rĈ j t u r�t
88 �� �� r�|
89 ||
90 | d �r�| d rv| ||||
91 | |d
92 �}n | ||||
93 | d �}t
94 �|�}| �|��2 d}z|I dH }W � j�d � |s�� j�d
95 |� n� j�d � |s�� j�d
96 |� w w W d � n1 s�w Y n| d r҈ j||||d�}n� j|||d�}W | �� n| �� w |jdur�|jn|j}|jr�d|d< n%d|d< |�� j�pg tt|�j �t|�d�� t | j | j dd�}||d< � j d |d< d|v �r5| j!�"dd�#|d �� |d dk�rG� �$|�pBi �|d < ni |d < � j%�&� |d < � j%�'� |S ) zHandle code execution.N�run_cell_async�should_run_asyncr� c S s dS )NFr! )�cellr! r! r" �<lambda>� s z*IPythonKernel.do_execute.<locals>.<lambda>c � s �� j | i |��S rg )�run_cellr �r. r! r" r� � s �z*IPythonKernel.do_execute.<locals>.run_cell)�transformed_cell�preprocessing_exc_tuple)�
97 store_history�silentr� r� r� )r� r� r� r� Z post_executeZ
98 post_run_cell)r� r� r� )r� r� �okrz r{ �� tracebackr| Zevalue�execute�Z engine_uuidZ engine_id�method� engine_infor r� r� z Exception in execute request:
99 %s�
100 �user_expressions�payload)(r. r� rP r� r� r r� Ztransform_cell� Exception�sys�exc_infor Z loop_runnerrq �get_event_loop�
101 is_running�
102 ensure_futurer� �events�triggerr� Zerror_before_execZ
103 error_in_exec�successrx �_last_traceback�str�type�__name__�dictru �int_idr� r= �info�joinr� Zpayload_managerZ read_payloadZ
104 clear_payload)r0 �coder� r� r� r� r� r� r~ r� r� Zaccepts_paramsr� r� �coroZ coro_future�res�err�e_infor! r� r" �
105 do_executel s� �
106 ��
107
108 ���� �
109 � ��� �� ��
110
111 ��
112 �
113 zIPythonKernel.do_executec C sv t r | jr | �||�S |du rt|�}t||�\}}|| }| jdus%J �| j�d||�\}}|||t|� i dd�S )zHandle code completion.N� r� ��matchesZ
114 cursor_endZ cursor_startr} rz )� _use_experimental_60_completion� use_experimental_completions�_experimental_do_complete�lenr r. Zcomplete)r0 r� �
115 cursor_pos�line�offsetZ line_cursor�txtr� r! r! r" � do_complete� s
116
117 �zIPythonKernel.do_completec � s( �ddl m} |r| j �|�I dH S dS )zHandle a debug request.r r_ N)r< r5 �process_request)r0 rc r5 r! r! r" �do_debug_request s
118 � z IPythonKernel.do_debug_requestc
119 C s� |du rt |�}t� �6 | jdusJ �| jj�||�}tt||��}g }|D ]}|�t|j |j
120 |j |j |j
121 d�� q&W d � n1 sDw Y |r]|d j }|d j
122 }dd� |D �} n|}|}g } | ||t|idd�S )zD
123 Experimental completions from IPython, using Jedi.
124 N)rp �endrW r� � signaturer c S s g | ]}|j �qS r! )rW )�.0�cr! r! r" �
125 <listcomp>2 � z;IPythonKernel._experimental_do_complete.<locals>.<listcomp>r� r� )r� �_provisionalcompleterr. Z Completer� completions�list�_rectify_completionsrK r� rp r� rW r� r� �_EXPERIMENTAL_KEY_NAME)
126 r0 r� r� Zraw_completionsr� �comps�comp�s�er� r! r! r" r� s@ ��� �
127
128 �z'IPythonKernel._experimental_do_completer! c C s� t ||�}ddi}i |d< i |d< | jdusJ �z0tjdkr(| jj|||d�}n| jj||d�}|d �|� | jjsB|d �d � d
129 |d < W |S tyV d |d < Y |S w )
130 zHandle code inspection.rz r� �datar} N)� )� detail_level�
131 omit_sections)r� z text/htmlT�foundF) r r. r � version_infoZobject_inspect_mimerx Zenable_html_pager�pop�KeyError)r0 r� r� r� r� r[ r~ Zbundler! r! r" �
132 do_inspect@ s,
133
134 �
135 �
136 �zIPythonKernel.do_inspectc
137 C s� | j dusJ �|dkr| j jj|||dd�}
138 n$|dkr(| j jj|||||d�}
139 n|dkr9| j jj||||| d�}
140 ng }
141 d t|
142 �d
143 �S ) zHandle code history.N�tailT)�raw�outputZinclude_latest�range)r� r� �search)r� r� �n�uniquer� )rz �history)r. Zhistory_managerZget_tailZ get_ranger� r� ) r0 Zhist_access_typer� r� r? rp �stopr� �patternr� �histr! r! r" �
144 do_history\ s"
145 �
146 �
147 ��zIPythonKernel.do_historyc C s | j rd| j _td|d�S )zHandle kernel shutdown.Tr� )rz �restart)r. rn r� )r0 r� r! r! r" � do_shutdown� s zIPythonKernel.do_shutdownc C sZ t | jdd�}|du r| jdusJ �| jj}|�|�\}}d|i}|dkr+d| |d< |S )z Handle an is_complete request.Zinput_transformer_managerNrz �
148 incomplete� �indent)rL r. Zinput_splitterZcheck_complete)r0 r� Ztransformer_managerrz Z
149 indent_spaces�rr! r! r" �do_is_complete� s z IPythonKernel.do_is_completec
150 C s z
151 ddl m}m} W n ty ddlm}m} Y nw | j}|dus%J �z�|j}dt|��dd� d } |||dd �\}
152 } } t |
153 d
154 d �}
155 | d }
156 | d }| d
157 }| d }|
158 |
159 || || |di}|�
160 |� |� d|
161 � d|� d|� d�}zt ||j |j� |�
162 |�}W |D ]}|�|� q�n |D ]}|�|� q�w | jdus�J �||| jj| jjd�}W nU ty� } zI|�� |jp�g tt|�j�t|�d�}t| j| jdd�}||d< | j| jd|| � d�d� | j � dd� |d �� g }d|d < W Y d}~||fS d}~ww d d i}||fS ) zHandle an apply request.r )�serialize_object�unpack_apply_messager N�_�-r� Frl r� r� r r �resultz = z(*z,**�))�buffer_threshold�item_thresholdr� �applyr� r� r{ �ru z Exception in apply request:
163 %sr� r� rz r� ) Zipyparallel.serializer r � ImportError� serializer. r2 r� �replacerL rx �execZuser_global_nsrE r� r? r r �
164 BaseException�
165 showtracebackr� r� r� r� ru r� Z
166 send_responserF rG r= r� r� )r0 �contentZbufsZmsg_idZreply_metadatar r r. Zworking�prefixr� r r �fname�argnameZ kwargnameZ
167 resultname�nsr� r �keyZ
168 result_bufr� r~ r� r! r! r" �do_apply� sn �
169
170 � �
171 � ����zIPythonKernel.do_applyc C s | j r | j �d� tdd�S )zClear the kernel.Fr� )rz )r. �resetr� ri r! r! r" �do_clear� s
172 zIPythonKernel.do_clearc C s
173 || _ dS )zBStore the parent header to associate it with new top-level threadsN)rN )r0 r� r! r! r" r� � s
174 z3IPythonKernel._associate_new_top_level_threads_withc sl | j �| j�t�� �| �tjj�tjj� dtjf�����fdd� }dtjf� fdd� }|tj_|tj_dS )z=Store thread hierarchy and thread-parent_header associations.r0 c sh z| j }W n
175 ty Y dS w ��fD ]}t|t�r-|�kr'�j|j| j< q||j| j< q� | � dS )aY Wrap the `threading.Thread.start` to intercept thread identity.
176
177 This is needed because there is no "start" hook yet, but there
178 might be one in the future: https://bugs.python.org/issue14073
179
180 This is a no-op if the `self._stdout` and `self._stderr` are not
181 sub-classes of `OutStream`.
182 N)� _ipykernel_parent_thread_ident�AttributeErrorrJ r rN �_thread_to_parent_headerru �_thread_to_parent)r0 r6 r� )�_threading_Thread_runr8 �kernel_thread_ident�stderr�stdoutr! r" � run_closure� s
183
184 �
185 �� � z;IPythonKernel._initialize_thread_hooks.<locals>.run_closurec s$ � | g|�R i |�� t �� | _d S rg )� threading� get_identr )r0 r r )�_threading_Thread__init__r! r" � init_closure s z<IPythonKernel._initialize_thread_hooks.<locals>.init_closureN)�_stdout�_stderrr' r( �Thread�runr; )r0 r& r* r! )r) r" r8 r# r$ r% r" rO � s z&IPythonKernel._initialize_thread_hooks�phase)rp r� r� c
186 C s� |dkrdS dd� t �� D �}| j| jfD ]E}t|t�rZ|j}t|�� �D ]}||vr:z||= W q% t y9 Y q%w q%|j
187 }t|�� �D ]}||vrYz||= W qD t yX Y qDw qDqdS )a� Clean parent frames of threads which are no longer running.
188 This is meant to be invoked by garbage collector callback hook.
189
190 The implementation enumerates the threads because there is no "exit" hook yet,
191 but there might be one in the future: https://bugs.python.org/issue14073
192
193 This is a no-op if the `self._stdout` and `self._stderr` are not
194 sub-classes of `OutStream`.
195 rp Nc S s h | ]}|j �qS r! r )r� �threadr! r! r" � <setcomp> r� z<IPythonKernel._clean_thread_parent_frames.<locals>.<setcomp>) r' � enumerater+ r, rJ r r r� �keysr� r! )r0 r/ r� Zactive_threadsr� Zthread_to_parent_header�identityZthread_to_parentr! r! r" rR s2
196
197
198 ��
199 ����z)IPythonKernel._clean_thread_parent_framesr� )F)TNF)r r! )r r NNNF)Dr� �
200 __module__� __qualname__�__doc__r r. r
201 r r@ r �tagr� r r> r r+ r r r1 r� r2 r3 r� Z_sys_eval_inputr; r r� r� Z
202 help_links�implementationr r\ �implementation_version�splitZ
203 language_infore �propertyrh rk rp rt rw r r� r� r� �setterr r� r� r� r� r� r� r� r� r� r r r r� rO �t�Literal�Dictr� rR �
204 __classcell__r! r! rT r" r' H s� �� ;��������� �" �
205
206
207
208
209
210
211
212 +
213 � �
214 (!
215 �$
216 B(��r' c s e Zd ZdZ� fdd�Z� ZS )r z3DEPRECATED. An alias for the IPython kernel class.c s. ddl }|jdtdd� t� j|i |�� dS )z DEPRECATED.r Nz@Kernel is a deprecated alias of ipykernel.ipkernel.IPythonKernelrV )�
217 stacklevel)�warnings�warn�DeprecationWarningr: r; )r0 r r rC rT r! r" r; / s �zKernel.__init__)r� r5 r6 r7 r; rA r! r! rT r" r , s r )Fr7 rq r� rQ r� rC r� r� r' �typingr> �
218 contextlibr � functoolsr rH Z IPython.corer ZIPython.utils.tokenutilr r Zjupyter_client.sessionr Z traitletsr r r
219 r r r
220 r r Zzmq.eventloop.zmqstreamr Z comm.commr Z comm.managerr �compilerr Z
221 eventloopsr Ziostreamr Z
222 kernelbaser Z
223 KernelBaser Zzmqshellr Z IPython.core.interactiveshellr r ZIPython.core.completerr r� r r� r� r� r# �Lockr% r$ �Optional�__annotations__r& Z create_commrI r' r! r! r! r" �<module> sh ( � �
224 i