Format: application/x-python-code
1 o
2
3 �hD � @ s� d dl mZ d dlmZ d dlmZmZ d dlmZ d dlm Z d dlm
4 Z
5 d dlm Z d dl m
6 Z
7 d d lmZmZmZ d d
8 lmZmZmZmZmZ d d lmZmZmZm Z m Z d d l m Z m Z m!Z! d d
9 l"m#Z#m$Z$ d dl%m&Z& d dlm'Z' d dl(m)Z) d dl*m+Z+ d dl,m-Z- G dd� de�Z.G dd� de �Z/G dd� d�Z0G dd� de0e/ed�Z1G d d � d e�Z2G d d � d e1�Z3G d d!� d!e#e$�Z4G d"d#� d#e4�Z5G d$d%� d%e4�Z6G d&d'� d'e0e �Z7d(d)� Z8d*S )+� )�tree)�debug)� inference_state_method_cache�CachedMetaClass)�compiled)� recursion)�
10 docstrings)�
11 flow_analysis��
12 TreeSignature)�ParserTreeFilter�FunctionExecutionFilter� AnonymousFunctionExecutionFilter)� ValueName�AbstractNameDefinition�AnonymousParamName� ParamName� NameWrapper)�ContextualizedNode� NO_VALUES�ValueSet� TreeValue� ValueWrapper)�LazyKnownValues�LazyKnownValue�
13 LazyTreeValue)� ValueContext�TreeContextMixin)�iterable)� parser_utils)�get_yield_exprs)�values_from_qualified_names��TupleGenericManagerc @ s0 e Zd ZdZdZdd� Zedd� �Zdd� Zd S )
14 �
15 LambdaNamez<lambda>�functionc C s || _ |j| _d S �N)�
16 _lambda_value�parent_context)�selfZ lambda_value� r* �Sc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/inference/value/function.py�__init__ s zLambdaName.__init__c C s
17 | j jjS r& )r' � tree_node� start_pos�r) r* r* r+ r. " s
18 zLambdaName.start_posc C s t | jg�S r& )r r' r/ r* r* r+ �infer& s zLambdaName.inferN) �__name__�
19 __module__� __qualname__� string_name�api_typer, �propertyr. r0 r* r* r* r+ r$ s
20 r$ c @ s e Zd Zdd� ZdS )�FunctionAndClassBasec C sF | j �� r| j �� }|d u rd S || �� f S | j �� r!| �� fS d S r& )r( �is_class�get_qualified_names�
21 py__name__Z is_module)r) �nr* r* r+ r9 + s
22
23
24
25 z(FunctionAndClassBase.get_qualified_namesN)r1 r2 r3 r9 r* r* r* r+ r7 * s r7 c @ sj e Zd ZdZddd�Zdd� Zdd� Zed d
26 � �Zd d � Z d
27 d� Z
28 ddd�Z dd� Z ddd�Z
29 dd� ZdS )�
30 FunctionMixinr% Nc c s. � | � � }|�� D ] }|j|d�E d H q d S )N)� origin_scope)� py__class__Zexecute_with_values� get_filters)r) r= �cls�instancer* r* r+ r? ; s
31 � �zFunctionMixin.get_filtersc C s4 ddl m} |d u rt| g�S t|||�� | �g�S )Nr )� BoundMethod)Z jedi.inference.value.instancerB r �
32 as_context)r) rA Z class_valuerB r* r* r+ � py__get__@ s
33 zFunctionMixin.py__get__c s � fdd�� j �� D �S )Nc s g | ]}t � |j��qS r* )r �name��.0�paramr/ r* r+ �
34 <listcomp>H s �z1FunctionMixin.get_param_names.<locals>.<listcomp>)r- �
35 get_paramsr/ r* r/ r+ �get_param_namesG s
36 �z FunctionMixin.get_param_namesc C s" | j jdkr
37 t| �S t| | j j�S )N�lambdef)r- �typer$ r rE r/ r* r* r+ rE K s zFunctionMixin.namec C s dS )NTr* r/ r* r* r+ � is_functionQ � zFunctionMixin.is_functionc C s | j jS r& )rE r4 r/ r* r* r+ r: T s zFunctionMixin.py__name__Tc s� | j j}|d u r3dd� � | �� }|�� }|�� }| �� dd�� fdd�|�� D �� }|d u r2|S n|jdd�}| �� | j j d jdd� }|d
38 | S ) Nc S sJ | j }| �� �� }|d ur|d| 7 }| jd ur#|d| jjdd� 7 }|S )Nz: �=F�Zinclude_prefix)r4 r0 �
39 get_type_hint� default_node�get_code)r; �s�
40 annotationr* r* r+ �param_name_to_strZ s
41 z6FunctionMixin.get_type_hint.<locals>.param_name_to_strz(%s)�, c � g | ]}� |��qS r* r* )rG r; �rW r* r+ rI f s ��z/FunctionMixin.get_type_hint.<locals>.<listcomp>FrQ � z -> )
42 r- rV rC r0 rR r: �joinrK rT �children)r) �add_class_info�return_annotation�function_execution�resultZ return_hint�bodyr* rZ r+ rR W s �� zFunctionMixin.get_type_hintc C s | � |�}|�� S r& )rC r0 )r) � argumentsr` r* r* r+ �
43 py__call__r s
44 zFunctionMixin.py__call__c C s |d u rt | �S t| |�S r& )�AnonymousFunctionExecution�FunctionExecutionContext)r) rc r* r* r+ � _as_contextv s
45 zFunctionMixin._as_contextc C s dd� | � � D �S )Nc S s g | ]}t |��qS r* r
46 �rG �fr* r* r+ rI | � z0FunctionMixin.get_signatures.<locals>.<listcomp>)�get_signature_functionsr/ r* r* r+ �get_signatures{ � z FunctionMixin.get_signaturesr& �T)r1 r2 r3 r5 r? rD rK r6 rE rN r: rR rd rg rl r* r* r* r+ r< 8 s
47
48
49
50 r< c @ s0 e Zd Zedd� �Zdd� Zdd� Zdd� Zd S )
51 �
52 FunctionValuec sx � ��fdd��t t�|��}����� s��� r$�j���� s��� s�|�}|r:t|t t�fdd�|D ����S |S )Nc s* �� � r
53 t�j��| d�S � �j�| d�S )N)r( r- )r8 � MethodValue�inference_state)r- )r@ �contextr( r* r+ �create� s ��z*FunctionValue.from_context.<locals>.createc rY r* r* rh )rs r* r+ rI � rj z.FunctionValue.from_context.<locals>.<listcomp>)�list�_find_overload_functionsr8 Z is_instancer( �OverloadedFunctionValue�reversed)r@ rr r- Zoverloaded_funcsr% r* )r@ rr rs r( r+ � from_context� s ��zFunctionValue.from_contextc C s t | jdd�\}|S )N�types� FunctionType)r! rq )r) �cr* r* r+ r> � s zFunctionValue.py__class__c C � | j S r& )r( r/ r* r* r+ �get_default_param_context� � z'FunctionValue.get_default_param_contextc C s | gS r& r* r/ r* r* r+ rk � r~ z%FunctionValue.get_signature_functionsN)r1 r2 r3 � classmethodrx r> r} rk r* r* r* r+ ro s
54 ro )� metaclassc s$ e Zd Z� fdd�Zdd� Z� ZS )�FunctionNameInClassc s t � �|� || _d S r& )�superr, �_class_context)r) �
55 class_contextrE �� __class__r* r+ r, � �
56 z FunctionNameInClass.__init__c C �
57 | j �� S r& )r� � get_valuer/ r* r* r+ � get_defining_qualified_value� �
58 z0FunctionNameInClass.get_defining_qualified_value)r1 r2 r3 r, r� �
59 __classcell__r* r* r� r+ r� � s r� c s< e Zd Z� fdd�Zdd� Zdd� Ze� fdd��Z� ZS ) rp c s$ t � j|g|�R i |�� || _d S r& )r� r, r� )r) rq r� �args�kwargsr� r* r+ r, � s
60 zMethodValue.__init__c C r| r& )r� r/ r* r* r+ r} � r~ z%MethodValue.get_default_param_contextc C s$ | j �� }|d u r d S || �� f S r& )r� r9 r: )r) �namesr* r* r+ r9 � s
61 z MethodValue.get_qualified_namesc s t | jt� j�S r& )r� r� r� rE r/ r� r* r+ rE � s zMethodValue.name) r1 r2 r3 r, r} r9 r6 rE r� r* r* r� r+ rp � s rp c @ sl e Zd Zdd� Zeed�e�� ddd���Zdd� Z eje
62 g �d�dd d
63 ��Z dd d �Z d
64 d� Z
65 dd� ZdS )� BaseFunctionExecutionContextc C s t �r& )�NotImplementedErrorr/ r* r* r+ �infer_annotations� rO z.BaseFunctionExecutionContext.infer_annotations)�defaultFc C s | j }|jdkr| �|jd �S |rt}t| j|�}n| �� }|r#|S |t� | j
66 �O }|� � }|D ][}|rE|t �
67 dd� | �|�D ��O }q1t�| ||�}|tju rXt�d|� n&z|j}W n tyt t�| jd�}|t |g�O }Y n
68 w || �|d �O }|tju r�t�d|� |S q1|S ) NrL �����c s � � | ]}|� � V qd S r& �r0 �rG �
69 lazy_valuer* r* r+ � <genexpr>� �
70 � �
71 �zABaseFunctionExecutionContext.get_return_values.<locals>.<genexpr>zReturn unreachable: %s�None� zReturn reachable: %s)r- rM Z
72 infer_noder] r r rq r� r �infer_return_types�_valueZiter_return_stmtsr � from_sets�_get_yield_lazy_valuer Zreachability_checkZ UNREACHABLEr �dbg�AttributeErrorr �builtin_from_nameZ REACHABLE) r) � check_yields�funcdefZ value_set�returns�r�checkr] �ctxr* r* r+ �get_return_values� s@
73 �
74
75 �
76 �z.BaseFunctionExecutionContext.get_return_valuesc c st � |j dkrt�| jd�}t|�V d S |jd }|j dkr2t| |jd �}|�� �|�E d H d S t | |�V d S )N�keywordr� r� Z yield_arg)
77 rM r r� rq r r] r r0 �iterater )r) Z
78 yield_exprr� �node�cnr* r* r+ r� � s �
79
80
81
82 z2BaseFunctionExecutionContext._get_yield_lazy_valuec
83 c s� � dd� t | j| j�D �}g }d }|D ]Z\}}|j}|jdkr"|j}|jdkrH|| jkrHt�|�rH||kr?|d d �|� n,|�||gf� n#|| jkrV|�d |gf� n| jdd�}|rht |d t
84 d
85 �d �V d S |}q|D ]^\}} |d u r�| D ]
86 }| � |�E d H qzqp|� � }
87 t
88 | |
89 �} | �� �| �} t| �} | D ]1}
90 t|jd j�|
91 �� i}| �||�� | D ]
92 }| � |�E d H q�W d � n1 s�w Y q�qpd S ) Nc
93 S s" g | ]
94 }|t �|d ddd�f�qS )�for_stmtr� Z
95 while_stmtZif_stmt)r Zsearch_ancestor)rG �yr* r* r+ rI s
96 �
97 �zFBaseFunctionExecutionContext.get_yield_lazy_values.<locals>.<listcomp>Zsuiter� r� r� T)r� r �inf)�min�max)r rq r- �parentrM r Zfor_stmt_defines_one_name�appendr� r �floatr� Z get_testlistr r0 r� rt �strr] �valueZpredefine_names)r) �is_asyncZ for_parentsZ yields_orderZ
98 last_for_stmtZyield_r� r� ry ZyieldsZ
99 input_noder� Zorderedr� �dctZyield_in_same_for_stmtr* r* r+ �get_yield_lazy_values sT � �
100 �
101 �
102 � ����z2BaseFunctionExecutionContext.get_yield_lazy_valuesc C s t �dd� | �� D ��S )Nc s r� r& r� r� r* r* r+ r� 6 r� zBBaseFunctionExecutionContext.merge_yield_values.<locals>.<genexpr>)r r� r� )r) r� r* r* r+ �merge_yield_values5 s
103 �z/BaseFunctionExecutionContext.merge_yield_valuesc C s t t| j| j��S r& )�boolr rq r- r/ r* r* r+ � is_generator; rm z)BaseFunctionExecutionContext.is_generatorc s� | j }| jjjdv }ddlm� |rU| �� r6|j�d�}| j dd�}|�
104 � t f�t � �fdd�|D ���
105 � S |j�d �}| �� }|�
106 � t t f�t � �fd
107 d�|D ���
108 � S | �� rf| �� sft t�|| �g�S | �� S ) z4
109 Created to be used by inheritance.
110 )Z
111 async_stmtZ
112 async_funcdefr )� GenericClass�AsyncGeneratorT)r� c 3 � � | ]
113 }� |t ���V qd S r& r" �rG r{ �r� Zgenericsr* r+ r� N s
114 � �
115 �z5BaseFunctionExecutionContext.infer.<locals>.<genexpr>� Coroutinec 3 r� r& r" r� r� r* r+ r� W s �
116 �)rq r- r� rM Zjedi.inference.gradual.baser� r� Z
117 typing_moduleZpy__getattribute__r� r> r r Zexecute_annotationr� r� r � Generator)r) rq Z is_coroutineZasync_generator_classesZ yield_valuesZ
118 async_classesZ
119 return_valuesr* r� r+ r0 > s4 � �� ��z"BaseFunctionExecutionContext.inferN)F)r1 r2 r3 r� r r r Z execution_recursion_decoratorr� r� �iterr� r� r� r0 r* r* r* r+ r� � s (
120 . r� c s6 e Zd Z� fdd�Zd
121 dd�Zdd� Zdd � Z� ZS ) rf c � t � �|� || _d S r& )r� r, �
122 _arguments)r) Zfunction_valuerc r� r* r+ r, c r� z!FunctionExecutionContext.__init__Nc c s � t | | j||| jd�V d S )N)�until_positionr= rc )r
123 r� r� �r) r� r= r* r* r+ r? g s � �z$FunctionExecutionContext.get_filtersc C s ddl m} || j| j�S )Nr )r� )Z!jedi.inference.gradual.annotationr� r� r� )r) r� r* r* r+ r� o s z*FunctionExecutionContext.infer_annotationsc s � fdd�� j j�� D �S )Nc s g | ] }t � j|j� j��qS r* )r r� rE r� rF r/ r* r+ rI t s ��z<FunctionExecutionContext.get_param_names.<locals>.<listcomp>)r� r- rJ r/ r* r/ r+ rK s s
124
125 �z(FunctionExecutionContext.get_param_names�NN)r1 r2 r3 r, r? r� rK r� r* r* r� r+ rf b s
126
127 rf c @ s& e Zd Zdd� Zddd�Zdd� ZdS ) re c C s t S r& )r r/ r* r* r+ r� { s z,AnonymousFunctionExecution.infer_annotationsNc c s � t | | j||d�V d S )N)r� r= )r r� r� r* r* r+ r? � s � �z&AnonymousFunctionExecution.get_filtersc C r� r& )r� rK r/ r* r* r+ rK � r� z*AnonymousFunctionExecution.get_param_namesr� )r1 r2 r3 r� r? rK r* r* r* r+ re z s
128 re c s6 e Zd Z� fdd�Zdd� Zdd� Zd
129 dd �Z� ZS ) rv c r� r& )r� r, �_overloaded_functions)r) r% Zoverloaded_functionsr� r* r+ r, � r� z OverloadedFunctionValue.__init__c C sp t jd| jdd� g }| �� D ]}|j�|�}|�|� |�|�r'|�� S q| j j
130 r.t S t �
131 dd� |D ��S )Nz Execute overloaded function %sZBLUE)�colorc s r� r& r� )rG Zfer* r* r+ r� � � � z5OverloadedFunctionValue.py__call__.<locals>.<genexpr>)r r� Z_wrapped_valuerl r� rC r� Zmatches_signaturer0 rq Z is_analysisr r r� )r) rc Zfunction_executions� signaturer` r* r* r+ rd � s
132
133 �z"OverloadedFunctionValue.py__call__c C r| r& )r� r/ r* r* r+ rk � r~ z/OverloadedFunctionValue.get_signature_functionsTc C s dd� dd� | jD �� S )Nz Union[%s]rX c s r� r& )rR rh r* r* r+ r� � r� z8OverloadedFunctionValue.get_type_hint.<locals>.<genexpr>)r\ r� )r) r^ r* r* r+ rR � s z%OverloadedFunctionValue.get_type_hintrn )r1 r2 r3 r, rd rk rR r� r* r* r� r+ rv � s
134 rv c c s� � dd� }|j dkr d S ||�r|V t| |jd�}|�|jj�}t|t�s)J �|s-d S d}|D ]}|jj }|j dkrG||�rG|}d}|V q1|sLd S q)Nc S sb | j jdkr/| j jd }|jdkr|g}n|j}|D ]}|jd }|jdkr.|jdkr. dS qdS ) NZ decoratedr � decoratorr� rE �overloadTF)r� rM r] r� )r� Z
135 decoratorsr� Z dotted_namer* r* r+ �_is_overload_decorated� s
136
137 �z8_find_overload_functions.<locals>._is_overload_decoratedrL T)r� Fr� )
138 rM r r. �getrE r� �
139 isinstancert Z tree_namer� )rr r- r� �filterr� �foundrE r� r* r* r+ ru � s4 �
140 ���ru N)9Z parso.pythonr Zjedir Zjedi.inference.cacher r Zjedi.inferencer r r r Zjedi.inference.signaturer Zjedi.inference.filtersr r
141 r Zjedi.inference.namesr r r r r Zjedi.inference.base_valuer r r r r Zjedi.inference.lazy_valuer r r Zjedi.inference.contextr r Zjedi.inference.valuer r Zjedi.inference.parser_cacher Zjedi.inference.helpersr! Z jedi.inference.gradual.genericsr# r$ r7 r< ro r� rp r� rf re rv ru r* r* r* r+ �<module> s< G-