Format: application/x-python-code
1 o
2
3 �hgC � @ s d Z ddlZddlmZ ddlmZmZ ddlmZm Z m
4 Z
5 m Z m Z ddl
6 mZ ddlmZ ddlmZ dd lmZmZ dd
7 lmZ dd lmZmZmZ dd l m Z dd
8 l m Z m Z d�!� Z"ddddddddd�Z#d�!� Z$G dd� de�Z%G dd � 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'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/d0� d0e�Z0G d1d2� d2e�Z1G d3d4� d4e�Z2G d5d6� d6e �Z3G d7d8� d8e
9 �Z4G d9d:� d:e �Z5G d;d<� d<e�Z6G d=d>� d>e �Z7dS )?z�
10 We need to somehow work with the typing objects. Since the typing objects are
11 pretty bare we need to add all the Jedi customizations to make them work as
12 values.
13
14 This file deals with all the typing.py cases.
15 � N)�debug)�builtin_from_name�create_simple_object)�ValueSet� NO_VALUES�Value�LazyValueWrapper� ValueWrapper)�LazyKnownValues)�repack_with_argument_clinic)�
16 FilterWrapper)� NameWrapper� ValueName)�
17 ClassMixin)�BaseTypingValue�BaseTypingClassWithGenerics�BaseTypingInstance)� TypeVarClass)�LazyGenericManager�TupleGenericManagerz$Tuple Generic Protocol Callable Typez
18 builtins.listz
19 builtins.dictz builtins.setzbuiltins.frozensetzcollections.ChainMapzcollections.Counterzcollections.defaultdictzcollections.deque)�List�Dict�Set� FrozenSet�ChainMap�Counter� DefaultDict�Dequez!Optional Union ClassVar Annotatedc @ � e Zd Zdd� Zdd� ZdS )�TypingModuleNamec C s t | �� �S �N)r �_remap��self� r$ �Sc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/inference/gradual/typing.py�infer' s zTypingModuleName.inferc c s� � | j }| jj}zt| }W n ty Y nw t�|| j| j|�V d S |tv r5t �|| j| j�V d S |t
20 v rEt �|| j| j�V d S |dkrKd S |dkr^| j �
21 � \}t�||�V d S |dkrnt�|| j| j�V d S |dkrzt|d�V d S |dkr�t�|| j| j�V d S |dkr�| j �
22 � \}t�||�V d S |dkr�| j �
23 � \}t�||�V d S |d kr�t�|| j| j�V d S | j �
24 � E d H d S )
25 NZruntime�TypeVar�Any�
26 TYPE_CHECKING�True�overload�NewType�cast� TypedDict)� string_name�parent_context�inference_state�_TYPE_ALIAS_TYPES�KeyError� TypeAlias�
27 create_cachedZ tree_name�_PROXY_CLASS_TYPES�ProxyTypingClassValue� _PROXY_TYPES�ProxyTypingValueZ
28 _wrapped_namer& r �AnyClassr �OverloadFunction�NewTypeFunction� CastFunction�TypedDictClass)r# �namer1 �actual�cls�vZcast_fnr$ r$ r% r! * s\ � � �
29
30 �
31
32 �
33
34 �
35
36 �
37
38 �zTypingModuleName._remapN)�__name__�
39 __module__� __qualname__r& r! r$ r$ r$ r% r & s r c @ � e Zd ZeZdS )�TypingModuleFilterWrapperN)rC rD rE r Zname_wrapper_classr$ r$ r$ r% rG ` � rG c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
40 �ProxyWithGenericsc C s� | j j}|dkr| �� �� S |dkr!| �� �� tt| jd�g�B S |dkr*| jd S |dv r5| jd �� S tt t
41 t d�}|| }t|| j | | j | jd�g�S ) N�Union�Optional�None�Typer )�ClassVar� Annotated)�Tuple�Generic�Protocol�Callable��generics_manager)
42 �
43 _tree_name�value�gather_annotation_classes�execute_annotationr r r1 �_generics_managerrP rQ rR rS r0 )r# r/ ZmappedrA r$ r$ r% rY e s.
44 �
45 �
46 �z$ProxyWithGenerics.execute_annotationc C s t �| j�� �S r )r � from_setsrZ �to_tupler" r$ r$ r% rX � � z+ProxyWithGenerics.gather_annotation_classesc C � t | j| j|�S r )rI r0 rV �r# rU r$ r$ r% � _create_instance_with_generics� �
47 �z0ProxyWithGenerics._create_instance_with_genericsc sL | � � }|si S | �� }|dkr$t| jd�� |d �|�� fdd���S i S )NrK rL r c s | � kS r r$ ��x��noner$ r% �<lambda>� s z3ProxyWithGenerics.infer_type_vars.<locals>.<lambda>)� get_generics�
48 py__name__r r1 �infer_type_vars�filter)r# � value_set�annotation_generics�annotation_namer$ rd r% ri � s �z!ProxyWithGenerics.infer_type_varsN)rC rD rE rY rX r` ri r$ r$ r$ r% rI d s
49 ! rI c @ s e Zd ZeZdd� Zdd� ZdS )r9 c C s | j j| j| j| jt|�d�S )NrT )� index_classr5 r1 r0 rV r )r# Zgenerics_tupler$ r$ r% �
50 with_generics� s �z ProxyTypingValue.with_genericsc s t � �fdd�|D ��S )Nc 3 s4 � | ]}�j j�j�j�jt� j|d �d�V qdS ))Zcontext_of_index� index_valuerT N)rn r5 r1 r0 rV r �context)�.0rp ��contextualized_noder# r$ r% � <genexpr>� s � ���
51 �z1ProxyTypingValue.py__getitem__.<locals>.<genexpr>�r �r# Zindex_value_setrt r$ rs r% �
52 py__getitem__� s �z ProxyTypingValue.py__getitem__N)rC rD rE rI rn ro rx r$ r$ r$ r% r9 � s r9 c @ s( e Zd Zdd� Zdd� Zedd� �ZdS )�_TypingClassMixinc C s t | jj�d��gS )N�object)r
53 r1 �builtins_module�py__getattribute__r" r$ r$ r% � py__bases__� s �z _TypingClassMixin.py__bases__c C � g S r r$ r" r$ r$ r% �get_metaclasses� � z!_TypingClassMixin.get_metaclassesc C � t | | j�S r )r rV r" r$ r$ r% r? � � z_TypingClassMixin.nameN)rC rD rE r} r �propertyr? r$ r$ r$ r% ry � s
54 ry c @ r )�TypingClassWithGenericsc C s� i }| � � }|s
55 |S | �� }|dkr|d �|�� �S |dkr0t|�dkr.|d �|�� �S |S |dkr>| �� \}|�|�S |S )NrM r rS � � rP )rg rh ri rY �len)r# rk �
56 type_var_dictrl rm Ztuple_annotationr$ r$ r% ri � s& � ��
57
58 z'TypingClassWithGenerics.infer_type_varsc C r^ r )r� r0 rV r_ r$ r$ r% r` � ra z6TypingClassWithGenerics._create_instance_with_genericsN)rC rD rE ri r` r$ r$ r$ r% r� � s r� c @ rF )r7 N)rC rD rE r� rn r$ r$ r$ r% r7 � rH r7 c @ sH e Zd Zdd� Zedd� �Zdd� Zdd� Zd d
59 � Zd d � Z d
60 d� Z
61 dS )r4 c C s |j | _ || _|| _|| _d S r )r1 r0 �_origin_tree_name�_actual)r# r0 Zorigin_tree_namer@ r$ r$ r% �__init__� s
62 zTypeAlias.__init__c C r� r )r r� r" r$ r$ r% r? � r� zTypeAlias.namec C � | j jS r )r? r/ r" r$ r$ r% rh � s zTypeAlias.py__name__c C s d| j j| jf S )Nz<%s: %s>)� __class__rC r� r" r$ r$ r% �__repr__ s zTypeAlias.__repr__c C sd | j �d�\}}ddlm} || j|g| jj��� \}|�|�}t|�dks*J |��t t
63 |��}|S )N�.r )�Importerr� ) r� �splitZjedi.inference.importsr� r1 r{ �followr| r� �next�iter)r# � module_name�
64 class_namer� �module�classesrA r$ r$ r% �_get_wrapped_value s ��
65 z TypeAlias._get_wrapped_valuec C s t | �� g�S r )r r� r" r$ r$ r% rX s z#TypeAlias.gather_annotation_classesc C r~ r r$ r" r$ r$ r% �get_signatures r� zTypeAlias.get_signaturesN) rC rD rE r� r� r? rh r� r� rX r� r$ r$ r$ r% r4 � s
66 r4 c @ r )rS c C sR z | j d }| j d }W n ty t�d� t Y S w ddlm} ||||�S )zG
67 def x() -> Callable[[Callable[..., _T]], _T]: ...
68 r r� z+Callable[...] defined without two arguments)�infer_return_for_callable)rZ �
69 IndexErrorr �warningr �!jedi.inference.gradual.annotationr� )r# � argumentsZ param_valuesZ
70 result_valuesr� r$ r$ r% �
71 py__call__ s
72
73 � zCallable.py__call__c C s
74 t | g�S r rv )r# �instanceZ class_valuer$ r$ r% � py__get__) �
75 zCallable.py__get__N)rC rD rE r� r� r$ r$ r$ r% rS s rS c @ sJ e Zd Zdd� Zdd� Zddd�Zdd � Zd
76 d � Zed d
77 � �Z dd� Z
78 dS )rP c C �
79 | j �� S r )rZ Zis_homogenous_tupler" r$ r$ r% �_is_homogenous. s
80 zTuple._is_homogenousc C s< | � � r
81 | j�d�S t|t�r| j�|�S t�d| � tS )Nr z The getitem type on Tuple was %s)r� rZ �get_index_and_execute�
82 isinstance�intr Zdbgr �r# �indexr$ r$ r% �py__simple_getitem__3 s
83 zTuple.py__simple_getitem__Nc c sB � | � � rt| j�d��V d S | j�� D ] }t|�� �V qd S �Nr )r� r
84 rZ r� r\ rY )r# rt rB r$ r$ r% �
85 py__iter__= s ��zTuple.py__iter__c C s( | � � r
86 | j�d�S t�| j�� ��� S r� )r� rZ r� r r[ r\ rY rw r$ r$ r% rx D s ��zTuple.py__getitem__c C s | j j�d��� \}|S �N�tuple)r1 r{ r| rY )r# �tuple_r$ r$ r% r� L s
87 �zTuple._get_wrapped_valuec C r� r )Z_wrapped_valuer? r" r$ r$ r% r? Q � z
88 Tuple.namec C s� ddl m}m} |�dd� �}| �� r | j�� d �|�� �S i }|D ] }z|j }W n t
89 y3 Y q#w |� }|||| j|�� q#|S )Nr )�merge_pairwise_generics�merge_type_var_dictsc S s | � � �� dkS r� )rh �lowerrb r$ r$ r% rf Z s z'Tuple.infer_type_vars.<locals>.<lambda>) r� r� r� rj r� Z _class_valuerg ri Zmerge_types_of_iterateZget_annotated_class_object�AttributeError)r# rk r� r� r� �element�methodZpy_classr$ r$ r% ri U s* ��
90
91 �
92 �zTuple.infer_type_varsr ) rC rD rE r� r� r� rx r� r� r? ri r$ r$ r$ r% rP - s
93
94
95 rP c @ � e Zd ZdS )rQ N�rC rD rE r$ r$ r$ r% rQ ~ � rQ c @ r� )rR Nr� r$ r$ r$ r% rR � r� rR c @ � e Zd Zdd� ZdS )r: c C s t �d� tS )Nz Used Any - returned no results)r r� r r" r$ r$ r% rY � s
96 zAnyClass.execute_annotationN)rC rD rE rY r$ r$ r$ r% r: � � r: c @ � e Zd Zed�dd� �ZdS )r; zfunc, /c C s |S r r$ )r# Zfunc_value_setr$ r$ r% r� � s zOverloadFunction.py__call__N�rC rD rE r r� r$ r$ r$ r% r; � � r; c @ r� )r< c sH |� � }t|d� t|d�\}� � d u rtS t� �fdd�|�� D ��S )N)NNc 3 s( � | ]}t �j|j|j� �� �V qd S r )r, r1 rq �noder& )rr rt �Z
97 second_argr# r$ r% ru � s � ��
98 �z-NewTypeFunction.py__call__.<locals>.<genexpr>)�unpackr� r r Zget_calling_nodes)r# r� Z ordered_args�_r$ r� r% r� � s
99 �zNewTypeFunction.py__call__N)rC rD rE r� r$ r$ r$ r% r< � r� r< c sF e Zd Z� fdd�Zdd� Zdd� Zedd� �Zd efd
100 d �Z � Z
101 S ) r, c s t � �||� || _|| _d S r )�superr� �_type_value_set� tree_node)r# r1 r0 r� �type_value_set�r� r$ r% r� � s
102 zNewType.__init__c C s | j �� \}|S r )r� � py__class__)r# �cr$ r$ r% r� � s zNewType.py__class__c C r� r )r� rY )r# r� r$ r$ r% r� � r� zNewType.py__call__c C s ddl m} || d�S )Nr )�CompiledValueNamer, )Z jedi.inference.compiled.valuer� )r# r� r$ r$ r% r? � s
103 z NewType.name�returnc C s d| j | jf S )Nz<NewType: %s>%s)r� r� r" r$ r$ r% r� � r] zNewType.__repr__) rC rD rE r� r� r� r� r? �strr� �
104 __classcell__r$ r$ r� r% r, � s
105 r, c @ r� )r= ztype, object, /c C s |� � S r )rY )r# r� Zobject_value_setr$ r$ r% r� � r� zCastFunction.py__call__Nr� r$ r$ r$ r% r= � r� r= c @ s e Zd ZdZdS )r> zr
106 This class has no responsibilities and is just here to make sure that typed
107 dicts can be identified.
108 N)rC rD rE �__doc__r$ r$ r$ r% r> � rH r> c @ s< e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
109 d � Z d S )
110 r. z6Represents the instance version of ``TypedDictClass``.c C s" |j | _ |j| _|j| _|| _d S r )r1 r0 r� �_definition_class)r# Zdefinition_classr$ r$ r% r� � s
111 zTypedDict.__init__c C s t | | jj�S r )r r� r? r" r$ r$ r% r? � s zTypedDict.namec s0 t � t�rt�� fdd�| jjdd�D ��S tS )Nc 3 s( � | ]}|� � �D ]}|�� V q qd S r )�getr& )rr rj r? �r� r$ r% ru � s � �� �z1TypedDict.py__simple_getitem__.<locals>.<genexpr>T�Z is_instance)r� r� r r[ r� � get_filtersr r� r$ r� r% r� � s
112
113 �z TypedDict.py__simple_getitem__c s6 t j�dd� � jjdd�D ��}t� fdd�|D ��S )Nc s s � | ]}|� � V qd S r )�values)rr �fr$ r$ r% ru � s
114 � �
115 �z+TypedDict.get_key_values.<locals>.<genexpr>Tr� c s h | ] }t � j|j��qS r$ )r r1 r/ )rr rB r" r$ r% � <setcomp>� s ��z+TypedDict.get_key_values.<locals>.<setcomp>)� itertools�chain�
116 from_iterabler� r� r )r# Zfiltered_valuesr$ r" r% �get_key_values� s � �zTypedDict.get_key_valuesc C s | j j�d�\}|�� \}|S )N�dict)r1 r{ r| Zexecute_with_values)r# �d�resultr$ r$ r% r� � s
117 z TypedDict._get_wrapped_valueN)
118 rC rD rE r� r� r� r? r� r� r� r$ r$ r$ r% r. � s
119
120 r. )8r� r� Zjedir Zjedi.inference.compiledr r Zjedi.inference.base_valuer r r r r Zjedi.inference.lazy_valuer
121 Zjedi.inference.argumentsr Zjedi.inference.filtersr Zjedi.inference.namesr
122 r Zjedi.inference.value.klassr Zjedi.inference.gradual.baser r r Z jedi.inference.gradual.type_varr Z jedi.inference.gradual.genericsr r r� r6 r2 r8 r rG rI r9 ry r� r7 r4 rS rP rQ rR r: r; r< r, r= r> r. r$ r$ r$ r% �<module> sV �
123 :?%&Q