U nu f*@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)cleandocWeakKeyDictionary)tree) parser_cache) split_linesfuncdefclassdefZ import_from import_nametestZor_testZand_testZnot_testZ comparisonexprZxor_exprZand_exprZ shift_exprZ arith_expr atom_exprZtermZfactorpoweratom) tryexceptfinallyelseifelifwithforwhileFcCsg}|j}|dkrB|}|dkr|jjdkr|dkr||n|dkrr|||jD]}|t|dd7}qZn|dkr|jd d kr|jd }|d kr|t|7}nPz |j}Wntk rYn2X|jtkr|s|||D]}|t||7}q|S) z For static analysis. nameFparam= expr_stmtT) last_added decorator)()type get_next_leafparentappendchildrenget_executable_nodesAttributeError_EXECUTE_NODES)noderresulttypZ next_leafchildr(r05/tmp/pip-unpacked-wheel-ol4ehs9f/jedi/parser_utils.pyr)s0      r)ccsV|V|jd}|jdkr(|jdVn|jdkr:|Vn |jdksFqR|jd}qdS)Ncomp_for sync_comp_forZcomp_ifr(r$)r3lastr0r0r1get_sync_comp_fors:s    r8cCs|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 r4rr6)for_stmtr0r0r1for_stmt_defines_one_nameGsr:cCsj|j}|j|kr|jks(ntdd}t|jD].\}}||jkrP|S|}|tkr6|}q6dS)Nz!The node is not part of the flow.) start_posend_pos ValueError enumerater(get_first_leaf_FLOW_KEYWORDS)Z flow_noder,r;keywordir/Z first_leafr0r0r1get_flow_branch_keywordQs rCcCs"|}|dk rtt|jSdS)z3 Returns a cleaned version of the docstring token. N)Z get_doc_nodersafe_literal_evalvalue)Z scope_noder,r0r0r1clean_scope_docstring`srGcCsP|jdkrL|j}|}|dk rL|jdkrL|jd}|jdkrLtt|jSdS)NrZ simple_stmtrstringrD)r$r&Zget_next_siblingr(rrErF)Z tree_nodeZ maybe_stringr0r0r1find_statement_documentationls    rIcCs0|dd}|ddks$|dkr(dSt|S)Nrf)frrfrD)lowerr)rFZ first_twor0r0r1rExsrEHc 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 Nlambdefzr4r#rDcss|]}|VqdSN)get_code).0rr0r0r1 sz get_signature..r!z\s+ z -> ) r$rrF get_paramsjoinstripresub annotationrRtextwrapwrap) r widthZ call_stringZomit_first_paramZomit_return_annotationparamspZrtypecoder0r0r1 get_signatures      rccCsFz |j}Wn"tk r,|j|7_YnX|D]}t||q2dS)z$ Move the `Node` start_pos. N)r(r*linemove)r, line_offsetr(cr0r0r1res  recCszj|jdkr|jdj}nJ|jdkr:|jdj}n.|jdkrZ|jdj}n|j}Wn*tk rYdStk rYdSXd|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 # r9Z with_stmtr N# rV) r$r(r?prefixr%Z get_last_leafr*r=index)r, whitespacecommentr0r0r1get_following_comment_same_lines(   rqcCs&|j}|dkr|jdjdkS|dkS)Nr3r4r5)Z file_inputr r rPr5)r$r()r,tr0r0r1is_scopesrscstdfdd }|S)NFcsz|dkr||Sz |}Wn tk r>i}|<YnXz ||WStk rt||}||<|YSXdSrQ)KeyError)Zparso_cache_noder, include_flowsZ for_moduler-cachefuncr0r1wrappers   z(_get_parent_scope_cache..wrapper)Fr)rxryr0rvr1_get_parent_scope_cachesrzcsj}|dkrdSt|r|jdkr|jd}|j|jjkrjjdkr\jjkr\n&jjdkrzjjdkrzn|j}q|S|rt|tj r|jdkrt fdd | Ds|S|j}qdS) z' Returns the underlying scope. N)r r rP:rZtfpdefrZif_stmtc3s,|]$}|jjko|jknVqdSrQ)r;r<)rSnr,r0r1rTsz#get_parent_scope..) r&rsr$r(rnr;r isinstancerZFlowanyZget_test_nodes)r,ruZscopernr0r}r1get_parent_scopes,    rcCs 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_nodelinesZgrammarpathr0r0r1get_cached_code_linessrcCst|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_hashedrr0r0r1rsrcCsjt|jddd|d|jd}|d}|j|dkrD||j8}|sLdS|dd||d<d|S)z3 Cuts of the value of the leaf at position T)keependsNrr4rDr2)rrFrdcolumnrX)Zleafpositionrrr0r0r1cut_value_at_position"s$ rcCs|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()`. rrirr#r4Frawaitcss|]}|jddkVqdS)r.N)r()rSrgr0r0r1rT?sz!expr_is_dotted..Nr)r$lenr(expr_is_dottedall)r,r(r0r0r1r0s    rcsfdd}|S)Ncs.|D] }|jd}|krdSqdS)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. r4TF)Zget_decoratorsr(rR)Z function_noderZ dotted_name method_namesr0r1ryDs    z&_function_is_x_method..wrapperr0)rryr0rr1_function_is_x_methodCs r staticmethod classmethodpropertycached_property)F)rONFF)F)&rZr]astrinspectrweakrefrZ parso.pythonrZ parso.cacherZparsorr+r@r)r8r:rCrGrIrErcrerqrsrzrZget_cached_parent_scoperrrrrZfunction_is_staticmethodZfunction_is_classmethodZfunction_is_propertyr0r0r0r1sb        %      !