Format: application/x-python-code
1 o
2
3 �h [ � @ s d Z ddlmZ ddlmZ ddlmZmZmZ ddlm Z m
4 Z
5 m Z m Z ddl
6 mZmZ ddlmZ ddlmZmZ dd lmZmZmZmZmZmZm Z dd
7 l m Z dd l m Z dd l!m"Z" G d
8 d� d�Z#G dd� dee#�Z$G dd� de$�Z%dd� Z&G dd� d�Z'G dd� d�Z(G dd� dee#�Z)G dd � d e'�Z*G d d � d e*e)�Z+G d d � d e*e)�Z,G d!d"� d"e*e$�Z-G d#d$� d$�Z.G d%d&� d&e'e)e.�Z/G d'd(� d(e)�Z0G d)d*� d*e(e0e.�Z1G d+d,� d,e)�Z2G d-d.� d.e2�Z3G d/d0� d0e2�Z4G d1d2� d2e(e)e.�Z5G d3d4� d4e)�Z6d5d6� Z7G d7d8� d8e �Z8d9S ):zd
9 Contains all classes and functions to deal with lists, dicts, generators and
10 iterators in general.
11 � )�compiled)�analysis)�LazyKnownValue�LazyKnownValues�
12 LazyTreeValue)�get_int_or_none� is_string�reraise_getitem_errors�SimpleGetItemNotFound)�
13 safe_property�to_list)� inference_state_method_cache)�LazyAttributeOverwrite�publish_method)�ValueSet�Value� NO_VALUES�ContextualizedNode�iterate_values�sentinel�LazyValueWrapper)�get_sync_comp_fors��CompForContext)�check_array_additionsc @ s$ e Zd Zddd�Zdd� ZejZdS )�
14 IterableMixinNc C s
15 | � |�S �N)�
16 py__iter__��self�contextualized_node� r! �Sc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/inference/value/iterable.py�
17 py__next__ �
18 zIterableMixin.py__next__c C � t t�| jd�g�S �N�None�r r �builtin_from_name�inference_state�r r! r! r" �py__stop_iteration_returns � z(IterableMixin.py__stop_iteration_returnsr )�__name__�
19 __module__� __qualname__r# r, r �get_safe_valuer! r! r! r" r s
20
21 r c @ sl e Zd ZdZdd� Zdd� Zdd� Zed�d d
22 � �Zed �ed �d
23 d� ��Z dd� Z
24 e dd� �Z dd� Z
25 dS )�
26 GeneratorBaseNc C s | � � �� \}|S r )�_get_cls�execute_annotation)r �instancer! r! r" �_get_wrapped_value) s z GeneratorBase._get_wrapped_valuec C s | j j�d�\}|S �N� Generator)r* Z
27 typing_moduleZpy__getattribute__)r � generatorr! r! r" r3 - s zGeneratorBase._get_clsc C s dS )NTr! r+ r! r! r" �
28 py__bool__1 � zGeneratorBase.py__bool__�__iter__c C s
29 t | g�S r )r �r � argumentsr! r! r" �_iter4 �
30 zGeneratorBase._iter�send�__next__c C � t �dd� | �� D ��S )Nc s � � | ]}|� � V qd S r ��infer��.0�
31 lazy_valuer! r! r" � <genexpr>; � � z&GeneratorBase._next.<locals>.<genexpr>�r � from_setsr r= r! r! r" �_next8 s zGeneratorBase._nextc C r% r& r( r+ r! r! r" r, = r- z(GeneratorBase.py__stop_iteration_returnsc C s t �| d�S r7 )r �CompiledValueNamer+ r! r! r" �name@ � zGeneratorBase.namec C s4 ddl m} | �� �� }||ttf�}| �� �|�S )Nr ��TupleGenericManager)� jedi.inference.gradual.genericsrS �merge_types_of_iterate� py__class__r r3 Z
32 with_generics)r rS Z
33 gen_valuesZgmr! r! r" �get_annotated_class_objectD s z(GeneratorBase.get_annotated_class_object)r. r/ r0 �
34 array_typer6 r3 r: r r? rN r, �propertyrP rW r! r! r! r" r2 & s
35
36 r2 c s: e Zd ZdZ� fdd�Zd dd�Zdd� Zd d
37 � Z� ZS ) r8 z Handling of `yield` functions.c � t � �|� || _d S r )�super�__init__�_func_execution_context)r r* Zfunc_execution_context�� __class__r! r" r\ M �
38 zGenerator.__init__Nc C s" | j �� }|r |�|�S | j �� S r )r] Zinfer_annotations�iterateZget_yield_lazy_values)r r � iteratorsr! r! r" r Q s
39
40
41 zGenerator.py__iter__c C �
42 | j �� S r )r] Zget_return_valuesr+ r! r! r" r, W r$ z$Generator.py__stop_iteration_returnsc C � dt | �j| jf S �Nz
43 <%s of %s>)�typer. r] r+ r! r! r" �__repr__Z r- zGenerator.__repr__r ) r. r/ r0 �__doc__r\ r r, rg �
44 __classcell__r! r! r^ r" r8 K s
45 r8 c C s� |j d }|j d }|dkr9|j d j d dkr6|j d }|jdkr'|j d }t| |||j d |j d d�S t}n
46 |d kr@t}n|d
47 krFt}|j d }|jdkrU|j d }|| |||j d d �S ) Nr � �{�:� �comp_for� )�sync_comp_for_node�key_node�
48 value_node�(�[)�defining_contextrp �
49 entry_node)�childrenrf �DictComprehension�SetComprehension�GeneratorComprehension�ListComprehension)r* �value�atomZbracketZtest_list_comp�
50 sync_comp_for�clsr! r! r" �comprehension_from_atom^ s8
51
52
53
54
55 �
56
57
58 �r� c @ sL e Zd Ze� dd� �Zd
59 dd�Zeg d�edd� ��Zd
60 d d
61 �Zd d � Z dS )�ComprehensionMixinc C s
62 t ||�S r r )r �parent_contextrn r! r! r" �_get_comp_for_context� r@ z(ComprehensionMixin._get_comp_for_contextNc
63 c s � |d }|j jdk}|jd }|p| j}|�|�}t||�}|j||d�}|jd } t|�D ][\}
64 } | �� } t || | �}
65 | �
66 ||�}|� ||
67 ��9 z| � |dd � |�E d H W n! t
68 yz |�| j�}| jdkru||�| j�fV n|V Y nw W d � n1 s�w Y q/d S )Nr rn rm )�is_asyncrj �dict)�parentrf rw �_defining_context�
69 infer_noder ra � enumeraterF �unpack_tuple_to_dictr� Zpredefine_names�_nested�
70 IndexError� _entry_noderX � _value_node)r � comp_forsr� rn r� Z
71 input_nodeZ input_typesZcnZiterated�exprlist�irI �types�dct�contextr! r! r" r� � s< �
72
73
74
75
76 �
77 ��� ���zComprehensionMixin._nested��defaultc c s$ � t t| j��}| �|�E d H d S r )�tupler �_sync_comp_for_noder� )r r� r! r! r" �_iterate� s �zComprehensionMixin._iteratec c s � | � � D ]}t|�V qd S r �r� r )r r �set_r! r! r" r � s � �z ComprehensionMixin.py__iter__c C rd re )rf r. r� r+ r! r! r" rg � r- zComprehensionMixin.__repr__r )
78 r. r/ r0 r
79 r� r� r r� r rg r! r! r! r" r� � s
80
81
82 r� c @ s e Zd Zdd� ZdS )�
83 _DictMixinc C s t dd� | �� D ��S )Nc s rD r )rV )rH Zc_setr! r! r" rJ � rK z+_DictMixin._get_generics.<locals>.<genexpr>)r� �get_mapping_item_valuesr+ r! r! r" �
84 _get_generics� s z_DictMixin._get_genericsN)r. r/ r0 r� r! r! r! r" r� � s r� c @ sZ e Zd ZdZedd� �Zdd� Zedd�dd � �Zd
85 d � Z d d
86 � Z
87 e dd� �Z dd� Z
88 dS )�Sequencer5 c C s t �| | j�S r )r rO rX r+ r! r! r" rP � s z
89 Sequence.namec C s | � � �� fS r )rU rV r+ r! r! r" r� � � zSequence._get_genericsr! r� c C � | � � S r )r� r+ r! r! r" �_cached_generics� � zSequence._cached_genericsc C sD ddl m} ddlm} t�| j| j�}|||| �� ��� � \}|S )Nr )� GenericClassrR )
90 Zjedi.inference.gradual.baser� rT rS r r) r* rX r� r4 )r r� rS �klass�cr! r! r" r6 � s
91 ��zSequence._get_wrapped_valuec C s d S r r! r+ r! r! r" r: � r; zSequence.py__bool__c C s | j jS r )r* Zbuiltins_moduler+ r! r! r" r� � r� zSequence.parentc C s | j dkr | �� S tt| g��S )Nr� )rX � _dict_valuesr r )r Zindex_value_setr r! r! r" �
92 py__getitem__� s
93 zSequence.py__getitem__N)r. r/ r0 Zapi_typerY rP r� r
94 r� r6 r: r r� r� r! r! r! r" r� � s
95
96
97
98 r� c s e Zd Z� fdd�Z� ZS )�_BaseComprehensionc s0 |j dksJ �t� �|� || _|| _|| _d S �Nr~ )rf r[ r\ r� r� r� )r r* ru rp rv r^ r! r" r\ � s
99
100 z_BaseComprehension.__init__)r. r/ r0 r\ ri r! r! r^ r" r� � s r� c @ s e Zd ZdZdd� ZdS )r{ �listc C s` t |t�r
101 t| g�S t| �� �}ttt�� || }W d � |�� S 1 s'w Y |�� S r ) �
102 isinstance�slicer r� r r r� � TypeErrorrF )r �indexZ all_typesrI r! r! r" �py__simple_getitem__� s
103
104
105
106 ��z&ListComprehension.py__simple_getitem__N)r. r/ r0 rX r� r! r! r! r" r{ � s r{ c @ � e Zd ZdZdS )ry �setN�r. r/ r0 rX r! r! r! r" ry � � ry c @ s e Zd ZdS )rz N)r. r/ r0 r! r! r! r" rz � s rz c @ s e Zd Zdd� Zdd� ZdS )�
107 _DictKeyMixinc C s | � � | �� fS r )�
108 _dict_keysr� r+ r! r! r" r� � s z%_DictKeyMixin.get_mapping_item_valuesc C r� r )r� r+ r! r! r" �get_key_values� r� z _DictKeyMixin.get_key_valuesN)r. r/ r0 r� r� r! r! r! r" r� � s r� c sj e Zd ZdZ� fdd�Zddd�Zdd� Zd d
109 � Zd d � Ze d
110 �dd� �Z
111 e d�dd� �Z dd� Z � Z
112 S )rx r� c s6 |j dksJ �t� �|� || _|| _|| _|| _d S r� )rf r[ r\ r� r� r� r� )r r* ru rp rq rr r^ r! r" r\ s
113 zDictComprehension.__init__Nc c s" � | � � D ] \}}t|�V qd S r r� )r r �keys�valuesr! r! r" r s � �z DictComprehension.py__iter__c C s@ | � � D ]\}}|D ]}|jt� d�|kr| S q
114 qt� �)Nr� )r� r1 �objectr
115 )r r� r� r� �kr! r! r" r� s ��z&DictComprehension.py__simple_getitem__c C rC )Nc s s � | ]\}}|V qd S r r! �rH r� r� r! r! r" rJ rK z/DictComprehension._dict_keys.<locals>.<genexpr>�r rM r� r+ r! r! r" r� � z DictComprehension._dict_keysc C rC )Nc s s � | ]\}}|V qd S r r! r� r! r! r" rJ rK z1DictComprehension._dict_values.<locals>.<genexpr>r� r+ r! r! r" r� r� z DictComprehension._dict_valuesr� c C � t | �� �}tt| j|g�g�S r �r r� r �FakeListr* �r r> rI r! r! r" �_imitate_values" � z!DictComprehension._imitate_values�itemsc �( � fdd�� � � D �}tt� j|�g�S )Nc s, g | ]\}}t t� jt|�t|�g���qS r! )r � FakeTupler* r )rH �keyr| r+ r! r" �
116 <listcomp>) s �����z4DictComprehension._imitate_items.<locals>.<listcomp>)r� r r� r* �r r> Z lazy_valuesr! r+ r" �_imitate_items' s
117 � z DictComprehension._imitate_itemsc C s g S r r! r+ r! r! r" �exact_key_items6 s z!DictComprehension.exact_key_itemsr )r. r/ r0 rX r\ r r� r� r� r r� r� r� ri r! r! r^ r" rx s
118
119
120 rx c sb e Zd ZdZdddd�Z� fdd�Z� fdd �Zd
121 d � Zdd
122 d�Zdd� Z dd� Z
123 dd� Z � Z S )�SequenceLiteralValue)�testlist_star_expr�testlistZ
124 subscriptlistr� r� r� )rs rt rk c sH t � �|� || _|| _| jj| jv rd| _d S tj|j d | _ d S )Nr� r )
125 r[ r\ r} r� rf � _TUPLE_LIKErX r� �mappingrw �r r* ru r} r^ r! r" r\ B s
126 z SequenceLiteralValue.__init__c s* | j dkrtdd� | �� D ��S t� �� S )Nr� c s s � | ] }|� � �� V qd S r )rF rV )rH �xr! r! r" rJ O s � z5SequenceLiteralValue._get_generics.<locals>.<genexpr>)rX r� r r[ r� r+ r^ r! r" r� M s
127
128 z"SequenceLiteralValue._get_genericsc C sn t |t�r
129 t| g�S tttt�� | �� | }W d � n1 s!w Y |dks/|jdkr1t S | j
130 � |�S )�9Here the index is an int/str. Raises IndexError/KeyError.Nrl � subscript) r� r� r r r� �KeyErrorr� �get_tree_entriesrf r r� r� )r r� �noder! r! r" r� R s
131
132 � z)SequenceLiteralValue.py__simple_getitem__Nc c s^ � | � � D ] }|dks|jdkr tt| jddd��V qt| j|�V qt| j| �E dH dS )��
133 While values returns the possible values for any array field, this
134 function returns the value for a certain index.
135 rl r� N)r� rf r �Slicer� r r )r r r� r! r! r" r ] s � z SequenceLiteralValue.py__iter__c C s t | �� �S r )�lenr� r+ r! r! r" � py__len__k rQ z SequenceLiteralValue.py__len__c C s | j j}| j j| jv r|d d d� S |d }|dv r g S |jdkr.dd� |jd d d� D �S |jdkr|g }t|j�}|D ]=}|dkrLt|� t|d � q<t|d �}|d u sY|d kre|jd
136 kr_q<|�|� q<|d kskJ �|�|t|�f� t|d � q<|S |jd
137 kr�g S |gS ) Nro rj )�]�}�)�
138 testlist_compc S s g | ] }|j d kr|�qS )� star_expr)rf )rH r| r! r! r" r� { s
139 �z9SequenceLiteralValue.get_tree_entries.<locals>.<listcomp>Zdictorsetmakerz**�,r� rl )r} rw rf r� �iter�next�append)r r� Z
140 array_node�kv�iteratorr� �opr! r! r" r� o s6
141
142
143
144
145
146 z%SequenceLiteralValue.get_tree_entriesc C � d| j j| jf S re )r_ r. r} r+ r! r! r" rg � � z SequenceLiteralValue.__repr__r )
147 r. r/ r0 r� r� r\ r� r� r r� r� rg ri r! r! r^ r" r� < s �
148 -r� c @ sb e Zd ZdZdd� Zdd� Zddd�Zed �d
149 d � �Zed �d
150 d� �Z dd� Z
151 dd� Z dd� Z dS )�DictLiteralValuer� c C s t �| |� || _|| _d S r )r� r\ r� r} r� r! r! r" r\ � s
152 zDictLiteralValue.__init__c C sp t �| j|�}| �� D ]&\}}| j�|�D ]}|�|d�D ]}|�� r/| j�|� S q qq td| ��)r� z==z No key found in dictionary %s.) r �create_simple_objectr* r� r� r� Zexecute_operationr1 r
153 )r r� Zcompiled_value_indexr� r| r� Zkey_vr! r! r" r� � s ��� z%DictLiteralValue.py__simple_getitem__Nc c s@ � t }| �� D ] \}}|| j�|�O }q|D ]}t|�V qdS )r� N)r r� r� r� r )r r r� r� �_r! r! r" r � s � �zDictLiteralValue.py__iter__r� c C r� r r� r� r! r! r" r� � r� z DictLiteralValue._imitate_valuesr� c r� )Nc s4 g | ]\}}t t� jt� j|�t� j|�f���qS r! )r r� r* r r� )rH rq rr r+ r! r" r� � s �
154
155 ���z3DictLiteralValue._imitate_items.<locals>.<listcomp>)r� r r� r* r� r! r+ r" r� � s
156 �z DictLiteralValue._imitate_itemsc c sH � | � � D ] \}}| j�|�D ]}t|�r |�� t| j|�fV qqdS )z�
157 Returns a generator of tuples like dict.items(), where the key is
158 resolved (as a string) and the values are still lazy values.
159 N)r� r� r� r r1 r )r rq r| r� r! r! r" r� � s ����z DictLiteralValue.exact_key_itemsc � t �� fdd�� �� D ��S )Nc 3 s � | ] \}}� j �|�V qd S r �r� r� �rH r� �vr+ r! r" rJ � �
160 �
161 �
162 �z0DictLiteralValue._dict_values.<locals>.<genexpr>�r rM r� r+ r! r+ r" r� � � �z DictLiteralValue._dict_valuesc r� )Nc 3 s � | ] \}}� j �|�V qd S r r� r� r+ r! r" rJ � r� z.DictLiteralValue._dict_keys.<locals>.<genexpr>r� r+ r! r+ r" r� � r� zDictLiteralValue._dict_keysr )
163 r. r/ r0 rX r\ r� r r r� r� r� r� r� r! r! r! r" r� � s
164
165
166
167
168 r� c s> e Zd Z� fdd�Zdd� Zd dd�Zdd � Zd
169 d � Z� ZS )
170 �
171 _FakeSequencec s t � �|� || _dS )z7
172 type should be one of "tuple", "list"
173 N)r[ r\ �_lazy_value_list)r r* Zlazy_value_listr^ r! r" r\ � s
174 z_FakeSequence.__init__c C sV t |t�r
175 t| g�S ttt�� | j| }W d � |�� S 1 s"w Y |�� S r )r� r� r r r� r� r� rF �r r� rI r! r! r" r� � s
176
177
178 ��z"_FakeSequence.py__simple_getitem__Nc C s | j S r )r� r r! r! r" r � s z_FakeSequence.py__iter__c C s t t| j��S r )�boolr� r� r+ r! r! r" r: � r� z_FakeSequence.py__bool__c C rd re )rf r. r� r+ r! r! r" rg r- z_FakeSequence.__repr__r ) r. r/ r0 r\ r� r r: rg ri r! r! r^ r" r� � s
179 r� c @ r� )r� r� Nr� r! r! r! r" r� r� r� c @ r� )r� r� Nr� r! r! r! r" r� r� r� c sb e Zd ZdZ� fdd�Zddd�Zdd� Zed �d
180 d � �Zd d
181 � Z dd� Z
182 dd� Z dd� Z � Z
183 S )�FakeDictr� c rZ r )r[ r\ �_dct)r r* r� r^ r! r" r\ r` zFakeDict.__init__Nc c s& � | j D ] }tt�| j|��V qd S r )r� r r r� r* )r r r� r! r! r" r s �
184 �zFakeDict.py__iter__c C sB t tt�� | j| }W d � |�� S 1 sw Y |�� S r )r r� r� r� rF r� r! r! r" r� s
185 ��z FakeDict.py__simple_getitem__r� c C s t t| jt| �� �g�g�S r )r r� r* r r� r= r! r! r" �_values s �zFakeDict._valuesc C s t �dd� | j�� D ��S )Nc s rD r rE rG r! r! r" rJ $ rK z(FakeDict._dict_values.<locals>.<genexpr>)r rM r� r� r+ r! r! r" r� # s zFakeDict._dict_valuesc C rC )Nc s rD r rE rG r! r! r" rJ ' rK z&FakeDict._dict_keys.<locals>.<genexpr>rL r+ r! r! r" r� & r� zFakeDict._dict_keysc C rc r )r� r� r+ r! r! r" r� ) r$ zFakeDict.exact_key_itemsc C r� )Nz<%s: %s>)r_ r. r� r+ r! r! r" rg , r� zFakeDict.__repr__r )r. r/ r0 rX r\ r r� r r� r� r� r� rg ri r! r! r^ r" r� s
186
187 r� c s. e Zd Z� fdd�Zddd�Zdd� Z� ZS ) � MergedArrayc s" t � �|� |d j| _|| _d S )N�����)r[ r\ rX �_arrays)r r* Zarraysr^ r! r" r\ 1 s
188 zMergedArray.__init__Nc c s � | j D ] }|�� E d H qd S r )r� r )r r �arrayr! r! r" r 6 s �
189 �zMergedArray.py__iter__c C rC )Nc s rD r rE rG r! r! r" rJ ; rK z3MergedArray.py__simple_getitem__.<locals>.<genexpr>rL )r r� r! r! r" r� : r� z MergedArray.py__simple_getitem__r )r. r/ r0 r\ r r� ri r! r! r^ r" r� 0 s
190 r� c
191 C s( |j dkr
192 |j|iS |j dkr |jd dv r t| ||jd �S |j dv ri }t|jddd� �}d}|�t| |��D ],}|d7 }zt|�}W n tyZ t j
193 | d |d
194 | d � Y q:w |� t| |� � |�� q:t|d�}|r}|dur}t j
195 | d |d
196 | d � |S |j dks�|j dkr�i S |j dkr�i S t
197 �)zG
198 Unpacking tuple assignments in for statements and expr_stmts.
199 rP r} r )rs rt rj )r� r� r� r� Nro zvalue-error-too-many-valuesz3ValueError: too many values to unpack (expected %s))�messagezvalue-error-too-few-valuesz.ValueError: need more than %s values to unpack�powerZ atom_exprr� )rf r| rw r� r� ra r r� �
200 StopIterationr �add�updaterF �NotImplementedError) r� r� r� r� �parts�nrI �partZ has_partsr! r! r" r� > s:
201
202
203
204
205 ��
206
207 �
208 r� c @ s( e Zd Zdd� Zdd� Zefdd�ZdS )r� c C s$ |j | _ || _|| _|| _|| _d S r )r* �_context�_start�_stop�_step)r Zpython_context�start�stop�stepr! r! r" r\ e s
209
210 zSlice.__init__c C s t �| jjd�}|�� \}|S )Nr� )r r) r r* Zexecute_with_values)r r| � slice_valuer! r! r" r6 m s
211 zSlice._get_wrapped_valuec sL � fdd�}zt |� j�|� j�|� j��W S ty% t ddd� Y S w )ze
212 Imitate CompiledValue.obj behavior and return a ``builtin.slice()``
213 object.
214 c s6 | d u rd S � j �| �}t|�dkrt�|\}t|�S )Nrj )r r� r� r� r )�element�resultr| r+ r! r" �getw s z!Slice.get_safe_value.<locals>.getN)r� r
215 r r r� )r r� r r! r+ r" r1 r s
216 �zSlice.get_safe_valueN)r. r/ r0 r\ r6 r r1 r! r! r! r" r� d s r� N)9rh Zjedi.inferencer r Zjedi.inference.lazy_valuer r r Zjedi.inference.helpersr r r r
217 Zjedi.inference.utilsr r Zjedi.inference.cacher
218 Zjedi.inference.filtersr r Zjedi.inference.base_valuer r r r r r r Zjedi.parser_utilsr Zjedi.inference.contextr Z#jedi.inference.value.dynamic_arraysr r r2 r8 r� r� r� r� r� r{ ry rz r� rx r� r� r� r� r� r� r� r� r� r! r! r! r" �<module> sD $ %$0%
219
220 8dJ$&