B bZ@sddlmZddlmZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZmZddlmZdd lmZdd lmZdd lmZmZdd lmZd dZGdddZGdddeZGdddeZGdddZ Gddde eZ!GdddeZ"GdddZ#Gddde#Z$Gdd d e$eZ%Gd!d"d"e%Z&Gd#d$d$e&Z'Gd%d&d&e&Z(Gd'd(d(e#Z)Gd)d*d*eZ*Gd+d,d,e*Z+Gd-d.d.Z,Gd/d0d0Z-Gd1d2d2e-e"Z.Gd3d4d4e eZ/Gd5d6d6e-e/Z0d7S)8)abstractmethod) Parameter)OptionalTuple)search_ancestor)find_statement_documentationclean_scope_docstring)unite)ValueSet NO_VALUES)inference_state_method_cache) docstrings)memoize_method) deep_ast_copyinfer_call_of_leaf)plugin_managercCs.d}x$|D]}|r|d7}||7}q W|S)Nz ------------------------------ ) py__doc__)namesdocnamer3lib/python3.7/site-packages/jedi/inference/names.py_merge_name_docss  rc@seZdZUdZeeeefed<eed<dZ dZ dZ e ddZ e ddZdd d Zd d ZddZddZddZddZddZeddZddZdS)AbstractNameDefinitionN start_pos string_nameTcCstdS)N)NotImplementedError)selfrrrinfer&szAbstractNameDefinition.infercCs|hS)Nr)rrrrgoto*szAbstractNameDefinition.gotoFcCs6|}|dks|s|S|j}|dkr.dS||S)N)_get_qualified_namesget_root_context string_names)rinclude_module_namesZqualified_names module_namesrrrget_qualified_names0s  z*AbstractNameDefinition.get_qualified_namescCsdS)Nr)rrrrr!:sz+AbstractNameDefinition._get_qualified_namescCs |jS)N)parent_contextr")rrrrr">sz'AbstractNameDefinition.get_root_contextcCs|jS)N)r)rrrrget_public_nameAsz&AbstractNameDefinition.get_public_namecCs2|jdkrd|jj|jfSd|jj|j|jfS)Nz<%s: string_name=%s>z!<%s: string_name=%s start_pos=%s>)r __class____name__r)rrrr__repr__Ds zAbstractNameDefinition.__repr__cCsdS)NFr)rrrr is_importJsz AbstractNameDefinition.is_importcCsdS)Nrr)rrrrrMsz AbstractNameDefinition.py__doc__cCs|jjS)N)r'api_type)rrrrr-PszAbstractNameDefinition.api_typecCsdS)z Returns either None or the value that is public and qualified. Won't return a function, because a name in a function is never public. Nr)rrrrget_defining_qualified_valueTsz3AbstractNameDefinition.get_defining_qualified_value)F)r* __module__ __qualname__rrrint__annotations__strr' tree_name is_value_namerrr r&r!r"r(r+r,rpropertyr-r.rrrrrs     rc@s$eZdZdZdZddZddZdS)AbstractArbitraryNamez When you e.g. want to complete dicts keys, you probably want to complete string literals, which is not really a name, but for Jedi we use this concept of Name for completions as well. FcCs||_||_|j|_dS)N)inference_staterZbuiltins_moduler')rr8stringrrr__init__dszAbstractArbitraryName.__init__cCstS)N)r )rrrrriszAbstractArbitraryName.inferN)r*r/r0__doc__r5r:rrrrrr7\sr7cs^eZdZddZdfdd ZddZdd Zd d Zd d Ze ddZ e ddZ Z S)AbstractTreeNamecCs||_||_dS)N)r'r4)rr'r4rrrr:nszAbstractTreeName.__init__Fcsdt|jdd}|dk rX|jdkr0|sX|rT|jsTtdd||jDSdSt |S)N import_name import_fromcss|] }|jVqdS)N)value).0nrrr |sz7AbstractTreeName.get_qualified_names..) rr4levelr" get_value is_packagetupleZget_path_for_namesuperr&)rr$Z import_node)r)rrr&rs z$AbstractTreeName.get_qualified_namescCs$|j}|dkrdS||jjfS)N)r'r&r4r@)rZ parent_namesrrrr!s z%AbstractTreeName._get_qualified_namescCs&|rtdn|jr"|jSdS)Nz&Shouldn't really happen, please report)r,rr'rE)rrrrr.s   z-AbstractTreeName.get_defining_qualified_valuecs|j|jjdd}|dk rp|j}|dkrFjjdk}|rn|gSq|dkrhddlm}|}|S|gSn,ddlm}|}|dk rd d |DSj}|j} | d kr|jd d kr|jdkr|j} | jdkr| j} | jdkr<| jdkr | jd } nt| jj | } | jjd| } | ddkrH| d | d} ddl m }x"| d dD]} || | } qpWg}xN| D]F}x>|D]2}x*|D]}|jjkr||qWqWqW|Sn\| dkr<|j }|dkrr=r) goto_import)%follow_error_node_imports_if_possiblecSsg|] }|jqSr)r)rAr@rrr sz)AbstractTreeName.goto..argumentr?=ZarglistclassdefZ decoratorawait) infer_trailerZ dotted_namec3s|]}|jdVqdS)) name_contextN)r )rAr@)contextrrrrCsz(AbstractTreeName.goto..rK.)Zcut_own_trailer)rTZlambdef)Zposition)r'r4get_definitiontypeparentjedi.inference.importsrLrMchildren infer_nodeindexpopjedi.inference.syntax_treerSZget_signaturesZget_param_namesrr@appendrr rr rr)r definitionZtype_Zis_simple_namerLr%rMvaluesZparZ node_typerKZ value_setiZto_inferrSZ param_namesr@Z signature param_namer]Z new_dottedZstmtr)rUrrr sx      *              zAbstractTreeName.gotocCst|jdd}|dk S)Nr>r=)rr4)rimprrrr,szAbstractTreeName.is_importcCs|jjS)N)r4r@)rrrrrszAbstractTreeName.string_namecCs|jjS)N)r4r)rrrrrszAbstractTreeName.start_pos)F) r*r/r0r:r&r!r.r r,r6rr __classcell__rr)r)rr<msH r<csHeZdZddZddZddZfddZd d Zed d Z Z S) ValueNameMixincCs t|jgS)N)r _value)rrrrrszValueNameMixin.infercCsF|j}|sB|jrBddlm}||gdd}||krBt|S|S)Nr) convert_namesF)prefer_stub_to_compiled)rhris_stub!jedi.inference.gradual.conversionrir)rrrirrrrrs  zValueNameMixin.py__doc__cCs |jS)N)rhr&)rrrrr!sz#ValueNameMixin._get_qualified_namescs|jdkr|jStS)N)r'rh as_contextrHr")r)r)rrr"s  zValueNameMixin.get_root_contextcCs$|j}|s|r |jSdS)N)r'Z is_moduleZis_classrE)rrUrrrr.s z+ValueNameMixin.get_defining_qualified_valuecCs|jjS)N)rhr-)rrrrr-szValueNameMixin.api_type) r*r/r0rrr!r"r.r6r-rfrr)r)rrgs   rgcs$eZdZfddZddZZS) ValueNamecst|j|||_dS)N)rHr:r'rh)rr@r4)r)rrr:szValueName.__init__cCst|jjgS)N)r rhr)rrrrr szValueName.goto)r*r/r0r:r rfrr)r)rrns rnc@sXeZdZeddddddZddZedd Zd d Zed d Z e ddddZ dS)TreeNameDefinitionmodulefunctionparamclass)r=r>ZfuncdefrrrQcCs ddlm}||jj|j|jS)Nr)tree_name_to_values)r_rtr'r8r4)rrtrrrrs  zTreeNameDefinition.infercCs*|jjdd}|dkrdS|j|jdS)NT)rI statement)r4rW _API_TYPESgetrX)rrarrrr-szTreeNameDefinition.api_typec Csg}d}|jj}|j}x|dk r|jdkrxjt|jD]T\}}||kr6t|d}|rxtt|j|d}t|| }|d||fPq6Wt dd}n|jdkrd}n |jd krP|}|j}qW|S) a Returns an array of tuple(int, node) of the indexes that are used in tuple assignments. For example if the name is ``y`` in the following code:: x, (y, z) = 2, '' would result in ``[(1, xyz_node), (0, yz_node)]``. When searching for b in the case ``a, *b, c = [...]`` it will return:: [(slice(1, -1), abc_node)] FN)ZtestlistZ testlist_compZtestlist_star_exprZexprlistrzCouldn't find the assignment.Z star_exprT)rJZ sync_comp_for) r4rYrX enumerater[r1lensliceinsert LookupError) rZindexesZ is_star_exprnodeZcomparercZchildr]Zfrom_endrrrassignment_indexes&s.       z%TreeNameDefinition.assignment_indexescCs|jjS)N)r'r8)rrrrr8Osz"TreeNameDefinition.inference_stater)defaultcCs|j}|dkrT|jrFddlm}||gdd}||krFt|St|j S|dkrt| }||krtt|S|dkr|j rt |j SdS) N)rqrsr6r)riF)rjrprur) r-r'r"rkrlrirrr4rWr Z is_definitionr)rr-rirrrrrTs zTreeNameDefinition.py__doc__N) r*r/r0dictrvrr6r-rr8r rrrrrro s ) roc@s2eZdZd ddZd ddZddZdd d Zd S) _ParamMixinTcCs(tjtjg}|r|tj||kS)N)rPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDr`VAR_POSITIONALget_kind)rZ include_staroptionsrrrmaybe_positional_argumentls  z%_ParamMixin.maybe_positional_argumentcCs(tjtjg}|r|tj||kS)N)r KEYWORD_ONLYrr` VAR_KEYWORDr)rZ include_starsrrrrmaybe_keyword_argumentrs  z"_ParamMixin.maybe_keyword_argumentcCs(|}|tjkrdS|tjkr$dSdS)N*z**r)rrrr)rkindrrr _kind_stringxs   z_ParamMixin._kind_stringFcCsdS)Nr)rr$rrrr&sz_ParamMixin.get_qualified_namesN)T)T)F)r*r/r0rrrr&rrrrrks  rc@s<eZdZdZddZddZddZedd Zd d Z d S) ParamNameInterfacerrcCstdS)N)r)rrrrrszParamNameInterface.get_kindcCstdS)N)r)rrrr to_stringszParamNameInterface.to_stringcCsdS)as For dealing with type inference and working around the graph, we sometimes want to have the param name of the execution. This feels a bit strange and we might have to refactor at some point. For now however it exists to avoid infering params when we don't really need them (e.g. when we can just instead use annotations. Nr)rrrrget_executed_param_names z*ParamNameInterface.get_executed_param_namecCs(|}|tjkrdS|tjkr$dSdS)Nr?rxr)rrrr)rrrrr star_counts   zParamNameInterface.star_countcCstS)N)r )rrrr infer_defaultsz ParamNameInterface.infer_defaultN) r*r/r0r-rrrr6rrrrrrrs  rc@s,eZdZdZdZddZddZddZdS)BaseTreeParamNameNcCsX||}|j}|j}|dk r8|d|jdd7}|dk rT|d|jdd7}|S)Nz: F)Zinclude_prefixrP)rr(annotation_node default_nodeget_code)routput annotationrrrrrszBaseTreeParamName.to_stringcCs |j}|dr|dd}|S)N__rx)r startswith)rrrrrr(s  z!BaseTreeParamName.get_public_namecKs|gS)Nr)rkwargsrrrr szBaseTreeParamName.goto)r*r/r0rrrr(r rrrrrs  rcs^eZdZfddZddZeddZdd d Zd d Zed dZ ddZ ddZ Z S)_ActualTreeParamNamecst||||_dS)N)rHr:Zget_default_param_contextfunction_value)rrr4)r)rrr:s z_ActualTreeParamName.__init__cCs t|jdS)Nrr)rr4)rrrr_get_param_nodesz$_ActualTreeParamName._get_param_nodecCs |jS)N)rr)rrrrrsz$_ActualTreeParamName.annotation_nodeTFcCs0ddlm}||j||d}|r,|}|S)Nr) infer_param) ignore_stars)Z!jedi.inference.gradual.annotationrrrexecute_annotation)rrrrrbrrrinfer_annotations  z%_ActualTreeParamName.infer_annotationcCs|j}|dkrtS|j|S)N)rr r'r\)rr~rrrrsz"_ActualTreeParamName.infer_defaultcCs |jS)N)rr)rrrrrsz!_ActualTreeParamName.default_nodecCs|}|jdkrtjS|jdkr(tjS|jjdr)r)r*r)rrrrr+:szParamNameWrapper.__repr__N)r*r/r0r:rr+rrrrr3src@sZeZdZdZdZddZdddZedd Ze d d Z d d Z eddZ ddZ dS) ImportName)r?rrcCs||_||_dS)N)_from_module_contextr)rr'rrrrr:BszImportName.__init__FcCsH|rD|jr<|jdkstd|jj}|dkr0|S||jfS|jfSdS)Nr?z(Everything else is not supported for nowr)_levelAssertionErrorrr#r)rr$r%rrrr&Fs zImportName.get_qualified_namescCs&|j}|}|s|Stt|S)N)rrnextiterrm)rmZ import_valuesrrrr'Qs zImportName.parent_contextcCs.ddlm}|j}||j|jg||jdS)Nr)Importer)rD)rZrrr8rrZfollow)rrrrrrr[s zImportName.infercCsdd|DS)NcSsg|] }|jqSr)r)rArrrrrNbsz#ImportName.goto..)r)rrrrr aszImportName.gotocCsdS)Nrpr)rrrrr-dszImportName.api_typecCs t|S)N)rr )rrrrrhszImportName.py__doc__N)F)r*r/r0rrr:r&r6r'rrr r-rrrrrr>s  rc@seZdZdZdS) SubModuleNamer?N)r*r/r0rrrrrrlsrc@s$eZdZddZddZddZdS) NameWrappercCs ||_dS)N) _wrapped_name)rZ wrapped_namerrrr:qszNameWrapper.__init__cCs t|j|S)N)rr)rrrrrrtszNameWrapper.__getattr__cCsd|jj|jfS)Nz%s(%s))r)r*r)rrrrr+wszNameWrapper.__repr__N)r*r/r0r:rr+rrrrrpsrcseZdZfddZZS) StubNameMixincshddlm}|g}|jdkr>d|jjkr>dd|D}||dd}||kr\tSt |SdS) Nr)rirurPcSsg|] }|jqSr)r)rAvrrrrNsz+StubNameMixin.py__doc__..F)rj) rlrir-r4rWr[rrHrr)rrir)r)rrr|s   zStubNameMixin.py__doc__)r*r/r0rrfrr)r)rr{srcseZdZfddZZS)StubNamecsJt}|jdkrF|dkrFddlmtfdd|DS|S)N version_infosysr) VersionInfoc3s|]}|VqdS)Nr)rAc)rrrrCsz!StubName.infer..)rHrrr"Z py__name__Z!jedi.inference.gradual.stub_valuerr )rZinferred)r))rrrs   zStubName.infer)r*r/r0rrfrr)r)rrsrc@s$eZdZdZddZeddZdS) ModuleName)r?rcCs||_||_dS)N)rh_name)rr@rrrrr:szModuleName.__init__cCs|jS)N)r)rrrrrszModuleName.string_nameN)r*r/r0rr:r6rrrrrrsrc@s eZdZdS)StubModuleNameN)r*r/r0rrrrrsrN)1abcrinspectrtypingrrZ parso.treerZjedi.parser_utilsrrZjedi.inference.utilsr Zjedi.inference.base_valuer r Zjedi.inference.cacher Zjedi.inferencer Z jedi.cacherZjedi.inference.helpersrrZ jedi.pluginsrrrr7r<rgrnrorrrrrrrrrrrrrrrrrrs@         @w  ^!E .