Format: application/x-python-code
1 o
2
3 �h�i �
4 @ s� d Z ddlZddlZddlZddlZddlmZ ddlZddl m
5 Z
6 ddl m Z ddl
7 mZ ddlmZ ddlmZ zdd lmZ W n eyQ dd
8 lmZ Y nw d d lmZmZmZ zdd
9 lm Z ddl m Z ddl m Z m!Z! dZ"W n' ey dZ"Y n e#y� Z$ ze$j%j&dkr�dZ"ne$�W Y dZ$[$ndZ$[$ww e'edd�p�ej(Z)G dd� d�Z*G dd� d�Z+G dd� d�Z,G dd� d�Z-G d d � d �Z.G d d � d �Z/G d d!� d!�Z0dS )"z/Debugger implementation for the IPython kernel.� N)�Path)� get_ipython)�leading_empty_lines)�Event)�Queue)�jsonapi)� json_default)� date_default� )�
10 get_file_name�get_tmp_directory�get_tmp_hash_seed)�api)�pydevd_frame_utils)�SuspendedFramesManager�_FramesTrackerTFZDebuggerInitializationError�
11 ROUTING_IDc @ � e Zd ZdZdd� ZdS )� _FakeCodezFake code class.c C s || _ || _dS ��Init.N)� co_filename�co_name)�selfr r � r �Hc:\Users\shubh\Videos\agent\nova\Lib\site-packages\ipykernel/debugger.py�__init__3 s
12 z_FakeCode.__init__N��__name__�
13 __module__� __qualname__�__doc__r r r r r r 0 � r c @ r )�
14 _FakeFramezFake frame class.c C s || _ || _|| _d| _dS r )�f_code� f_globals�f_locals�f_back)r r$ r% r& r r r r < s
15 z_FakeFrame.__init__Nr r r r r r# 9 r" r# c @ r )�
16 _DummyPyDBzFake PyDb class.c C s ddl m} |�� | _dS )r r )� PyDevdAPIN)Z_pydevd_bundle.pydevd_apir) ZVariablePresentationZvariable_presentation)r r) r r r r G s z_DummyPyDB.__init__Nr r r r r r( D r" r( c @ s2 e Zd ZdZdd� Zdd� Zdd� Zd d d
17 �ZdS ) �VariableExplorerzA variable explorer.c C s* t � | _t� | _t| j| j�| _d| _dS )zInitialize the explorer.N)r �suspended_frame_managerr( Zpy_dbr �tracker�frame�r r r r r Q s
18 zVariableExplorer.__init__c C s: t � j}ttdtd��||�| _| j�dt� | j�� dS )zStart tracking.�<module>zsys._getframe()Zthread1N)
19 r Zuser_nsr# r r r- r, �trackr Z create_frames_list_from_frame)r �varr r r r0 X s zVariableExplorer.trackc C s | j �� dS )zStop tracking.N)r, � untrack_allr. r r r r2 ^ s z VariableExplorer.untrack_allNc C s0 |}|s t | j�}| j�|�}dd� |�� D �S )z1Get the child variables for a variable reference.c S s g | ]}|� � �qS r )Z get_var_data)�.0�xr r r �
20 <listcomp>h s z;VariableExplorer.get_children_variables.<locals>.<listcomp>)�idr- r+ Z get_variable�get_children_variables)r Z variable_refZvar_ref� variablesr r r r7 b s
21
22 z'VariableExplorer.get_children_variables�N)r r r r! r r0 r2 r7 r r r r r* N s r* c @ sH e Zd ZdZdZdZdZdZdd� Zdd � Z d
23 d � Z
24 d d
25 � Z dd� Z dS )�DebugpyMessageQueuezA debugpy message queue.zContent-Length: � z
26
27 � c C s& d| _ | �� || _t� | _|| _dS )zInit the queue.� N)�
28 tcp_buffer�_reset_tcp_pos�event_callbackr �
29 message_queue�log)r r@ rB r r r r s s
30
31 z DebugpyMessageQueue.__init__c C s d| _ d| _d| _d| _d S )N�����r )�
32 header_pos�
33 separator_pos� message_size� message_posr. r r r r? { s
34 z"DebugpyMessageQueue._reset_tcp_posc C s� | j �d� t�tjttjf t�|��}|d dkr.| j �d� | j �|� | � |� d S | j �d� | j �|� | j
35 � |� d S )NzQUEUE - _put_message:�type�eventzQUEUE - received event:zQUEUE - put message:) rB �debug�t�cast�Dict�str�Anyr �loadsr@ rA �
36 put_nowait)r Zraw_msg�msgr r r � _put_message� s z DebugpyMessageQueue._put_messagec C s� | j |7 _ | j�d� | jdkr| j �tj�| _| jdkr"dS | j�d| j� | jdkr>| jtj }| j �tj |�| _| jdkrEdS | j�d| j� | j
37 dkrj| jtj }| jtj | _
38 t | j || j� �| _
39 | j�d| j
40 � | j�d| j
41 � t| j �| j
42 | j
43 k r�dS | �| j | j
44 | j
45 | j
46 � � t| j �| j
47 | j
48 kr�| j�d � d
49 | _ | �� dS | j | j
50 | j
51 d� | _ | j�d | j � | �� q) z Put a tcp frame in the queue.zQUEUE - received frameTrC Nz QUEUE - found header at pos %iz!QUEUE - found separator at pos %iz QUEUE - found message at pos %izQUEUE - message size is %iz QUEUE - resetting tcp_bufferr= z QUEUE - slicing tcp_buffer: %s)r> rB rJ rD �findr: �HEADERrE �
52 HEADER_LENGTH� SEPARATORrG �SEPARATOR_LENGTH�intrF �lenrS r? )r r- �hintZsize_posr r r �
53 put_tcp_frame� sD
54
55
56
57
58 � �z!DebugpyMessageQueue.put_tcp_framec � s �| j �� I dH S )z Get a message from the queue.N)rA �getr. r r r � get_message� s �z DebugpyMessageQueue.get_messageN)
59 r r r r! rU rV rW rX r r? rS r\ r^ r r r r r: k s /r: c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
60 d � Zd d
61 � Z dd� Z
62 dd� Z dd� Z dd� Z
63 dd� ZdS )�
64 DebugpyClientzA client for debugpy.c C sL || _ || _|| _t| j| j �| _d| _d| _d| _d| _ t
65 � | _ d| _ dS )zInitialize the client.� 127.0.0.1rC NT)
66 rB �debugpy_streamr@ r: �_forward_eventrA � debugpy_host� debugpy_port�
67 routing_id�wait_for_attachr �
68 init_event�init_event_seq)r rB ra r@ r r r r � s
69 zDebugpyClient.__init__c C s | � � \}}d| d t|� S )N�tcp://�:)�
70 get_host_portrN )r �host�portr r r �
71 _get_endpoint� s zDebugpyClient._get_endpointc C s. |d dkr| j �� |d | _| �|� d S )NrI Z initialized�seq)rg �setrh r@ �r rR r r r rb � s
72
73 z DebugpyClient._forward_eventc C s� | j d u r
74 | jj�t�| _ tj|tddd�}tt |��}t
75 j | t
76 j �
77 d�}||7 }| j�d� | j�| j � | j�|� | j�| j |f� d S )NF)�default� ensure_ascii� allow_nan�asciizDEBUGPYCLIENT:)re ra �socket�
78 getsockoptr r �dumpsr rN rZ r: rU rW �encoderB rJ Zsend_multipart)r rR �content�content_length�bufr r r �
79 _send_request� s"
80 � � zDebugpyClient._send_requestc � s �| j �� I d H S r9 )rA r^ r. r r r �_wait_for_response� s �z DebugpyClient._wait_for_responsec � sN �| j �� I d H dt| j�d dd�}| �|� | �� I d H | �� I d H S )N�requestr
81 �configurationDone)rH ro �command)rg �waitrY rh r} r~ )r r� r r r �_handle_init_sequence� s � �
82 z#DebugpyClient._handle_init_sequencec C sn | j dkr1| jj}|�d| j � |�tj��d�| _ |�
83 | j � | j � d�}| j |d d� | _ | j| j fS )zGet the host debugpy port.rC ri �utf-8rj r
84 N) rd ra rv Zbind_to_random_portrc rw �zmqZ
85 LAST_ENDPOINT�decode�endpointZunbind�rfind)r rv �indexr r r rk s
86 zDebugpyClient.get_host_portc C s& | j j�| �� � | j j�t�| _dS )zConnect to the tcp socket.N)ra rv �connectrn rw r re r. r r r �connect_tcp_socket s z DebugpyClient.connect_tcp_socketc C s0 | j j�| �� � d| _t� | _d| _d| _dS )z Disconnect from the tcp socket.NrC T) ra rv �
87 disconnectrn re r rg rh rf r. r r r �disconnect_tcp_socket s
88
89 z#DebugpyClient.disconnect_tcp_socketc C s | j �|� dS )zReceive a dap frame.N)rA r\ )r r- r r r �receive_dap_frame s z DebugpyClient.receive_dap_framec � s` �| � |� | jr|d dkr| �� I dH }d| _|S | �� I dH }| j�d� | j�|� |S )zSend a dap request.r� �attachNFzDEBUGPYCLIENT - returning:)r} rf r� r~ rB rJ )r rR �repr r r �send_dap_request" s �
90 z DebugpyClient.send_dap_requestN)r r r r! r rn rb r} r~ r� rk r� r� r� r� r r r r r_ � s
91 r_ c @ s� e Zd ZdZg d�Zg d�Z d4dd�Zdd� Zd d
92 � Zd d � Z d
93 d� Z
94 dd� Z e dd� �Z
95 dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3S )5�DebuggerzThe debugger class.)�dumpCell�setBreakpoints�source�
96 stackTracer8 r� r� )� debugInfo�inspectVariables�richInspectVariables�modules�
97 copyToGlobalsTc C s� || _ t||| j�| _|| _|| _d| _|| _|| _t � | _
98 i | _ t j
99 D ]
100 }t| |�| j |< q$i | _t jD ]
101 }t| |�| j|< q5i | _t� | _d| _i | _d| _d| _d| _t� | _dS )zInitialize the debugger.Fr` r N)rB r_ �
102 _handle_event�debugpy_client� shell_socket�session�
103 is_startedr@ � just_my_coder �
104 stopped_queue�started_debug_handlersr� �started_debug_msg_types�getattr�static_debug_handlers�static_debug_msg_types�breakpoint_listrp �stopped_threads�debugpy_initialized�_removed_cleanuprc rd r� r* �variable_explorer)r rB ra r@ r� r� r� �msg_typer r r r G s,
105
106 zDebugger.__init__c C s� |d dkr%|d d r| j �|� d S | j�|d d � | �|� d S |d dkrG|d d r6t� | _n
107 | j�|d d � | �|� d S | �|� d S )NrI �stopped�bodyZallThreadsStoppedZthreadId� continuedZallThreadsContinued)r� rQ r� �addr@ rp �removerq r r r r� h s
108 zDebugger._handle_eventc � s �| j �|�I d H S r9 )r� r� rq r r r �_forward_messagey s �zDebugger._forward_messagec s4 � fdd�|D �}|d d|d d|d d|id�S ) Nc � g | ] }� � |d �r|�qS ��name��accept_variable�r3 r1 r. r r r5 } � z6Debugger._build_variables_response.<locals>.<listcomp>ro �responseTr� r8 )ro rH � request_seq�successr� r� r )r r r8 Zvar_listr r. r �_build_variables_response| s �z"Debugger._build_variables_responsec C s g d�}||vS )N)ZIPythonHistorySavingThreadzThread-2zThread-3zThread-4r )r Z thread_name� forbid_listr r r �_accept_stopped_thread� s z Debugger._accept_stopped_threadc � st �| j �� I dH }|d d ddd�}| �|�I dH }|d d D ]}| �|d �r2| j�|d � q!| �|� dS )
109 zHandle a stopped event.Nro r
110 r �threads)ro rH r� r� r� r6 )r� r] r� r� r� r� r@ )r rI �reqr� �threadr r r �handle_stopped_event� s ��z Debugger.handle_stopped_eventc C s | j S r9 )r� r. r r r �
111 tcp_client� s zDebugger.tcp_clientc
112 C s� | j sQt� }t|��� st|�jdd� | j�� \}}d}|d| d | d 7 }|dd�}| j�| j d|d | j �
113 t �� | jj | j d
114 d �\}}|d d
115 dk| _ t
116 � jj}t|v rg|�t�} |�| �| j| < | j�� | j S )zStart the debugger.T)�parentszimport debugpy;zdebugpy.listen(("z",z)))�codeZsilentZexecute_requestNr )�moderz �status�ok)r� r r �exists�mkdirr� rk r� �sendr� rw r �recvr �input_transformer_manager�cleanup_transformsr r� �popr� r� )
117 r Ztmp_dirrl rm r� rz �identrR r� r� r r r �start� s.
118
119 �
120
121
122 zDebugger.startc C s@ | j �� t� jj}t| j�D ]}| j�|�}|�||� qdS )zStop the debugger.N) r� r� r r� r� �sortedr� r� �insert)r r� r� �funcr r r �stop� s
123
124 �z
125 Debugger.stopc � sj �|d d }t |�}t|ddd��
126 }|�|� W d � n1 s"w Y d|d d |d
127 d |id �S )
128 zHandle a dump cell message.� argumentsr� �wr� ��encodingNr� ro Tr� Z
129 sourcePath�rH r� r� r� r� )r �open�write)r �messager� � file_name�fr r r r� � s � ��zDebugger.dumpCellc � s^ �|d d d }|d d | j |< | �|�I dH }|�d�r-dd� |d d D �| j |< |S )
130 z!Handle a set breakpoints message.r� r� �path� breakpointsNr� c S s g | ]}d |d i�qS )�liner )r3 �
131 breakpointr r r r5 � s
132 ��z+Debugger.setBreakpoints.<locals>.<listcomp>r� )r� r� r] )r r� r� Zmessage_responser r r r� � s �
133
134 �zDebugger.setBreakpointsc � s� �d|d |d d�}|d d d }t |��� r>t|dd ��}d
135 |d < d |�� i|d
136 < W d � |S 1 s7w Y |S d|d < d|d< i |d
137 < |S )zHandle a source message.r� ro r� )rH r� r� r� r� r� r� r� Tr� rz r� NFzsource unavailabler� )r �is_filer� �read)r r� �reply� source_pathr� r r r r� � s �
138 ���zDebugger.sourcec � s~ �| � |�I dH }z+|d d }t|�tdd� tt|�d�D �� }|d d d|d � |d d< W |S ty> Y |S w )z Handle a stack trace message.Nr� Z stackFramesc s s, � | ]\}}|d dkr|dkr|V qdS )r� r/ r
139 Nr )r3 �i�vr r r � <genexpr> s � "�z&Debugger.stackTrace.<locals>.<genexpr>r
140 )r� rZ �next� enumerate�reversed�
141 StopIteration)r r� r� Zsf_listZ
142 module_idxr r r r� � s �
143
144 �" ��zDebugger.stackTracec C s> g d�}||v}|ot t�d|�� }|o |dd� dk}|S )zAccept a variable by name.)r r! � __package__�
145 __loader__�__spec__�__annotations__� __builtins__� __builtin__Z __display__r Zdebugpy�exit�quit�InZOutZ_ohZ_dh�_�__Z___z^_\dr � Z_i)�bool�re�search)r Z
146 variable_namer� Zcondr r r r� s
147 zDebugger.accept_variablec � sb �i }� j s� j�|d d �}� �||�S � �|�I dH }� fdd�|d d D �|d d< |S )zHandle a variables message.r� ZvariablesReferenceNc r� r� r� r� r. r r r5 0 s
148 ��z&Debugger.variables.<locals>.<listcomp>r� r8 )r� r� r7 r� r� )r r� r� r8 r r. r r8 % s �
149 �
150
151 �zDebugger.variablesc � sR �| j �� \}}||d�|d d< d|d d< | js!dg|d d< | �|�I dH S ) zHandle an attach message.)rl rm r� r� TZ logToFileZ DebugStdLibZ debugOptionsN)r� rk r� r� )r r� rl rm r r r r� 5 s � zDebugger.attachc � s �|d d|d d|d d�S )z$Handle a configuration done message.ro r� Tr� )ro rH r� r� r� r )r r� r r r r� B s ��zDebugger.configurationDonec � sl �g }| j �� D ] \}}|�||d�� qd|d d|d | jdt� t� tj d|t| j �ddgdd �
152 d
153 �S ) z Handle a debug info message.)r� r� r� ro Tr� ZMurmur2�.pyzPython Exceptions)
154 Z isStartedZ
155 hashMethodZhashSeedZ
156 tmpFilePrefixZ
157 tmpFileSuffixr� ZstoppedThreadsZ
158 richRenderingZexceptionPathsr� r� )
159 r� �items�appendr� r
160 r �os�sep�listr� )r r� r� �key�valuer r r r� L s( �
161 ��zDebugger.debugInfoc � s4 �| j �� t� | _ | j �� | j �� }| �||�S )z$Handle an inspect variables message.)r� r2 r* r0 r7 r� )r r� r8 r r r r� d s �
162
163
164 zDebugger.inspectVariablesc � s> �d|d d|d d�}|d d }t �|�}|s.i i d�|d < |d
165 ks(|d kr,d |d
166 < |S i � i }| jsTt� �||i�| }|�dd�dkrS|�di �� |�di �}n3d|� d�}|d d }|d } | �dd| d ||dd�d��I d H }|d
167 r�t|d d i i �\� }� � fd d �|�� D �d�}
168 |
169 |d < d |d
170 < |S ) z(Handle a rich inspect variables message.r� ro Fr� )rH Z sequence_seqr� r� r� Z variableName)�data�metadatar� zspecial variableszfunction variablesTr� r� �errorr� r r z'get_ipython().display_formatter.format(�)�frameIdr �evaluater
171 � clipboard)�
172 expressionr �context�rH r� ro r� N�resultc s i | ] \}}|� v r||�qS r r )r3 �kr� �Z repr_datar r �
173 <dictcomp>� r� z1Debugger.richInspectVariables.<locals>.<dictcomp>) rN � isidentifierr� r Zuser_expressionsr] r� �evalr ) r r� r� �var_nameZ
174 valid_nameZ
175 repr_metadatar r� Zframe_idro r� r r r r� o sN ��
176 �
177 �
178 ��z Debugger.richInspectVariablesc � s` �|d d }|d d }|d d }d|� d�}|d }| � dd |d
179 |||d �d ��I d H S )
180 Nr� ZdstVariableNameZsrcVariableNameZ
181 srcFrameIdz globals()['z']ro r Z
182 setExpressionr
183 )r r r r )r� )r r� Z dst_var_nameZ src_var_nameZ src_frame_idr ro r r r r� � s � ��
184 �zDebugger.copyToGlobalsc � s� �t tj�� �}|�dd�}|�dt|��}g }t||�D ]!}|| }tt|dd�dd�}|r>|�d�r>|� ||j
185 |d�� q d |t|�d
186 �iS ) zHandle a modules message.� startModuler � moduleCountr� N�originr )r6 r� r� r� )r� Z totalModules) r �sysr� �valuesr] rZ �ranger� �endswithr r ) r r� r� r r �modsr� �module�filenamer r r r� � s � �zDebugger.modulesc � s� �i }|d dkr,| j r| j�d� n| �� | _ | j r"| j�d� n
187 d|d dddd �}| j�|d d
188 �}|d
189 urA||�I d
190 H }n | j ra| j�|d d
191 �}|d
192 urY||�I d
193 H }n| �|�I d
194 H }|d d kr{| �� i | _ t
195 � | _ d| _ | j�d � |S )
196 zProcess a request.r� Z
197 initializez The debugger has already startedzThe debugger has startedro � Fr� )r� r� ro r� rH Nr� zThe debugger has stopped) r� rB �infor� r� r] r� r� r� r� rp r� )r r� r� �handlerr r r �process_request� s: �
198 � zDebugger.process_requestN)T) r r r r! r� r� r r� r� r� r� r� �propertyr� r� r� r� r� r� r� r� r8 r� r� r� r� r� r� r� r& r r r r r� 0 s:
199 �!
200
201
202
203 4 r� )1r! r r� r �typingrK �pathlibr r� ZIPython.core.getipythonr Z IPython.core.inputtransformer2r Z
204 tornado.locksr Ztornado.queuesr Z zmq.utilsr Zjupyter_client.jsonutilr � ImportErrorr �compilerr r r
205 Zdebugpy.serverr Z_pydevd_bundler Z&_pydevd_bundle.pydevd_suspended_framesr r Z_is_debugpy_available� Exception�e� __class__r r� ZIDENTITYr r r# r( r* r: r_ r� r r r r �<module> sP � ���
206
207 Vo