Format: application/x-python-code
1 o
2
3 �h�< � @ s6 d dl mZ d dlmZmZmZmZmZmZ d dl m
4 Z
5 d dl m Z d dl m
6 Z
7 d dlmZ d dlmZmZ d dlmZ d d lmZ G d
8 d � d e�ZG d d
9 � d
10 �ZG dd� de�ZG dd� de�ZG dd� dee
11 �ZG dd� d�Z G dd� de�Z G dd� de�Z G dd� de�Z G d d � d e�Z G d d � d e�Z!d S )!� )� inference_state_method_cache)�ValueSet� NO_VALUES�Value�iterator_to_value_set�LazyValueWrapper� ValueWrapper)�builtin_from_name�� ClassFilter)�
12 ClassMixin)�to_list)�AbstractNameDefinition� ValueName)� ClassContext)�TupleGenericManagerc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
13 S ) �_BoundTypeVarNamez>
14 This type var was bound to a certain type, e.g. int.
15 c C s || _ |j| _|| _d S �N)� _type_var�parent_context�
16 _value_set)�self�type_var� value_set� r �Qc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/inference/gradual/base.py�__init__ s
17 z_BoundTypeVarName.__init__c s � fdd�}t |� �S )Nc 3 s> � � j D ]} ddlm} t| |�r� jjE d H q| V qd S )Nr )�AnyClass)r Z jedi.inference.gradual.typingr �
18 isinstancer � constraints)�valuer �r r r �iter_ s �
19
20 �z&_BoundTypeVarName.infer.<locals>.iter_)r )r r" r r! r �infer s
21 z_BoundTypeVarName.inferc C �
22 | j �� S r )r �
23 py__name__r! r r r r% ! �
24 z _BoundTypeVarName.py__name__c C s d| j j| �� | jf S )Nz
25 <%s %s -> %s>)� __class__�__name__r% r r! r r r �__repr__$ s z_BoundTypeVarName.__repr__N)r( �
26 __module__� __qualname__�__doc__r r# r% r) r r r r r
27 s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �_TypeVarFilterz�
28 A filter for all given variables in a class.
29
30 A = TypeVar('A')
31 B = TypeVar('B')
32 class Foo(Mapping[A, B]):
33 ...
34
35 In this example we would have two type vars given: A and B
36 c C s || _ || _d S r )� _generics�
37 _type_vars)r �genericsZ type_varsr r r r 3 s
38 z_TypeVarFilter.__init__c C s\ t | j�D ]&\}}|�� |kr+z t|| j| �gW S ty* |jg Y S w qg S r )� enumerater/ r% r r. �
39 IndexError�name)r r3 �ir r r r �get7 s ��z_TypeVarFilter.getc C � g S r r r! r r r �values@ s z_TypeVarFilter.valuesN)r( r* r+ r, r r5 r7 r r r r r- ( s
40
41 r- c s e Zd Z� fdd�Z� ZS )�_AnnotatedClassContextc / s. � t � j|i |��}|E d H | j�� V d S r )�super� get_filters�_value�get_type_var_filter)r �args�kwargs�filters�r' r r r: G s ���
42 z"_AnnotatedClassContext.get_filters)r( r* r+ r: �
43 __classcell__r r r@ r r8 F s r8 c @ sJ e Zd Zdd� Zdd� Ze� dd� �Zdd� Zd d
44 � Zd d � Z d
45 d� Z
46 dS )�DefineGenericBaseClassc C s
47 || _ d S r )�_generics_manager�r �generics_managerr r r r S r& z DefineGenericBaseClass.__init__c C s t �r )�NotImplementedErrorrD r r r � _create_instance_with_genericsV � z5DefineGenericBaseClass._create_instance_with_genericsc C r$ r )rC Zto_tupler! r r r � get_genericsY s
48 z#DefineGenericBaseClass.get_genericsc C s� ddl m} d}g }| �� D ]/}t}|D ]#}t|t|f�r0|�|�}||O }|t|h�kr/d}q|t|g�O }q|�|� q|sEt| g�S t| � t
49 t |���g�S )Nr ��TypeVarFT) � jedi.inference.gradual.type_varrK rI r r rB �define_genericsr �appendrG r �tuple) r �
50 type_var_dictrK �changedZ new_genericsZ generic_setr7 Zgeneric�resultr r r rM ] s&
51 �
52
53 �z&DefineGenericBaseClass.define_genericsc C sZ t |t�sdS | j|jkrdS | �� }|�� }t|�t|�kr!dS tdd� t||�D ��S )NFc 3 s. � | ]\}� t � fd d�|�� D ��V qdS )c 3 s( � | ]}� � � D ]}|�|�V qqd S r )�gather_annotation_classes�
54 is_same_class)�.0Zcls1Zcls2�Z
55 class_set2r r � <genexpr>� s � � � �zADefineGenericBaseClass.is_same_class.<locals>.<genexpr>.<genexpr>N)�anyrS )rU Z
56 class_set1r rV r rW � s � ��
57 �z7DefineGenericBaseClass.is_same_class.<locals>.<genexpr>)r rB � tree_noderI �len�all�zip)r �otherZ
58 given_params1Z
59 given_params2r r r rT w s
60 �z$DefineGenericBaseClass.is_same_classc C r6 r r r! r r r �get_signatures� rH z%DefineGenericBaseClass.get_signaturesc C s d| j j| jt| �� �f S )Nz
61 <%s: %s%s>)r' r( �_wrapped_value�listrI r! r r r r) � s
62
63 �z DefineGenericBaseClass.__repr__N) r( r* r+ r rG r rI rM rT r^ r) r r r r rB R s
64 rB c s~ e Zd ZdZ� fdd�Zdd� Zddd�Zd d
65 � Z� fd d �Zd
66 d� Z e
67 dd� �Z dd� Z � fdd�Z
68 dd� Zdd� Z� ZS )� GenericClassz�
69 A class that is defined with generics, might be something simple like:
70
71 class Foo(Generic[T]): ...
72 my_foo_int_cls = Foo[int]
73 c s t � �|� || _d S r )r9 r � _class_value)r � class_valuerE r@ r r r � s
74 zGenericClass.__init__c C � | j S r �rb r! r r r �_get_wrapped_value� � z GenericClass._get_wrapped_valueTc C s> | � � }tddddd��||�}|| j�� }|r d| S |S )N�List�Dict�Set�Tuple)r` �dict�setrO zType[%s])r% rl r5 rC �
75 get_type_hint)r �add_class_info�n�sr r r rn � s zGenericClass.get_type_hintc C s t | �� | �� �S r )r- rI Zlist_type_varsr! r r r r< � � z GenericClass.get_type_var_filterc s t � �|�\}tt|�g�S r )r9 �
76 py__call__r �_GenericInstanceWrapper)r � arguments�instancer@ r r rs � s zGenericClass.py__call__c C s t | �S r )r8 r! r r r � _as_context� s zGenericClass._as_contextc c s& � | j �� D ]
77 }t| || j�V qd S r )r_ � py__bases__�_LazyGenericBaseClassrC �r �baser r r rx � s ��zGenericClass.py__bases__c C s t | j|�S r )ra rb rD r r r rG � � z+GenericClass._create_instance_with_genericsc s t � �|�rdS | j�|�S )NT)r9 �is_sub_class_ofrb )r rc r@ r r r} � s z GenericClass.is_sub_class_ofc C s | j �|�S r )rb �
78 with_generics)r Zgenerics_tupler r r r~ � r| zGenericClass.with_genericsc
79 C s� ddl m}m} | �� }i }|dkr#| �� }|r!|d �|�� �S |S |D ]/}|�� r4|�� s4|� � }nq%|j
80 dkr;q%|� � D ]}|�� } || krS|||| |�� nq?q%|S )Nr )�merge_pairwise_generics�merge_type_var_dicts�Iterable�class) Z!jedi.inference.gradual.annotationr r� r% rI �infer_type_varsZmerge_types_of_iterateZ is_instanceZ is_compiled�get_annotated_class_object�api_type� py__mro__)
81 r r r r� Zannotation_namerP Zannotation_genericsZpy_classZ parent_class�
82 class_namer r r r� � s6 � �
83
84 ���z GenericClass.infer_type_vars�T)r( r* r+ r, r rf rn r< rs rw r
85 rx rG r} r~ r� rA r r r@ r ra � s
86
87 ra c @ s0 e Zd Zdd� Zedd� �Zdd� Zdd� Zd S )
88 ry c C s || _ || _|| _d S r )rb �_lazy_base_classrC )r rc Zlazy_base_classrE r r r r � s
89 z _LazyGenericBaseClass.__init__c
90 c sh � | j �� D ]+}t|t�r t�|j|jtt| � |����V q|�
91 � r.t�|j|| j �V q|V qd S r ) r� r# r ra Z
92 create_cached�inference_stater_ r rO �_remap_type_varsZis_class_mixinrC rz r r r r# s �
93 ���z_LazyGenericBaseClass.inferc c s~ � ddl m} | j�� }|�� D ],}t}|D ]"}t||�r1|�|�� �}|t �
94 dd� |D ��O }q|t |g�O }q|V qd S )Nr rJ c s s � | ]}|� � V qd S r )r# )rU r3 r r r rW ! s �
95 �z9_LazyGenericBaseClass._remap_type_vars.<locals>.<genexpr>) rL rK rb r< rI r r r5 r% r Z from_sets)r r{ rK �filterZ type_var_set�newr �namesr r r r� s �
96
97 ��z&_LazyGenericBaseClass._remap_type_varsc C � d| j j| jf S �Nz<%s: %s>)r' r( r� r! r r r r) + rr z _LazyGenericBaseClass.__repr__N)r( r* r+ r r r# r� r) r r r r ry � s
98 ry c @ s e Zd Zdd� Zddd�ZdS )rt c C sx | j j�� D ]0}|�� dkr%|�� }z |d �� W S ty$ Y qw |�� dkr6tt| j d�g� S q| j �
99 � S )N� Generator� �Iterator�None) r_ rc r� r% rI �execute_annotationr2 r r r� �py__stop_iteration_returns)r �clsr0 r r r r� 0 s � �
100 z2_GenericInstanceWrapper.py__stop_iteration_returnsTc C s | j jjdd�S )NF)ro )r_ rc rn )r ro r r r rn < s z%_GenericInstanceWrapper.get_type_hintNr� )r( r* r+ r� rn r r r r rt / s rt c s\ e Zd ZdZdZ� fdd�Zedd� �Zdd� Zd d
101 � Z ed d � �Z
102 d
103 d� Z dd� Z � Z
104 S )�_PseudoTreeNameClassa2
105 In typeshed, some classes are defined like this:
106
107 Tuple: _SpecialForm = ...
108
109 Now this is not a real class, therefore we have to do some workarounds like
110 this class. Essentially this class makes it possible to goto that `Tuple`
111 name, without affecting anything else negatively.
112 r� c s t � �|j|� || _d S r )r9 r r� �
113 _tree_name�r r � tree_namer@ r r r L s
114 �
115 z _PseudoTreeNameClass.__init__c C rd r )r� r! r r r rY S s z _PseudoTreeNameClass.tree_nodec o s � G dd� dt �}|� V d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z5_PseudoTreeNameClass.get_filters.<locals>.EmptyFilterc S s d S r r r! r r r r Z rH z>_PseudoTreeNameClass.get_filters.<locals>.EmptyFilter.__init__c [ r6 r r )r r3 r> r r r r5 ] rH z9_PseudoTreeNameClass.get_filters.<locals>.EmptyFilter.getc [ r6 r r )r r> r r r r7 ` rH z<_PseudoTreeNameClass.get_filters.<locals>.EmptyFilter.valuesN)r( r* r+ r r5 r7 r r r r � EmptyFilterY s r� r
116 )r r= r> r� r r r r: W s �
117 z _PseudoTreeNameClass.get_filtersc C s t | jd�S )N�type)r r� r! r r r � py__class__e s z _PseudoTreeNameClass.py__class__c C � t | | j�S r �r r� r! r r r r3 j � z_PseudoTreeNameClass.namec C s
118 | j jfS r )r� r r! r r r �get_qualified_namesn r& z(_PseudoTreeNameClass.get_qualified_namesc C � d| j j| jjf S �Nz%s(%s)�r' r( r� r r! r r r r) q � z _PseudoTreeNameClass.__repr__)r( r* r+ r, r� r �propertyrY r: r� r3 r� r) rA r r r@ r r� @ s
119
120 r� c @ s8 e Zd Zdd� Zedd� �Zdd� Zdd� Zd d
121 � Zd S ) �BaseTypingValuec C s |j | _ || _|| _d S r )r� r r� r� r r r r v s
122 zBaseTypingValue.__init__c C r� r r� r! r r r r3 { r� zBaseTypingValue.namec C � t | j| j�S r �r� r r� r! r r r rf � z"BaseTypingValue._get_wrapped_valuec C r$ r )r_ r^ r! r r r r^ � r& z BaseTypingValue.get_signaturesc C r� r� r� r! r r r r) � r� zBaseTypingValue.__repr__N) r( r* r+ r r� r3 rf r^ r) r r r r r� u s
123 r� c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�BaseTypingClassWithGenericsc s$ t � �|� |j| _|| _|| _d S r )r9 r r� r r� )r r r� rE r@ r r r � s
124 z$BaseTypingClassWithGenerics.__init__c C r� r r� r! r r r rf � r� z.BaseTypingClassWithGenerics._get_wrapped_valuec C s d| j j| jj| jf S )Nz%s(%s%s))r' r( r� r rC r! r r r r) � s �z$BaseTypingClassWithGenerics.__repr__)r( r* r+ r rf r) rA r r r@ r r� � s r� c @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zed d
125 � �Zd d � Z d
126 d� Z
127 dS )�BaseTypingInstancec C s$ |j | _ || _|| _|| _|| _d S r )r� r rb r� rC )r r rc r� rE r r r r � s
128
129 zBaseTypingInstance.__init__c C rd r re r! r r r r� � rg z BaseTypingInstance.py__class__c C rd r re r! r r r r� � rg z-BaseTypingInstance.get_annotated_class_objectc C s
130 | � � fS r )r% r! r r r r� � r& z&BaseTypingInstance.get_qualified_namesc C r� r r� r! r r r r3 � r� zBaseTypingInstance.namec C s t | jd��� \}|S )N�object)r r� r� )r Zobject_r r r rf � s z%BaseTypingInstance._get_wrapped_valuec C r� r� )r' r( rC r! r r r r) � rr zBaseTypingInstance.__repr__N) r( r* r+ r r� r� r� r� r3 rf r) r r r r r� � s
131 r� N)"Zjedi.inference.cacher Zjedi.inference.base_valuer r r r r r Zjedi.inference.compiledr Zjedi.inference.value.klassr r Zjedi.inference.utilsr
132 Zjedi.inference.namesr r Zjedi.inference.contextr Z jedi.inference.gradual.genericsr r r- r8 rB ra ry rt r� r� r� r� r r r r �<module> s( N\35