Format: application/x-python-code
1 o
2
3 �h C � @ sZ d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
4 m Z d dl m
5 Z
6 mZmZ d dlmZmZ d d lmZmZ d d
7 lmZ d d lmZ d d lmZ G d
8 d� d�ZG dd� de�Z G dd� d�Z G dd� de e �Z G dd� de e �Z G dd� de e �Z G dd� de e �Z!G dd � d e e�Z"G d d � d e �Z#G d d � d e#�Z$d!d"� Z%d#d$� Z&d%S )&� )�abstractmethod)�contextmanager)�Path)�Optional)�search_ancestor)�Name)�ParserTreeFilter� MergedFilter�GlobalNameFilter)�AnonymousParamName�TreeNameDefinition)� NO_VALUES�ValueSet)�get_parent_scope)�debug)� parser_utilsc @ s� e Zd Zdd� Zed)dd��Zdd� Z d*d d
9 �Zd d � Zd
10 d� Z dd� Z
11 dd� Z dd� Z dd� Z
12 dd� Zdd� Zdd � Zed d � �Zd d � Zed!d"� �Zd#d$� Zd%d&� Zed'd(� �ZdS )+�AbstractContextc C s || _ i | _d S �N)�inference_state�predefined_names)�selfr � r �Lc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/inference/context.py�__init__ s
13 zAbstractContext.__init__Nc C � t �r ��NotImplementedError�r �until_position� origin_scoper r r � get_filters � zAbstractContext.get_filtersc C sF ddl m} t| t|t�r|nd |�}|�||�}t�d|| |� |S )Nr )�finderzcontext.goto %s in (%s): %s)�jedi.inferencer" � _get_global_filters_for_name�
14 isinstancer Z filter_namer �dbg)r � name_or_str�positionr" �filters�namesr r r �goto s � zAbstractContext.gotoTc C sZ |du r| }| � ||�}t|t�r|jn|}d}| jrTt|t�rT|}|durTt�|�sT|j}|jdv rKz | j| } | | }
15 W n t yG Y q"w |
16 }n |durTt�|�r+|durt|rtddl
17 m } | j | | j
18 |d�} | | ju rqt}
19 n
20 |}
21 n
22 t�dd� |D ��}
23 |s�|
24 s�|r�t|t�r�ddl
25 m} d | }|�|d
26 ||� t�d ||
27 � |
28 r�|
29 S | �|||�S ) zZ
30 :param position: Position of the last statement -> tuple of line, column
31 N)Zif_stmtZfor_stmt�comp_for�
32 sync_comp_forr )�
33 flow_analysis)�contextZ value_scope�nodec s s � | ]}|� � V qd S r )Zinfer)�.0�namer r r � <genexpr>M s � z5AbstractContext.py__getattribute__.<locals>.<genexpr>)�analysisz$NameError: name '%s' is not defined.z
34 name-errorz context.names_to_types: %s -> %s)r+ r% r �valuer r �is_scope�parent�type�KeyErrorr# r. Zreachability_check� tree_nodeZ UNREACHABLEr
35 r Z from_setsr4 �addr r& � _check_for_additional_knowledge)r r' � name_contextr( Zanalysis_errorsr* Z string_nameZfound_predefined_typesr0 Z name_dict�typesr. �check�valuesr4 �messager r r �py__getattribute__& sN
36
37 �� �
38
39 z"AbstractContext.py__getattribute__c C s� |p| }t |t�r@|�� s@|}|jg}tdd� |D ��r tS ddlm} t|dd�}|||||�}|d ur8|S ||v r? tS q%tS )Nc s s � | ]}|j d v V qdS )�r, r- N)r8 )r1 �br r r r3 a s � zBAbstractContext._check_for_additional_knowledge.<locals>.<genexpr>r )�check_flow_informationT)Z
40 include_flows) r% r � is_instancer: �anyr
41 Zjedi.inference.finderrE r )r r' r= r( �
42 flow_scopeZ
43 base_nodesrE �nr r r r< Z s& ��z/AbstractContext._check_for_additional_knowledgec C s | j }|d u r | S |�� S r )�parent_context�get_root_context)r rJ r r r rK n s z AbstractContext.get_root_contextc C � dS �NFr �r r r r � is_modulet � zAbstractContext.is_modulec C rL rM r rN r r r �is_builtins_modulew rP z"AbstractContext.is_builtins_modulec C rL rM r rN r r r �is_classz rP zAbstractContext.is_classc C rL rM r rN r r r �is_stub} rP zAbstractContext.is_stubc C rL rM r rN r r r rF � rP zAbstractContext.is_instancec C rL rM r rN r r r � is_compiled� rP zAbstractContext.is_compiledc C rL rM r rN r r r �is_bound_method� rP z AbstractContext.is_bound_methodc C r r r rN r r r �
44 py__name__� r! zAbstractContext.py__name__c C r r r rN r r r � get_value� rP zAbstractContext.get_valuec C � d S r r rN r r r r2 � r! zAbstractContext.namec C rL )Nr r rN r r r �get_qualified_names� rP z#AbstractContext.get_qualified_namesc C rL )N� r rN r r r � py__doc__� rP zAbstractContext.py__doc__c c s, � | j }|||< z d V W ||= d S ||= w r )r )r rH �dctZ
45 predefinedr r r �predefine_names� s �z AbstractContext.predefine_names�NN)NNT)�__name__�
46 __module__� __qualname__r r r r+ rB r< rK rO rQ rR rS rF rT rU rV rW �propertyr2 rY r[ r r] r r r r r s4
47 �4
48
49 r c s� e Zd ZdZ� fdd�Zedd� �Zedd� �Zdd � Zd
50 d � Z d d
51 � Z
52 dd� Z dd� Z dd� Z
53 dd� Zdd� Zedd� �Zdd� Zd d � Zd d � Zd d!� Z� ZS )"� ValueContextzC
54 Should be defined, otherwise the API returns empty types.
55 c s t � �|j� || _d S r )�superr r �_value)r r5 �� __class__r r r � s
56 zValueContext.__init__c C � | j jS r )re r: rN r r r r: � � zValueContext.tree_nodec C rh r )re rJ rN r r r rJ � ri zValueContext.parent_contextc C �
57 | j �� S r )re rO rN r r r rO � �
58 zValueContext.is_modulec C s | j | jjkS r )re r �builtins_modulerN r r r rQ � s z ValueContext.is_builtins_modulec C rj r )re rR rN r r r rR � rk zValueContext.is_classc C rj r )re rS rN r r r rS � rk zValueContext.is_stubc C rj r )re rF rN r r r rF � rk zValueContext.is_instancec C rj r )re rT rN r r r rT � rk zValueContext.is_compiledc C rj r )re rU rN r r r rU � rk z ValueContext.is_bound_methodc C rj r )re rV rN r r r rV � rk zValueContext.py__name__c C rh r )re r2 rN r r r r2 � ri zValueContext.namec C rj r )re rY rN r r r rY � rk z ValueContext.get_qualified_namesc C rj r )re r[ rN r r r r[ � rk zValueContext.py__doc__c C � | j S r �re rN r r r rW � � zValueContext.get_valuec C � d| j j| jf S �Nz%s(%s))rg r_ re rN r r r �__repr__� � zValueContext.__repr__)r_ r` ra �__doc__r rb r: rJ rO rQ rR rS rF rT rU rV r2 rY r[ rW rr �
59 __classcell__r r rf r rc � s*
60
61
62 rc c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
63 �TreeContextMixinc C s ddl m} || |�S )Nr )�
64 infer_node)Zjedi.inference.syntax_treerw )r r0 rw r r r rw � s
65 zTreeContextMixin.infer_nodec C s� ddl m} || jkr| �� sJ �| �� S | �|�}|jdv rF|j�||�}|� � rD|j
66 � |j�}|� | j
67 |j
68 |�}|j||�� |d�}|S |jdkrS|�| j
69 ||�S td| ��)Nr )r5 )�funcdef�lambdef)�instanceZ
70 class_context�function�classdefz Probably shouldn't happen: %s)r# r5 r: rO rW �create_contextr8 Z
71 FunctionValueZ from_contextrR rJ � create_valueZAnonymousInstancer Z BoundMethod�
72 as_contextZ
73 ClassValuer )r r0 r5 rJ �funcZ class_valuerz r r r r~ � s*
74
75
76
77 ��
78 z TreeContextMixin.create_valuec sz d
79 � ���fdd� � dd� ����}|j dv r7|j|j�d� }�j|jk r7�j}|j dkr3|j�ks7�|�}� |dd �S ) NTc sj | �j kr�S | jdv r��| ��� S | jdv r/� �| j��}�j| jd jkr*|S t|| �S td| ��)N)rx ry r| rC �����z(There's a scope that was not managed: %s) r: r8 r~ r r7 � start_pos�children�CompForContext� Exception)�
80 scope_node� is_nestedrJ ��from_scope_noder0 � parent_scoper r r r� � s
81
82
83
84 z8TreeContextMixin.create_context.<locals>.from_scope_nodec S sn | j } t�| �r | S | jdv r | jd jdv r | jd S n| jdkr6| jdd� D ] }|jdv r5| S q*q)NT)ZargumentZ
85 testlist_comp� rC Zdictorsetmaker� )r7 r r6 r8 r� )r0 rI r r r r�
86 s
87
88
89 �
90
91 ��z5TreeContextMixin.create_context.<locals>.parent_scope)rx r| �:�param)r� )T)r8 r� �indexr� r7 r2 )r r0 r� �colonr7 r r� r r} � s
92
93 z TreeContextMixin.create_contextc C sT |� � }|r |jdkr |j|kr t|dd�}| �|�}t||�S | �|�}t||�S )Nr� rx ry )Zget_definitionr8 r2 r r~ r r} r )r Z tree_nameZ
94 definitionrx r� r/ r r r � create_name" s
95
96
97
98 z TreeContextMixin.create_nameN)r_ r` ra rw r~ r} r� r r r r rv � s
99 &rv c @ � e Zd Zddd�ZdS )�FunctionContextNc c s � t | j| ||d�V d S �N)rJ r r )r r r r r r r . s � �zFunctionContext.get_filtersr^ �r_ r` ra r r r r r r� - � r� c @ sP e Zd Zdee fdd�Zddd�Zdd� Zed d
100 � �Z ed d � �Z
101 d
102 d� Z dS )�
103 ModuleContext�returnc C rj r �re �
104 py__file__rN r r r r� 8 rk zModuleContext.py__file__Nc c s@ � | j �|�}t|d � tt| ||d�| �� �V |E d H d S r� )re r �nextr r �get_global_filter)r r r r) r r r r ; s �
105 ��zModuleContext.get_filtersc C s t | �S r )r
106 rN r r r r� I s z ModuleContext.get_global_filterc C rh r �re � string_namesrN r r r r� L ri zModuleContext.string_namesc C rh r )re �
107 code_linesrN r r r r� P ri zModuleContext.code_linesc C rm )z�
108 This is the only function that converts a context back to a value.
109 This is necessary for stub -> python conversion and vice versa. However
110 this method shouldn't be moved to AbstractContext.
111 rn rN r r r rW T s zModuleContext.get_valuer^ ) r_ r` ra r r r� r r� rb r� r� rW r r r r r� 7 s
112
113
114 r� c @ s< e Zd Zd dd�Zdd� Zedd� �Zdee fd d
115 �Z dS ) �NamespaceContextNc C rj r �re r r r r r r ^ rk z NamespaceContext.get_filtersc C rm r rn rN r r r rW a ro zNamespaceContext.get_valuec C rh r r� rN r r r r� d ri z NamespaceContext.string_namesr� c C rj r r� rN r r r r� h rk zNamespaceContext.py__file__r^ )
116 r_ r` ra r rW rb r� r r r� r r r r r� ] s
117
118 r� c @ s e Zd Zddd�Zddd�ZdS )� ClassContextNc c s � | � ||�V d S r )r� r r r r r m s �zClassContext.get_filtersc C s t | ||d�S r� �r r r r r r� p s
119 �z ClassContext.get_global_filterr^ )r_ r` ra r r� r r r r r� l s
120 r� c s> e Zd Z� fdd�Zd dd�Zdd� Zdd � Zd
121 d � Z� ZS )
122 r� c s t � �|j� || _|| _d S r )rd r r r: rJ )r rJ r, rf r r r y s
123 zCompForContext.__init__Nc c s � t | �V d S r r� r r r r r ~ s �zCompForContext.get_filtersc C rX r r rN r r r rW � rP zCompForContext.get_valuec C rL )Nz<comprehension context>r rN r r r rV � rP zCompForContext.py__name__c C rp rq )rg r_ r: rN r r r rr � rs zCompForContext.__repr__r^ ) r_ r` ra r r rW rV rr ru r r rf r r� x s
124 r� c @ r� )�CompiledContextNc C rj r r� r r r r r � rk zCompiledContext.get_filtersr^ r� r r r r r� � r� r� c @ s6 e Zd ZdZdd� Zedd� �Zdee fdd�Z dS ) �CompiledModuleContextNc C rm r rn rN r r r rW � ro z CompiledModuleContext.get_valuec C rh r r� rN r r r r� � ri z"CompiledModuleContext.string_namesr� c C rj r r� rN r r r r� � rk z CompiledModuleContext.py__file__)
125 r_ r` ra r� rW rb r� r r r� r r r r r� � s
126 r� c C s� |d ur:t |ddd�}d }|dkr|}t |dd�}|d ur:|jd }|d ur:||jk r:|d u s7||jd jk r:|j}t| ||�S )Nrx r| ry �����)r r� r� �get_global_filters)r/ Z name_or_noner( Zancestorry r� r r r r$ � s
127 r$ c c sv � | }ddl m} | dur'| j||d�E dH t| |tf�r d}| j} | dus
128 t|jj�� d�}|dus6J �|V dS )a�
129 Returns all filters in order of priority for name resolution.
130
131 For global name lookups. The filters will handle name resolution
132 themselves, but here we gather possible filters downwards.
133
134 >>> from jedi import Script
135 >>> script = Script('''
136 ... x = ['a', 'b', 'c']
137 ... def func():
138 ... y = None
139 ... ''')
140 >>> module_node = script._module_node
141 >>> scope = next(module_node.iter_funcdefs())
142 >>> scope
143 <Function: func@3-5>
144 >>> context = script._get_module_context().create_context(scope)
145 >>> filters = list(get_global_filters(context, (4, 0), None))
146
147 First we get the names from the function scope.
148
149 >>> print(filters[0]) # doctest: +ELLIPSIS
150 MergedFilter(<ParserTreeFilter: ...>, <GlobalNameFilter: ...>)
151 >>> sorted(str(n) for n in filters[0].values()) # doctest: +NORMALIZE_WHITESPACE
152 ['<TreeNameDefinition: string_name=func start_pos=(3, 4)>',
153 '<TreeNameDefinition: string_name=x start_pos=(2, 0)>']
154 >>> filters[0]._filters[0]._until_position
155 (4, 0)
156 >>> filters[0]._filters[1]._until_position
157
158 Then it yields the names from one level "lower". In this example, this is
159 the module scope (including globals).
160 As a side note, you can see, that the position in the filter is None on the
161 globals filter, because there the whole module is searched.
162
163 >>> list(filters[1].values()) # package modules -> Also empty.
164 []
165 >>> sorted(name.string_name for name in filters[2].values()) # Module attributes
166 ['__doc__', '__name__', '__package__']
167
168 Finally, it yields the builtin filter, if `include_builtin` is
169 true (default).
170
171 >>> list(filters[3].values()) # doctest: +ELLIPSIS
172 [...]
173 r )� BaseFunctionExecutionContextN)r r ) Z jedi.inference.value.functionr� r r% r� rJ r� r rl )r/ r r Z base_contextr� rD r r r r� � s �/ ��
174 r� N)'�abcr �
175 contextlibr �pathlibr �typingr Z
176 parso.treer Zparso.python.treer Zjedi.inference.filtersr r r
177 Zjedi.inference.namesr r Zjedi.inference.base_valuer
178 r Zjedi.parser_utilsr Zjedir r r rc rv r� r� r� r� r� r� r� r$ r� r r r r �<module> s2 9P
179 &