B b*@sBddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd d d d d dddddddddddddhZdZdFddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*ZdGd,d-Zd.d/Zd0d1Zd2d3Zd4d5ZdHd6d7ZeeZd8d9Zd:d;Zdd?Z!d@dAZ"e"dBZ#e"dCZ$e"dDdEZ%dS)IN) literal_eval)cleandoc)WeakKeyDictionary)tree) parser_cache) split_linesfuncdefclassdefZ import_fromZ import_nameZtestZor_testZand_testZnot_testZ comparisonexprZxor_exprZand_exprZ shift_exprZ arith_expr atom_exprZtermZfactorZpoweratom) tryexceptfinallyelseifelifwithforwhileFcCsg}|j}|dkrB|}|dkr@|jjdkr@|dkr@||n|dkrv||x|jD]}|t|dd7}q\Wn|dkr|jd d kr|jd }|d kr|t|7}nTy |j}Wntk rYn6X|jtkr|s||x|D]}|t||7}qW|S) z For static analysis. nameFparam= expr_stmtT) last_added decorator)()type get_next_leafparentappendchildrenget_executable_nodesAttributeError_EXECUTE_NODES)noderresulttypZ next_leafchildr$r,0lib/python3.7/site-packages/jedi/parser_utils.pyr%s0       r%ccsZ|V|jd}xD|jdkr*|jdVn|jdkr<|Vn |jdksHP|jd}qWdS)Ncomp_for sync_comp_forZcomp_if)r$r )r/Zlastr,r,r-get_sync_comp_fors:s    r2cCs|jdjdkS)z Returns True if only one name is returned: ``for x in y``. Returns False if the for loop is more complicated: ``for x, z in y``. :returns: bool r0r)r$r )for_stmtr,r,r-for_stmt_defines_one_nameGsr4cCsj|j}|j|kr|jks(ntdd}x8t|jD]*\}}||jkrN|S|}|tkr8|}q8WdS)Nz!The node is not part of the flow.) start_posend_pos ValueError enumerater$get_first_leaf_FLOW_KEYWORDS)Z flow_noder(r5keywordir+Z first_leafr,r,r-get_flow_branch_keywordQs r=cCs"|}|dk rtt|jSdS)z3 Returns a cleaned version of the docstring token. N)Z get_doc_nodersafe_literal_evalvalue)Z scope_noder(r,r,r-clean_scope_docstring`srAcCsP|jdkrL|j}|}|dk rL|jdkrL|jd}|jdkrLtt|jSdS)NrZ simple_stmtrstringr>)r r"Zget_next_siblingr$rr?r@)Z tree_nodeZ maybe_stringr,r,r-find_statement_documentationls    rCcCs0|dd}|ddks$|dkr(dSt|S)Nrf)frZrfr>)lowerr)r@Z first_twor,r,r-r?xsr?Hc Cs|dkr |jdkrd}n|jj}|}|r8|dd}dddd|Dd }td d |}|jr|sd |j }nd}|||}d t ||S)z Generate a string signature of a function. :param width: Fold lines if a line is longer than this value. :type width: int :arg func_name: Override function name when given. :type func_name: str :rtype: str Nlambdefzr0rr>css|]}|VqdS)N)get_code).0rr,r,r- sz get_signature..rz\s+ z -> ) r rr@Z get_paramsjoinstripresubZ annotationrJtextwrapwrap) rwidthZ call_stringZomit_first_paramZomit_return_annotationparamspZrtypecoder,r,r- get_signatures      rYcCsJy |j}Wn"tk r,|j|7_YnXx|D]}t||q4WdS)z$ Move the `Node` start_pos. N)r$r&linemove)r(Z line_offsetr$cr,r,r-r[s   r[cCsyj|jdkr|jdj}nJ|jdkr:|jdj}n.|jdkrZ|jdj}n|j}Wn&tk r~dStk rdSXd|krdS||dd}d |kr|d|d }d |kr|d|d }|S) zl returns (as string) any comment that appears on the same line, after the node, including the # r3Z with_stmtrN# rN) r r$r9prefixr!Z get_last_leafr&r7index)r(Z whitespacecommentr,r,r-get_following_comment_same_lines(   recCs&|j}|dkr|jdjdkS|dkS)Nr/r0r1)Z file_inputr rrIr1)r r$)r(tr,r,r-is_scopesrgcstdfdd }|S)NFcst|dkr||Sy |}Wn tk r>i}|<YnXy||Stk rn||}||<|SXdS)N)KeyError)Zparso_cache_noder( include_flowsZ for_moduler))cachefuncr,r-wrappers  z(_get_parent_scope_cache..wrapper)F)r)rkrlr,)rjrkr-_get_parent_scope_cachesrmcsj}|dkrdSxt|r|jdkr|jd}|j|jjkrjjdkr^jjkr^n&jjdkr|jjdkr|n|j}q|S|rt|tj r|jdkrt fdd | Ds|S|j}qWdS) z' Returns the underlying scope. N)r rrI:rZtfpdefrZif_stmtc3s,|]$}|jjko|jknVqdS)N)r5r6)rKn)r(r,r-rLsz#get_parent_scope..) r"rgr r$rcr5r isinstancerZFlowanyZget_test_nodes)r(riZscopercr,)r(r-get_parent_scopes*    rrcCs t||jS)z Basically access the cached code lines in parso. This is not the nicest way to do this, but we avoid splitting all the lines again. )get_parso_cache_nodelines)grammarpathr,r,r-get_cached_code_linessrwcCst|j|S)z This is of course not public. But as long as I control parso, this shouldn't be a problem. ~ Dave The reason for this is mostly caching. This is obviously also a sign of a broken caching architecture. )rZ_hashed)rurvr,r,r-rssrscCsjt|jddd|d|jd}|d}|j|dkrD||j8}|sLdS|dd||d<d|S)z3 Cuts of the value of the leaf at position T)keependsNrr0r>r.)rr@rZcolumnrO)ZleafZpositionrtryr,r,r-cut_value_at_position"s$ rzcCs|jdkr8t|jdkr4|jddkr4t|jdSdS|jdkr|j}|ddkrXdSt|dshdStd d |dd DS|jd kS) zP Checks if a path looks like `name` or `name.foo.bar` and not `name()`. r r^rrr0Fr awaitcss|]}|jddkVqdS)r.N)r$)rKr\r,r,r-rL?sz!expr_is_dotted..Nr)r lenr$expr_is_dottedall)r(r$r,r,r-r~0s    r~csfdd}|S)Ncs0x*|D]}|jd}|kr dSq WdS)z This is a heuristic. It will not hold ALL the times, but it will be correct pretty much for anyone that doesn't try to beat it. staticmethod/classmethod are builtins and unless overwritten, this will be correct. r0TF)Zget_decoratorsr$rJ)Z function_noderZ dotted_name) method_namesr,r-rlDs   z&_function_is_x_method..wrapperr,)rrlr,)rr-_function_is_x_methodCs r staticmethod classmethodpropertyZcached_property)F)rHNFF)F)&rQrSZastrinspectrweakrefrZ parso.pythonrZ parso.cacherZparsorr'r:r%r2r4r=rArCr?rYr[rergrmrrZget_cached_parent_scoperwrsrzr~rZfunction_is_staticmethodZfunction_is_classmethodZfunction_is_propertyr,r,r,r-sF         %