Format: application/x-python-code
1 o
2
3 �h�n � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m
4 Z
5 m Z d dl m
6 Z
7 d dlmZ d d lmZ d d
8 lmZ d d lmZ d d lmZ d d
9 lmZ d dlmZ d dlmZ d dlm Z d dl m Z m Z d dl m!Z! d dl"m#Z# d dl$m%Z% d dl&m'Z'm(Z( d dl)m*Z*m+Z+ d dl,m-Z- d dl.m/Z/ G dd� de'�Z0dd � Z1d d � Z2d d � Z3d!d"� Z4d#d$� Z5d%d&� Z6e/�7� d'd(� �Z8G d)d*� d*�Z9d+d,� Z:e �;d-�Z<d.d/� Z=d0d1� Z>d2d3� Z? 4 4d9d5d6�Z@d7d8� ZAdS ):� N)�dedent)� Parameter)�PythonTokenTypes)�tree)�search_ancestor�Leaf)� split_lines)�debug)�settings)�classes)�helpers)�keywords)�
10 complete_dict)�complete_file_name)�imports)�ValueSet)�infer_call_of_leaf�parse_dotted_names)�get_global_filters)� TreeInstance)�DocstringModule)�ParamNameWrapper�
11 SubModuleName)�convert_values�
12 convert_names��cut_value_at_position)�plugin_managerc @ s e Zd Zdd� ZdS )�ParamNameWithEqualsc C s
13 | j d S )N�=)� string_name��self� r# �Ic:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/api/completion.py�get_public_name s
14 z#ParamNameWithEquals.get_public_nameN)�__name__�
15 __module__� __qualname__r% r# r# r# r$ r s r c c sd � | D ],}t |j�D ]$\}}|j}||k r|tjkrq
16 |tjtjfv r.|j|vr.t|j�V q
17 qd S �N) � enumerate�params�kindr �POSITIONAL_OR_KEYWORD� KEYWORD_ONLY�namer �_name)�
18 signatures�positional_count� used_kwargsZcall_sig�i�pr, r# r# r$ �_get_signature_param_names" s �
19 ���r6 c C sp |rdS d}| D ]-}t |j�D ] \}}|j}|tju r dS ||kr/|tjtjfv r/d} nq|s5 |S q|S )NTF)r* r+ r, r �VAR_POSITIONALr- �POSITIONAL_ONLY)r1 r2 r3 Z
20 must_be_kwarg� signaturer4 r5 r, r# r# r$ �_must_be_kwarg. s&
21 ���r: c c s� � t � }tjr |�� }|D ]Q}|j} | |v r| |krq
22 tjr"| �� } tj| ||d�r^tj| ||t |�||d�}
23 |
24 j
25 |
26 j f} | |vr^|� | � |j
27 } | d ur[| �� }
28 |
29 d ur[|
30 jdkr[q
31 |
32 V q
33 d S )N��fuzzy)�is_fuzzy� cached_nameZdel_stmt)�setr
34 Zcase_insensitive_completion�lowerr r �matchr �
35 Completion�lenr/ �complete�add� tree_nameZget_definition�type)�inference_state�completion_names�stackZ like_namer< �imported_namesr> Zcomp_dctr/ �string�new�krF Z
36 definitionr# r# r$ � filter_namesD s< ��
37 ��rO c s dd� |D �� � fdd�| D �S )Nc S s h | ]}|j �qS r# �r/ )�.0�dr# r# r$ � <setcomp>d � z%_remove_duplicates.<locals>.<setcomp>c s g | ] }|j � vr|�qS r# rP �rQ �c��namesr# r$ �
38 <listcomp>e s z&_remove_duplicates.<locals>.<listcomp>r# )� completionsZother_completionsr# rW r$ �_remove_duplicatesc s r[ c C s | j j|dd�}| �|�S )zK
39 Returns the scope in which the user resides. This includes flows.
40 T�Zinclude_prefixes)� tree_node�get_leaf_for_position�create_context)�module_context�position�leafr# r# r$ �get_user_contexth s
41 rc c C s< | j |dd�}t|tjtjf�s |j}t|tjtjf�r|S )NTr\ )r^ �
42 isinstancer ZScopeZFlow�parent)� module_nodera �noder# r# r$ �get_flow_scope_nodep s
43 �rh c C � g S r) r# )�context�
44 function_nameZdecorator_nodesr# r# r$ �complete_param_namesx s rl c @ s| e Zd Z d dd�Zdd� Zdd� Zdd � Zd
45 d � Zd d
46 � Zdd� Z dd� Z
47 dd� Z d dd�Z d!dd�Z
48 dd� Zd d � Zd S )"rB Fc C sB || _ || _|j| _|| _t�| j||�| _|| _|| _ || _
49 d S r) ) �_inference_state�_module_contextr] � _module_node� _code_linesr Zget_on_completion_name�
50 _like_name�_original_position�_signatures_callback�_fuzzy)r" rH r` �
51 code_linesra �signatures_callbackr< r# r# r$ �__init__� s
52 zCompletion.__init__c
53 s( � j j� jdd�}t|� j�\}}}t� j� j|p|� j|d u r"d n|| � jd�}|d urF|sFtt � j
54 � j|||� j � j � j� j� j�
55 �}|d urX|sVd|v rV� �
56 ||�}|S � �|�\}}g }|jd urt|jjdv rt|�t|j�� tt� j
57 |� j� j � j||d��} t|| �t| � fdd�d � S )
58 NTr\ r; �
59 )�import_as_names�dotted_as_names)r> c s. | j �� j� | j �d�| j �d�| j �� fS )N�__�_)r/ �
60 startswithrq r@ )�xr! r# r$ �<lambda>� s
61
62
63 �z%Completion.complete.<locals>.<lambda>)�key)ro r^ rr � _extract_string_while_in_stringr rn rp rt �listr rm rq rs �_complete_in_string�_complete_pythonre rG �extend�extract_imported_namesrO rJ r[ �sorted)
64 r" rb rL �
65 start_leaf�quoteZprefixed_completionsr> rI rK rZ r# r! r$ rD � sH �� �
66 ���zCompletion.completec
67 s� | j j}d | _}| jd | jd t| j� f| _d}zt�|| j || j� | _}W n+ tj
68 yU } z |j j }|dkrE|g fW Y d}~S || �
69 � fW Y d}~S d}~ww t|�� �� d� v r�| jj| jdd�}|�� }| jd }|j| j kr|jks�n |jd }|dur�|} t| dd d
70 d d �} | du r�n=| j}
71 |
72 d kr�| jd } t| t�r�| j d
73 }
74 | jd |kr�|
75 dkr ddg7 � n|
76 d kr͈ g d�7 � n |
77 d krֈ �d� q�g } d}
78 t� fdd�t j t j fD ���r�dd� |D �}t |�}|�r
79 |d dv �r
80 |t| j dd��fS d|v �r-t!|d|v �\}}d|v �o d|v }| | j"|||d �7 } n|d d v �rZ|d dk�rZ| j�| j�}|jd k�rL|�� }| �#|�� �\}}| |7 } nR| �$� �rg| | �%|�7 } nE|d d v �r�|d d v �r�| j&| j� }|�r�|d j'}t|�(� �}|�)� }| t*|||�7 } t+|||�}
81 |
82 �s�| | �
83 � 7 } | | j dd�7 } |
84 �s�| j | jd d d| jd � }| | j,� | �p�|d d!v �o�|d"d� d#k d$�7 } || fS )%a�
85 Analyzes the current context of a completion and decides what to
86 return.
87
88 Technically this works by generating a parser stack and analysing the
89 current stack for possible grammar nodes.
90
91 Possible enhancements:
92 - global/nonlocal search global
93 - yield from / raise from <- could be only exceptions/generators
94 - In args: */**: no completion
95 - In params (also lambda): no completion before =
96 Nr � �.�ifTr\ Zif_stmtZfor_stmtZ
97 while_stmtZtry_stmt�
98 error_nodeZ_stmt�elif�else)�except�finallyr� Fc 3 s � | ]}|� v V qd S r) r# )rQ �t��allowed_transitionsr# r$ � <genexpr> � � z.Completion._complete_python.<locals>.<genexpr>c S � g | ]}|j �qS r# )� nonterminal)rQ �
99 stack_noder# r# r$ rY rT z/Completion._complete_python.<locals>.<listcomp>�����)�as�def�class)� is_functionZ import_stmtZ import_from�import�� only_modules)�trailerZ dotted_nameZ endmarker)�(�,)r� �arglist� decoratorz .;�����z...)� only_values)-rm �grammarrJ rr rC rq � _positionr Zget_stack_at_positionrp Z OnErrorLeaf�
100 error_leaf�value�_complete_global_scoper� Z)_allowed_transition_names_and_token_typesro r^ �get_previous_leaf� start_pos�end_posr rG �childrenrd r �append�anyr �NAME�INDENT�
101 _gather_nodes�_complete_inheritedr �_get_importer_names�_complete_trailer�_is_parameter_completion�_complete_paramsrs Z
102 _call_detailsZiter_used_keyword_argumentsZcount_positional_argumentsr6 r: �_complete_keywords)r" rb r� rJ r> �er� �
103 previous_leaf�indent�stmt�type_�firstrI Z kwargs_onlyZ nonterminals�nodes�levelrX r� �dot�nr1 Z call_detailsr3 r2 � current_liner# r� r$ r� � s�
104 � ���
105 �
106
107
108 �
109
110
111
112 � �
113
114 �
115
116
117
118 � "��zCompletion._complete_pythonc C sL | j d }|jdkrt|j�dkrdS |jdv rdS |jdv o%|jd dkS )Nr� Zlambdefr� T�
119 parameters)Z
120 typedargslistZ varargslistr� )rJ r� rC r� )r" Ztosr# r# r$ r� T s
121
122 z#Completion._is_parameter_completionc C s� | j d }|jdkr| j d }|jdkrTt| j| j�}t|dd�}|d urT|jdkrD|jd }|jdkr8|j}n|jdkrA|g}ng }n|�� }|j d }t
123 ||j |�S g S )
124 N�����r� r� �funcdefr� r �
125 decoratorsr� r� ) rJ r� rc rn r� r rG r� Zget_decoratorsr� rl r� )r" rb r� rj rg r� r� rk r# r# r$ r� a s$
126
127
128
129
130
131
132
133
134 zCompletion._complete_paramsc c s> � |D ]}t |t�r |�� r |r|dv r t�| j|�V qd S )N)�True�False�None)rd �str�isalphar
135 Z KeywordNamerm )r" r� r� rN r# r# r$ r� x s � ��z Completion._complete_keywordsc C sT t | j| j�}t�d|� t| j| j�}t|| j|�}g }|D ]}||�� 7 }q |S )Nzglobal completion scope: %s) rc rn r� r �dbgrh ro r �values)r" rj Zflow_scope_node�filtersrI �filterr# r# r$ r� ~ s �z!Completion._complete_global_scopec C sv | j �|�}t||�}tjd|dd� d }t|�dkr4|\}|�� r4t|j�dkr4|jd }|dv r4|}|| �|�fS )Nz trailer completion values: %sZMAGENTA)�colorr� r )�numpyZ
136 tensorflowZ
137 matplotlibZpandas) rn r_ r r r� rC Z is_moduleZ string_names� _complete_trailer_for_values)r" r� Zinferred_contextr� r> �v� module_namer# r# r$ r� � s
138
139 z Completion._complete_trailerc C s t | j| j�}t||�S r) )rc rn r� �complete_trailer)r" r� � user_contextr# r# r$ r� � s
140 z'Completion._complete_trailer_for_valuesr Tc C s2 dd� |D �}t �| j|| j|�}|j| j|d�S )Nc S r� r# )r� �rQ r� r# r# r$ rY � rT z2Completion._get_importer_names.<locals>.<listcomp>r� )r ZImporterrm rn rI )r" rX r� r� r4 r# r# r$ r� � s z Completion._get_importer_namesc c s� � | j j| jdd�}t�|d�}|du rdS | j�|�}|jd |jd kr(dS |jdd�}t |� |D ]}|�
141 � D ] }|j dk|krF|V q:q4dS )zP
142 Autocomplete inherited methods when overriding in child class.
143 Tr\ ZclassdefNr� )Z is_instance�function) ro r^ r� r r rn Z create_valuer� � get_filters�nextr� �api_type)r" r� rb �clsZ class_valuer� r� r/ r# r# r$ r� � s" � ���z Completion._complete_inheritedc sz � fdd�}t |�}t|dd�� t|� ��}|d dur(dd� |D �}| �|�S t�d � d �}|r;| �|�d
144 �g�S g S ) aU
145 To make it possible for people to have completions in doctests or
146 generally in "Python" code in docstrings, we use the following
147 heuristic:
148
149 - Having an indented block of code
150 - Having some doctest code that starts with `>>>`
151 - Having backticks that doesn't have whitespace inside it
152 c 3 sT � d}� D ]"}|s|� d�s|� d�r t�dd|�V nd V tt�d|��}qd S )NFz>>>� z
153 ^( *>>> ?| +)� z *>>>)r} �re�sub�boolrA )�linesZinclude_next_line�l�ru r# r$ �iter_relevant_lines� s ��z;Completion._complete_in_string.<locals>.iter_relevant_linesT)�keependsr� Nc S s g | ]
154 }|d u r
155 dn|�qS )Nrx r# rU r# r# r$ rY � s z2Completion._complete_in_string.<locals>.<listcomp>z
156 `([^`\s]+)r� )r r r� �_complete_code_linesr� �search�group)r" r� rL r� Zrelevant_code_linesrA r# r� r$ r� � s
157
158 z Completion._complete_in_stringc C sN | j j�d�|��}t| j| j ||d�}t| j |�� ||jdd� | j d��
159 � S )Nr� )Zin_module_contextrH rf ru c _ ri r) r# )�args�kwargsr# r# r$ r � s z1Completion._complete_code_lines.<locals>.<lambda>)ru ra rv r< ) rm r� �parse�joinr rn rB �
160 as_contextr� rt rD )r" ru rf Z module_valuer# r# r$ r� � s" ���z Completion._complete_code_linesN)F)r T)T)r& r' r( rw rD r� r� r� r� r� r� r� r� r� r� r� r# r# r# r$ rB s
161 �0
162
163
164 !rB c C s. g }| D ]}|j jdkrg }q||j7 }q|S )NZ
165 small_stmt)ZdfaZ from_ruler� )rJ r� r� r# r# r$ r� � s r� z^\w*(\'{3}|"{3}|\'|")c s � fdd�}� | j k r
166 dS | jdkr|| �S g }| d ur�| jdkrsd| jv s+d| jv rst| j�dkr6|| �S d }| jsV| �� }|d u sT|jd ksTtd
167 d � |j�� D ��sVd }d �� fd
168 d �|D ��|pd| |d u rjd n|jt | � � fS | j
169 � d kr} dS |� d| � | �� } | d us dS )Nc s� i }| j � d kr� d | j |d< tj| jfi |��}|s dS |�d�}| j � d kr9� d | j|�� k r9dS t| � �|�� d � | |fS )Nr r� �endpos�NNN)�line�column�
170 _string_startrA r� r� �endr )rb r� rA �start�ra r# r$ �return_part_of_leaf s
171 $ z<_extract_string_while_in_string.<locals>.return_part_of_leafr� rL r� �"�'r� r/ c s s � | ]}|d v V qdS )ZrubfNr# rU r# r# r$ r� r� z2_extract_string_while_in_string.<locals>.<genexpr>r� c 3 s � | ]}t |� �V qd S r) r )rQ r� r� r# r$ r� � � r ) r� rG r� rC �prefixr� �allr@ r� r r� �insert)rb ra r� �leavesZ prefix_leafr# r� r$ r� s<
172
173 ��� ��r� c C s� g }|D ]"}|j | jd�D ]}||�� 7 }q
174 |�� s&t|t�r&|t| |�7 }qt|�}|D ]}||vrC|j | jd�D ]}||�� 7 }q:q-|S )N)Z origin_scope)r� r] r� Zis_stubrd r �_complete_getattrr )r� r� rI r� r� Z
175 python_valuesrV r# r# r$ r� 0 s ��r� c C s4 |� d�p |� d�}t�dd� |D ��}|D ]�}|j}|du s$|jdkr%q|�� D ]m}|jdkr1q)|jd }|jd kr<q)|jd
176 }|jd } t|j�d ksW|jd ksW|jd
177 krXq)| jd }
178 |
179 jdksit|
180 j�dk rjq)|� � } |
181 jd
182 } |
183 jd }
184 | �
185 |
186 |
187 j �}t dd� |D ��s�q)| �
188 | �}t| |� S qg S )aV
189 A heuristic to make completion for proxy objects work. This is not
190 intended to work in all cases. It works exactly in this case:
191
192 def __getattr__(self, name):
193 ...
194 return getattr(any_object, name)
195
196 It is important that the return contains getattr directly, otherwise it
197 won't work anymore. It's really just a stupid heuristic. It will not
198 work if you write e.g. `return (getatr(o, name))`, because of the
199 additional parentheses. It will also not work if you move the getattr
200 to some other place that is not the return statement itself.
201
202 It is intentional that it doesn't work in all cases. Generally it's
203 really hard to do even this case (as you can see below). Most people
204 will write it like this anyway and the other ones, well they are just
205 out of luck I guess :) ~dave.
206 � __getattr__�__getattribute__c s s � | ]}|� � V qd S r) )�infer)rQ r/ r# r# r$ r� W s
207 � �
208 �z$_complete_getattr.<locals>.<genexpr>Nr� � return_stmtr� � atom_exprr � r/ �getattrr� � c s s � | ]}|j d kV qdS )�paramN)r� r� r# r# r$ r� v r� )Zget_function_slot_namesr Z from_setsr] rG Ziter_return_stmtsr� rC r� r� Zgotor� r� Z
209 infer_noder� )r� �instancerX Z functions�funcr] r r Zatomr� r� rj Z object_node� name_node� name_list�objectsr# r# r$ r A sD
210 �
211 �
212
213
214
215
216
217
218
219
220
221 � r Fc c s. � |d d� D ]1} g }
222 |D ] } | | j kr*| jd ur!| jdv r!|r!q
223 |
224 t|| �� �7 }
225 q
226 t�d|
227 |d d� � |
228 }q|d �� } |D ]S} | j �� }
229 |rRtj |
230 | |d�sX|s�|
231 | kr�t
232 | t �rgdd� | �� D �}n| g}|rpt |�}|D ]!}|r�t
233 j| |d t| �|d�}nt
234 �| |�}|r�||jkr�|V qrqAd S ) Nr� )�module� namespacez dot lookup on search %s from %s�
235 r; c S r� r# rP )rQ r� r# r# r$ rY � rT z$search_in_module.<locals>.<listcomp>)rJ Zlike_name_lengthr= )r rF r� r� r r r� r@ r rA rd r r r rB rC �NamerG )rH r` rX Z wanted_namesZ wanted_typerD r< Zignore_imports�convert�sZ new_namesr� � last_namerL Zn2Zdef_r# r# r$ �search_in_module� sR �
236 ���
237 �
238 � ��r c C s� g }| j dv rBt| j�D ]5\}}|j dkr5|dkr.| j|d j dkr.| j|d jdkr.q |�|j� q |j dkrA|�t|�� q |S )N)ry rz �import_as_namer/ r r� �keywordr� r )rG r* r� r� r� r� r� )rg rK �index�childr# r# r$ r� � s
239
240
241 �r� )FFFF)Br� �textwrapr �inspectr Zparso.python.tokenr Z parso.pythonr Z
242 parso.treer r Zparsor Zjedir r
243 Zjedi.apir r r
244 Zjedi.api.stringsr Zjedi.api.file_namer Zjedi.inferencer Zjedi.inference.base_valuer Zjedi.inference.helpersr r Zjedi.inference.contextr Zjedi.inference.valuer Z jedi.inference.docstring_utilsr Zjedi.inference.namesr r Z!jedi.inference.gradual.conversionr r Zjedi.parser_utilsr Z jedi.pluginsr r r6 r: rO r[ rc rh Zdecoraterl rB r� �compiler� r� r� r r r� r# r# r# r$ �<module> sZ
245 v
246
247 0A
248 � *