o h<@s6ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl m Z ddlmZddlmZmZddlmZdd lmZGd d d eZGd d d ZGdddeZGdddeZGdddee ZGdddZGdddeZGdddeZGdddeZGdddeZ GdddeZ!d S)!)inference_state_method_cache)ValueSet NO_VALUESValueiterator_to_value_setLazyValueWrapper ValueWrapper)builtin_from_name ClassFilter) ClassMixin)to_list)AbstractNameDefinition ValueName) ClassContext)TupleGenericManagerc@s0eZdZdZddZddZddZdd Zd S) _BoundTypeVarNamez> This type var was bound to a certain type, e.g. int. cCs||_|j|_||_dSN) _type_varparent_context _value_set)selftype_var value_setrQc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/inference/gradual/base.py__init__s z_BoundTypeVarName.__init__csfdd}t|S)Nc3s>jD]}ddlm}t||rjjEdHq|VqdS)Nr)AnyClass)rZjedi.inference.gradual.typingr isinstancer constraints)valuerrrriter_s   z&_BoundTypeVarName.infer..iter_)r)rr"rr!rinfers  z_BoundTypeVarName.infercC |jSr)r py__name__r!rrrr%! z_BoundTypeVarName.py__name__cCsd|jj||jfS)Nz <%s %s -> %s>) __class____name__r%rr!rrr__repr__$sz_BoundTypeVarName.__repr__N)r( __module__ __qualname____doc__rr#r%r)rrrrr s  rc@s(eZdZdZddZddZddZdS) _TypeVarFilterz A filter for all given variables in a class. A = TypeVar('A') B = TypeVar('B') class Foo(Mapping[A, B]): ... In this example we would have two type vars given: A and B cCs||_||_dSr) _generics _type_vars)rgenericsZ type_varsrrrr3s z_TypeVarFilter.__init__c Cs\t|jD]&\}}||kr+z t||j|gWSty*|jgYSwqgSr) enumerater/r%rr. IndexErrorname)rr3irrrrget7s  z_TypeVarFilter.getcCgSrrr!rrrvalues@sz_TypeVarFilter.valuesN)r(r*r+r,rr5r7rrrrr-(s   r-cseZdZfddZZS)_AnnotatedClassContextc/s.tj|i|}|EdH|jVdSr)super get_filters_valueget_type_var_filter)rargskwargsfiltersr'rrr:Gs z"_AnnotatedClassContext.get_filters)r(r*r+r: __classcell__rrr@rr8Fsr8c@sJeZdZddZddZeddZddZd d Zd d Z d dZ dS)DefineGenericBaseClasscCs ||_dSr)_generics_managerrgenerics_managerrrrrSr&zDefineGenericBaseClass.__init__cCstr)NotImplementedErrorrDrrr_create_instance_with_genericsVz5DefineGenericBaseClass._create_instance_with_genericscCr$r)rCZto_tupler!rrr get_genericsYs z#DefineGenericBaseClass.get_genericsc Csddlm}d}g}|D]/}t}|D]#}t|t|fr0||}||O}|t|hkr/d}q|t|gO}q||q|sEt|gSt| t t |gS)NrTypeVarFT) jedi.inference.gradual.type_varrKrIrrrBdefine_genericsrappendrGrtuple) r type_var_dictrKchangedZ new_genericsZ generic_setr7ZgenericresultrrrrM]s&      z&DefineGenericBaseClass.define_genericscCsZt|tsdS|j|jkrdS|}|}t|t|kr!dStddt||DS)NFc3s.|]\}tfdd|DVqdS)c3s(|]}D]}||VqqdSr)gather_annotation_classes is_same_class).0Zcls1Zcls2Z class_set2rr s  zADefineGenericBaseClass.is_same_class...N)anyrS)rUZ class_set1rrVrrWs  z7DefineGenericBaseClass.is_same_class..)rrB tree_noderIlenallzip)rotherZ given_params1Z given_params2rrrrTws   z$DefineGenericBaseClass.is_same_classcCr6rrr!rrrget_signaturesrHz%DefineGenericBaseClass.get_signaturescCsd|jj|jt|fS)Nz <%s: %s%s>)r'r(_wrapped_valuelistrIr!rrrr)s  zDefineGenericBaseClass.__repr__N) r(r*r+rrGrrIrMrTr^r)rrrrrBRs  rBcs~eZdZdZfddZddZdddZd d Zfd d Zd dZ e ddZ ddZ fddZ ddZddZZS) GenericClassz A class that is defined with generics, might be something simple like: class Foo(Generic[T]): ... my_foo_int_cls = Foo[int] cst|||_dSr)r9r _class_value)r class_valuerEr@rrrs  zGenericClass.__init__cC|jSrrbr!rrr_get_wrapped_valuezGenericClass._get_wrapped_valueTcCs>|}tddddd||}||j}|rd|S|S)NListDictSetTuple)r`dictsetrOzType[%s])r%rlr5rC get_type_hint)radd_class_infonsrrrrns zGenericClass.get_type_hintcCst||Sr)r-rIZlist_type_varsr!rrrr<z GenericClass.get_type_var_filtercst|\}tt|gSr)r9 py__call__r_GenericInstanceWrapper)r argumentsinstancer@rrrsszGenericClass.py__call__cCst|Sr)r8r!rrr _as_contextszGenericClass._as_contextccs&|jD] }t|||jVqdSr)r_ py__bases___LazyGenericBaseClassrCrbaserrrrxszGenericClass.py__bases__cCs t|j|Sr)rarbrDrrrrG z+GenericClass._create_instance_with_genericscst|rdS|j|S)NT)r9is_sub_class_ofrb)rrcr@rrr}s  zGenericClass.is_sub_class_ofcCs |j|Sr)rb with_generics)rZgenerics_tuplerrrr~r|zGenericClass.with_genericsc Csddlm}m}|}i}|dkr#|}|r!|d|S|S|D]/}|r4|s4| }nq%|j dkr;q%| D]}|} || krS|||||nq?q%|S)Nr)merge_pairwise_genericsmerge_type_var_dictsIterableclass) Z!jedi.inference.gradual.annotationrrr%rIinfer_type_varsZmerge_types_of_iterateZ is_instanceZ is_compiledget_annotated_class_objectapi_type py__mro__) rrrrZannotation_namerPZannotation_genericsZpy_classZ parent_class class_namerrrrs6   zGenericClass.infer_type_varsT)r(r*r+r,rrfrnr<rsrwr rxrGr}r~rrArrr@rras     rac@s0eZdZddZeddZddZddZd S) rycCs||_||_||_dSr)rb_lazy_base_classrC)rrcZlazy_base_classrErrrrs z_LazyGenericBaseClass.__init__c csh|jD]+}t|trt|j|jtt| |Vq| r.t|j||j Vq|VqdSr) rr#rraZ create_cachedinference_stater_rrO_remap_type_varsZis_class_mixinrCrzrrrr#s  z_LazyGenericBaseClass.inferccs~ddlm}|j}|D],}t}|D]"}t||r1||}|t dd|DO}q|t |gO}q|VqdS)NrrJcss|]}|VqdSr)r#)rUr3rrrrW!s z9_LazyGenericBaseClass._remap_type_vars..) rLrKrbr<rIrrr5r%rZ from_sets)rr{rKfilterZ type_var_setnewrnamesrrrrs      z&_LazyGenericBaseClass._remap_type_varscCd|jj|jfSNz<%s: %s>)r'r(rr!rrrr)+rrz_LazyGenericBaseClass.__repr__N)r(r*r+rrr#rr)rrrrrys   ryc@seZdZddZdddZdS)rtc Csx|jjD]0}|dkr%|}z |dWSty$Yqw|dkr6tt|j dgSq|j S)N GeneratorIteratorNone) r_rcrr%rIexecute_annotationr2rr rpy__stop_iteration_returns)rclsr0rrrr0s    z2_GenericInstanceWrapper.py__stop_iteration_returnsTcCs|jjjddS)NF)ro)r_rcrn)rrorrrrn<sz%_GenericInstanceWrapper.get_type_hintNr)r(r*r+rrnrrrrrt/s rtcs\eZdZdZdZfddZeddZddZd d Z ed d Z d dZ ddZ Z S)_PseudoTreeNameClassa2 In typeshed, some classes are defined like this: Tuple: _SpecialForm = ... Now this is not a real class, therefore we have to do some workarounds like this class. Essentially this class makes it possible to goto that `Tuple` name, without affecting anything else negatively. rcst|j|||_dSr)r9rr _tree_namerr tree_namer@rrrLs  z_PseudoTreeNameClass.__init__cCrdr)rr!rrrrYSsz_PseudoTreeNameClass.tree_nodecosGdddt}|VdS)Nc@s$eZdZddZddZddZdS)z5_PseudoTreeNameClass.get_filters..EmptyFiltercSsdSrrr!rrrrZrHz>_PseudoTreeNameClass.get_filters..EmptyFilter.__init__c[r6rr)rr3r>rrrr5]rHz9_PseudoTreeNameClass.get_filters..EmptyFilter.getc[r6rr)rr>rrrr7`rHz<_PseudoTreeNameClass.get_filters..EmptyFilter.valuesN)r(r*r+rr5r7rrrr EmptyFilterYs rr )rr=r>rrrrr:Ws z _PseudoTreeNameClass.get_filterscCs t|jdS)Ntype)r rr!rrr py__class__es z _PseudoTreeNameClass.py__class__cC t||jSrrrr!rrrr3j z_PseudoTreeNameClass.namecCs |jjfSr)rr r!rrrget_qualified_namesnr&z(_PseudoTreeNameClass.get_qualified_namescCd|jj|jjfSNz%s(%s)r'r(rr r!rrrr)qz_PseudoTreeNameClass.__repr__)r(r*r+r,rrpropertyrYr:rr3rr)rArrr@rr@s   rc@s8eZdZddZeddZddZddZd d Zd S) BaseTypingValuecCs|j|_||_||_dSr)rrrrrrrrvs zBaseTypingValue.__init__cCrrrr!rrrr3{rzBaseTypingValue.namecCt|j|jSrrrrr!rrrrfz"BaseTypingValue._get_wrapped_valuecCr$r)r_r^r!rrrr^r&zBaseTypingValue.get_signaturescCrrrr!rrrr)rzBaseTypingValue.__repr__N) r(r*r+rrr3rfr^r)rrrrrus  rcs,eZdZfddZddZddZZS)BaseTypingClassWithGenericscs$t||j|_||_||_dSr)r9rrrr)rrrrEr@rrrs  z$BaseTypingClassWithGenerics.__init__cCrrrr!rrrrfrz.BaseTypingClassWithGenerics._get_wrapped_valuecCsd|jj|jj|jfS)Nz%s(%s%s))r'r(rr rCr!rrrr)sz$BaseTypingClassWithGenerics.__repr__)r(r*r+rrfr)rArrr@rrs rc@sHeZdZddZddZddZddZed d Zd d Z d dZ dS)BaseTypingInstancecCs$|j|_||_||_||_||_dSr)rrrbrrC)rrrcrrErrrrs  zBaseTypingInstance.__init__cCrdrrer!rrrrrgzBaseTypingInstance.py__class__cCrdrrer!rrrrrgz-BaseTypingInstance.get_annotated_class_objectcCs |fSr)r%r!rrrrr&z&BaseTypingInstance.get_qualified_namescCrrrr!rrrr3rzBaseTypingInstance.namecCst|jd\}|S)Nobject)r rr)rZobject_rrrrfsz%BaseTypingInstance._get_wrapped_valuecCrr)r'r(rCr!rrrr)rrzBaseTypingInstance.__repr__N) r(r*r+rrrrrr3rfr)rrrrrs  rN)"Zjedi.inference.cacherZjedi.inference.base_valuerrrrrrZjedi.inference.compiledr Zjedi.inference.value.klassr r Zjedi.inference.utilsr Zjedi.inference.namesrrZjedi.inference.contextrZjedi.inference.gradual.genericsrrr-r8rBraryrtrrrrrrrrs(        N\35