Format: application/x-python-code
1 o
2
3 �h�Z � @ s� d dl mZ d dlmZ d dlmZmZ d dlmZ d dl m
4 Z
5 m Z d dl m
6 Z
7 d dlmZmZ d dlmZ d d lmZ d d
8 lmZ d d lmZmZ d d lmZ d
9 d� Z G dd� d�Z G dd� de �Z G dd� de �Z G dd� d�Z G dd� de e �Z!G dd� de �Z"G dd � d �Z#G d d � d e#�Z$G d d � d e$e �Z%G d!d"� d"e%�Z&G d#d$� d$e&�Z'G d%d&� d&e&�Z(G d'd(� d(e#�Z)G d)d*� d*e �Z*G d+d,� d,e*�Z+G d-d.� d.�Z,G d/d0� d0�Z-G d1d2� d2e-e"�Z.G d3d4� d4e e �Z/G d5d6� d6e-e/�Z0d7S )8� )�abstractmethod)� Parameter)�Optional�Tuple)�search_ancestor)� find_statement_documentation�clean_scope_docstring)�unite)�ValueSet� NO_VALUES)� inference_state_method_cache)�
10 docstrings)�memoize_method)�
11 deep_ast_copy�infer_call_of_leaf)�plugin_managerc C s* d}| D ]}|r |d7 }||� � 7 }q|S )N� z
12 ------------------------------
13 )� py__doc__)�names�doc�name� r �Jc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/inference/names.py�_merge_name_docs s r c @ s� e Zd ZU dZeeeef ed< eed< dZ dZ
14 dZ e dd� �Z
15 e dd� �Zd d
16 d �Zd d
17 � Zdd� Zdd� Zdd� Zdd� Zdd� Zedd� �Zdd� ZdS ) �AbstractNameDefinitionN� start_pos� string_nameTc C � t ��N��NotImplementedError��selfr r r �infer& � z AbstractNameDefinition.inferc C s | hS r r r! r r r �goto* s zAbstractNameDefinition.gotoFc C s6 | � � }|d u s
18 |s |S | �� j}|d u rd S || S r )�_get_qualified_names�get_root_context� string_names)r" �include_module_namesZqualified_names� module_namesr r r �get_qualified_names0 s
19 z*AbstractNameDefinition.get_qualified_namesc C � d S r r r! r r r r& : r$ z+AbstractNameDefinition._get_qualified_namesc C �
20 | j �� S r )�parent_contextr' r! r r r r' > �
21 z'AbstractNameDefinition.get_root_contextc C � | j S r )r r! r r r �get_public_nameA � z&AbstractNameDefinition.get_public_namec C s2 | j d u rd| jj| jf S d| jj| j| j f S )Nz<%s: string_name=%s>z!<%s: string_name=%s start_pos=%s>)r � __class__�__name__r r! r r r �__repr__D s
22
23 �z AbstractNameDefinition.__repr__c C � dS )NFr r! r r r � is_importJ � z AbstractNameDefinition.is_importc C r6 )Nr r r! r r r r M r8 z AbstractNameDefinition.py__doc__c C � | j jS r )r. �api_typer! r r r r: P � z AbstractNameDefinition.api_typec C r6 )z�
24 Returns either None or the value that is public and qualified. Won't
25 return a function, because a name in a function is never public.
26 Nr r! r r r � get_defining_qualified_valueT s z3AbstractNameDefinition.get_defining_qualified_value�F)r4 �
27 __module__� __qualname__r r r �int�__annotations__�strr. � tree_name�
28 is_value_namer r# r% r+ r& r' r1 r5 r7 r �propertyr: r< r r r r r s*
29
30
31
32
33
34 r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�AbstractArbitraryNamez�
35 When you e.g. want to complete dicts keys, you probably want to complete
36 string literals, which is not really a name, but for Jedi we use this
37 concept of Name for completions as well.
38 Fc C s || _ || _|j| _d S r )�inference_stater Zbuiltins_moduler. )r" rG �stringr r r �__init__d s z AbstractArbitraryName.__init__c C � t S r �r r! r r r r# i r8 zAbstractArbitraryName.inferN)r4 r> r? �__doc__rD rI r# r r r r rF \ s
39 rF c s^ e Zd Zdd� Zd� fdd� Zdd� Zdd � Zd
40 d � Zd d
41 � Ze dd� �Z
42 e dd� �Z � Z S )�AbstractTreeNamec C � || _ || _d S r )r. rC )r" r. rC r r r rI n �
43 zAbstractTreeName.__init__Fc sd t | jdd�}|d ur,|jdkr| �� �� �� s,|r*|js*tdd� |�| j�D ��S d S t� � |�S )N� import_name� import_from� c s s � | ]}|j V qd S r )�value)�.0�nr r r � <genexpr>| s � z7AbstractTreeName.get_qualified_names.<locals>.<genexpr>)
44 r rC �levelr' � get_value�
45 is_package�tupleZget_path_for_name�superr+ )r" r) Z import_node�r3 r r r+ r s �
46 z$AbstractTreeName.get_qualified_namesc C s$ | j �� }|d u r d S || jjf S r )r. r+ rC rS )r" Z parent_namesr r r r& � s
47 z%AbstractTreeName._get_qualified_namesc C s$ | � � rtd��| jr| j�� S d S )Nz&Shouldn't really happen, please report)r7 r r. rX r! r r r r< � s
48
49 z-AbstractTreeName.get_defining_qualified_valuec s| | j � | j��jdd�}|d ur7|j}|dkr#�jjdv}|r"| gS n*|dv r4ddlm} |� ��}|S | gS ddlm} |� ��}|d urMd d
50 � |D �S �j}|j} | d kr�|jd d
51 kr�|jd �kr�|j}
52 |
53 jdkrp|
54 j}
55 |
56 jdkr�|
57 jdkr�� � |
58 jd �} n6|
59 jj�
60 |
61 �} |
62 jjd | � }
63 |
64 d dkr�|
65 � d� � � |
66 d �} ddl m
67 } |
68 d d � D ]}
69 |� | |
70 �} q�g }| D ]}|�� D ]}|�� D ]
71 }|j�jkr�|�|� q�q�q�|S n.| dk�r
72 |j�
73 ��}|dk�r
74 t|�}g |j|d d �< � � |�}t� �fdd�|D ��S | dk�r%|jd dk�r%t� �dd�}|j�� d�S t�dd��p-�}|jdk�r6�}� j�|jd�S ) NT�Zimport_name_always� expr_stmt)�power�trailer)rQ rP r )� goto_import)�%follow_error_node_imports_if_possiblec S � g | ]}|j �qS r �r �rT rS r r r �
75 <listcomp>� � z)AbstractTreeName.goto.<locals>.<listcomp>ZargumentrR �=Zarglist�classdef� decorator�await)�
76 infer_trailerZ dotted_namec 3 s � | ]
77 }|j �� d �V qdS )�Z name_contextN)r% re ��contextr r r rV � s
78 � �
79 �z(AbstractTreeName.goto.<locals>.<genexpr>r` �.)Zcut_own_trailerrm Zlambdef)�position)r. rC �get_definition�type�parent�jedi.inference.importsra rb �children�
80 infer_node�index�pop�jedi.inference.syntax_treerl Zget_signaturesZget_param_namesr rS �appendr r r r% r r )r" �
81 definition�type_Zis_simple_namera r* rb �values�parZ node_typer` Z value_set�iZto_inferrl Z param_namesrS � signature�
82 param_namerx Z
83 new_dotted�stmtr rn r r% � s� �
84
85 $
86
87
88
89
90 ����
91
92
93 ��� zAbstractTreeName.gotoc C s t | jdd�}|d uS )NrQ rP �r rC )r" �impr r r r7 � s zAbstractTreeName.is_importc C r9 r )rC rS r! r r r r � r; z AbstractTreeName.string_namec C r9 r )rC r r! r r r r � r; zAbstractTreeName.start_posr= )
94 r4 r> r? rI r+ r& r< r% r7 rE r r �
95 __classcell__r r r\ r rM m s H
96 rM c sH e Zd Zdd� Zdd� Zdd� Z� fdd�Zd d
97 � Zed d � �Z � Z
98 S )
99 �ValueNameMixinc C s t | jg�S r )r
100 �_valuer! r r r r# � � zValueNameMixin.inferc C sF | j �� }|s!| j �� r!ddlm} || gdd�}| |vr!t|�S |S )Nr ��
101 convert_namesF�Zprefer_stub_to_compiled)r� r �is_stub�!jedi.inference.gradual.conversionr� r )r" r r� r r r r r � s
102 zValueNameMixin.py__doc__c C r- r )r� r+ r! r r r r& � r/ z#ValueNameMixin._get_qualified_namesc s | j d u r
103 | j�� S t� �� S r )r. r� �
104 as_contextr[ r' r! r\ r r r' � s
105
106
107 z ValueNameMixin.get_root_contextc C s, | j }|d ur|�� s|�� r| j �� S d S r )r. Z is_module�is_classrX )r" ro r r r r< � s
108 z+ValueNameMixin.get_defining_qualified_valuec C r9 r )r� r: r! r r r r: � r; zValueNameMixin.api_type) r4 r> r? r# r r& r' r< rE r: r� r r r\ r r� � s r� c s$ e Zd Z� fdd�Zdd� Z� ZS )� ValueNamec s t � �|j|� || _d S r )r[ rI r. r� )r" rS rC r\ r r rI s
109 zValueName.__init__c C s t | jjg�S r )r
110 r� r r! r r r r% s zValueName.goto)r4 r> r? rI r% r� r r r\ r r� s r� c @ sX e Zd Zedddddd�Zdd� Zedd � �Zd
111 d � Zed d
112 � �Z e
113 dd�dd� �Z dS )�TreeNameDefinition�module�function�param�class)rP rQ Zfuncdefr� ri c C s ddl m} || jj| j| j�S )Nr )�tree_name_to_values)rz r� r. rG rC )r" r� r r r r# s �zTreeNameDefinition.inferc C s* | j jdd�}|d u r
114 dS | j�|jd�S )NTr] � statement)rC rr �
115 _API_TYPES�getrs )r" r| r r r r: s zTreeNameDefinition.api_typec C s� g }d}| j j}| j }|dure|jdv rLt|j�D ]+\}}||krDt|d �}|r:tt|j�| d �}t|| �}|�d||f� nqt d��d}n|jdkrTd}n|jd v r\ |S |}|j}|dus|S )
116 a�
117 Returns an array of tuple(int, node) of the indexes that are used in
118 tuple assignments.
119
120 For example if the name is ``y`` in the following code::
121
122 x, (y, z) = 2, ''
123
124 would result in ``[(1, xyz_node), (0, yz_node)]``.
125
126 When searching for b in the case ``a, *b, c = [...]`` it will return::
127
128 [(slice(1, -1), abc_node)]
129 FN)ZtestlistZ
130 testlist_compZtestlist_star_exprZexprlist� r z Couldn't find the assignment.Z star_exprT)r^ Z
131 sync_comp_for)
132 rC rt rs � enumeraterv r@ �len�slice�insert� LookupError) r" ZindexesZ is_star_expr�node�comparer� �childrx Zfrom_endr r r �assignment_indexes& s4
133 �
134
135 ��z%TreeNameDefinition.assignment_indexesc C r9 r )r. rG r! r r r rG O s z"TreeNameDefinition.inference_stater )�defaultc C s� | j }|dv r*| j�� �� r#ddlm} || gdd�}| |vr#t|�S t| j� � �S |dkr:| �
136 � }| |vr:t|�S |dkrJ| j� � rJt | j� � �S dS ) N)r� r� rE r r� Fr� r� r� r )
137 r: r. r' r� r� r� r r rC rr r% Z
138 is_definitionr )r" r: r� r r r r r T s z TreeNameDefinition.py__doc__N) r4 r> r? �dictr� r# rE r: r� rG r r r r r r r�
139 s �
140 )
141 r� c @ s2 e Zd Zd dd�Zd dd�Zdd� Zd
142 d d
143 �Zd S )� _ParamMixinTc C �( t jt jg}|r|�t j� | �� |v S r )r �POSITIONAL_ONLY�POSITIONAL_OR_KEYWORDr{ �VAR_POSITIONAL�get_kind)r" Z include_star�optionsr r r �maybe_positional_argumentl � z%_ParamMixin.maybe_positional_argumentc C r� r )r � KEYWORD_ONLYr� r{ � VAR_KEYWORDr� )r" Z
144 include_starsr� r r r �maybe_keyword_argumentr r� z"_ParamMixin.maybe_keyword_argumentc C �( | � � }|tjkr dS |tjkrdS dS )N�*z**r �r� r r� r� �r" �kindr r r � _kind_stringx s
145
146 z_ParamMixin._kind_stringFc C r, r r )r" r) r r r r+ � r8 z _ParamMixin.get_qualified_namesN)Tr= )r4 r> r? r� r� r� r+ r r r r r� k s
147
148
149 r� c @ s< e Zd ZdZdd� Zdd� Zdd� Zedd � �Zd
150 d � Z d S )
151 �ParamNameInterfacer� c C r r r r! r r r r� � r8 zParamNameInterface.get_kindc C r r r r! r r r � to_string� r8 z ParamNameInterface.to_stringc C r6 )as
152 For dealing with type inference and working around the graph, we
153 sometimes want to have the param name of the execution. This feels a
154 bit strange and we might have to refactor at some point.
155
156 For now however it exists to avoid infering params when we don't really
157 need them (e.g. when we can just instead use annotations.
158 Nr r! r r r �get_executed_param_name� s z*ParamNameInterface.get_executed_param_namec C r� )NrR r� r r� r� r r r �
159 star_count� s
160
161 z ParamNameInterface.star_countc C rJ r rK r! r r r �
162 infer_default� r8 z ParamNameInterface.infer_defaultN)
163 r4 r> r? r: r� r� r� rE r� r� r r r r r� � s
164 r� c @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS )�BaseTreeParamNameNc C sX | � � | �� }| j}| j}|d ur |d|jdd� 7 }|d ur*|d|jdd� 7 }|S )Nz: F)Zinclude_prefixrh )r� r1 �annotation_node� default_node�get_code)r" �output�
165 annotationr� r r r r� � s zBaseTreeParamName.to_stringc C s | j }|�d�r|dd � }|S )N�__r� )r �
166 startswith�r" r r r r r1 � s
167 z!BaseTreeParamName.get_public_namec K s | gS r r )r" �kwargsr r r r% � r2 zBaseTreeParamName.goto)r4 r> r? r� r� r� r1 r% r r r r r� � s
168 r� c s^ e Zd Z� fdd�Zdd� Zedd� �Zdd d
169 �Zd d � Zed
170 d� �Z dd� Z
171 dd� Z � Z S )�_ActualTreeParamNamec s t � �|�� |� || _d S r )r[ rI Zget_default_param_context�function_value)r" r� rC r\ r r rI � s �
172 z _ActualTreeParamName.__init__c C s t | jd�S )Nr� r� r! r r r �_get_param_node� r� z$_ActualTreeParamName._get_param_nodec C �
173 | � � jS r )r� r� r! r r r r� � �
174 z$_ActualTreeParamName.annotation_nodeTFc C s0 ddl m} || j| �� |d�}|r|�� }|S )Nr )� infer_param)� ignore_stars)Z!jedi.inference.gradual.annotationr� r� r� �execute_annotation)r" r� r� r� r~ r r r �infer_annotation� s
175 �z%_ActualTreeParamName.infer_annotationc C s | j }|d u r tS | j�|�S r )r� r r. rw )r" r� r r r r� � s z"_ActualTreeParamName.infer_defaultc C r� r )r� r� r! r r r r� � r� z!_ActualTreeParamName.default_nodec C s� | � � }|jdkr tjS |jdkrtjS |jj�d�r tjS |j }d}|j
176 D ]*}|r4|dkr3tj S q&|dkr=tj S |j dkrP|jrJtj S ||krPd}q&tj
177 S ) NrR r� r� F�/r� r� T)r� r� r r� r� r rS r� r� rt rv r� rs r� )r" Z
178 tree_paramrt Zparam_appeared�pr r r r� � s.
179
180
181
182 �
183
184
185 �z _ActualTreeParamName.get_kindc C s& | � � }|r|S t�| j| �� �}|S r )r� r
186 r� r� r� )r" r~ Z
187 doc_paramsr r r r# � s
188 z_ActualTreeParamName.infer)TF)
189 r4 r> r? rI r� rE r� r� r� r� r� r# r� r r r\ r r� � s
190
191
192 r� c s@ e Zd Zejdd�� fdd��Zejdd�� fdd��Z� ZS )�AnonymousParamNameZgoto_anonymous_paramrd c s
193 t � �� S r )r[ r% r! r\ r r r% r� zAnonymousParamName.gotoZinfer_anonymous_paramc s� t � �� }|r |S ddlm} | �� }|| j|j�}|r |S |jdkr1ddlm } || jj
194 g �}n"|jdkrDddlm } || jj
195 i �}n|j d u rKt
196 S | jj�|j �S t|h�S )Nr )�dynamic_param_lookuprR )� FakeTupler� )�FakeDict)r[ r# Z jedi.inference.dynamic_paramsr� r� r� �position_indexr� Z jedi.inference.value.iterabler� rG r� r� r r. rw r
197 )r" r~ r� r� r� rS r� r\ r r r# s$
198
199
200
201
202 zAnonymousParamName.infer)r4 r> r? r Zdecorater% r# r� r r r\ r r� s
203
204
205 r� c s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )� ParamNamec s t � �||� || _d S r )r[ rI � arguments)r" r� rC r� r\ r r rI " s
206 zParamName.__init__c s t � �� }|r |S | �� �� S r )r[ r# r� )r" r~ r\ r r r# & s
207 zParamName.inferc C s( ddl m} || j| j�}|| �� j S )Nr )�get_executed_param_names)Zjedi.inference.paramr� r� r� r� r� )r" r� Z params_namesr r r r� - s z!ParamName.get_executed_param_name)r4 r> r? rI r# r� r� r r r\ r r� ! s r� c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�ParamNameWrapperc C �
208 || _ d S r )�_wrapped_param_name)r" r� r r r rI 4 r/ zParamNameWrapper.__init__c C � t | j|�S r )�getattrr� r� r r r � __getattr__7 r� z ParamNameWrapper.__getattr__c C � d| j j| jf S )Nz<%s: %s>)r3 r4 r� r! r r r r5 : � zParamNameWrapper.__repr__N�r4 r> r? rI r� r5 r r r r r� 3 � r� c @ sZ e Zd ZdZdZdd� Zddd�Zedd � �Ze d
209 d � �Z
210 d d
211 � Z edd� �Z dd� Z
212 dS )�
213 ImportName�rR r r c C rN r )�_from_module_contextr )r" r. r r r r rI B rO zImportName.__init__Fc C sH |r"| j r | j dksJ d��| jj}|d u r|S || jf S | jfS dS )NrR z(Everything else is not supported for nowr )�_levelr� r( r )r" r) r* r r r r+ F s z ImportName.get_qualified_namesc C s& | j }| �� }|s |S tt|���� S r )r� r# �next�iterr� )r" �mZ
214 import_valuesr r r r. Q s
215 zImportName.parent_contextc C s. ddl m} | j}||j| jg|| jd��� S )Nr )�Importer)rW )ru r� r� rG r r� �follow)r" r� r� r r r r# [ s zImportName.inferc C s dd� | � � D �S )Nc S rc r rd )rT r� r r r rf b rg z#ImportName.goto.<locals>.<listcomp>)r# r! r r r r% a r� zImportName.gotoc C r6 )Nr� r r! r r r r: d r$ zImportName.api_typec C s t | �� �S r )r r% r! r r r r h r� zImportName.py__doc__Nr= )r4 r> r? r r� rI r+ rE r. r r# r% r: r r r r r r� > s
216
217
218
219 r� c @ s e Zd ZdZdS )�
220 SubModuleNamerR N)r4 r> r? r� r r r r r� l s r� c @ r� )� NameWrapperc C r� r )�
221 _wrapped_name)r" Z wrapped_namer r r rI q r/ zNameWrapper.__init__c C r� r )r� r� r� r r r r� t r� zNameWrapper.__getattr__c C r� )Nz%s(%s))r3 r4 r� r! r r r r5 w r� zNameWrapper.__repr__Nr� r r r r r� p r� r� c � e Zd Z� fdd�Z� ZS )�
222 StubNameMixinc sd ddl m} | g}| jdkr d| j�� jv r dd� | �� D �}||dd�}| |v r.t� �� S t |�S ) Nr r� r� rh c S rc r rd )rT �vr r r rf � rg z+StubNameMixin.py__doc__.<locals>.<listcomp>Fr� )
223 r� r� r: rC rr rv r# r[ r r )r" r� r r\ r r r | s
224 zStubNameMixin.py__doc__)r4 r> r? r r� r r r\ r r� { � r� c r� )�StubNamec sJ t � �� }| jdkr#| �� �� dkr#ddlm� t� fdd�|D ��S |S )N� version_info�sysr �� VersionInfoc 3 s � | ]}� |�V qd S r r )rT �cr� r r rV � s � z!StubName.infer.<locals>.<genexpr>)r[ r# r r' Z
225 py__name__Z!jedi.inference.gradual.stub_valuer� r
226 )r" �inferredr\ r� r r# � s
227
228 zStubName.infer)r4 r> r? r# r� r r r\ r r� � r� r� c @ s$ e Zd ZdZdd� Zedd� �ZdS )�
229 ModuleNamer� c C rN r )r� �_name)r" rS r r r r rI � rO zModuleName.__init__c C r0 r )r� r! r r r r � s zModuleName.string_nameN)r4 r> r? r rI rE r r r r r r� � s
230 r� c @ s e Zd ZdS )�StubModuleNameN)r4 r> r? r r r r r� � s r� N)1�abcr �inspectr �typingr r Z
231 parso.treer Zjedi.parser_utilsr r Zjedi.inference.utilsr Zjedi.inference.base_valuer
232 r Zjedi.inference.cacher Zjedi.inferencer
233 Z
234 jedi.cacher Zjedi.inference.helpersr r Z jedi.pluginsr r r rF rM r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> sB @w ^!E .