B /d*@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_from import_nametestZor_testZand_testZnot_testZ comparisonexprZxor_exprZand_exprZ shift_exprZ arith_expr atom_exprtermfactorpoweratom) 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)r1^/work/yifan.wang/ringdown/master-ringdown-env/lib/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%)r4lastr1r1r2get_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 r5r)r)r%)for_stmtr1r1r2for_stmt_defines_one_nameGsr:cCsj|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-r;keywordir0Z first_leafr1r1r2get_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-r1r1r2clean_scope_docstring`srGcCsP|jdkrL|j}|}|dk rL|jdkrL|jd}|jdkrLtt|jSdS)NrZ simple_stmtrstringrD)r%r'Zget_next_siblingr)rrErF)Z tree_nodeZ maybe_stringr1r1r2find_statement_documentationls    rIcCs0|dd}|ddks$|dkr(dSt|S)Nrf)frrfrD)lowerr)rFZ first_twor1r1r2rExsrEHc 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 Nlambdefzr5r$rDcss|]}|VqdS)N)get_code).0rr1r1r2 sz get_signature..r"z\s+ z -> ) r%rrF get_paramsjoinstripresub annotationrQtextwrapwrap) rwidthZ call_stringZomit_first_paramZomit_return_annotationparamspZrtypecoder1r1r2 get_signatures      rbcCsJy |j}Wn"tk r,|j|7_YnXx|D]}t||q4WdS)z$ Move the `Node` start_pos. N)r)r+linemove)r- line_offsetr)cr1r1r2rds   rdcCsyj|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 # r9Z with_stmtrN# rU) r%r)r?prefixr&Z get_last_leafr+r=index)r- whitespacecommentr1r1r2get_following_comment_same_lines(   rpcCs&|j}|dkr|jdjdkS|dkS)Nr4r5r6)Z file_inputr rrPr6)r%r))r-tr1r1r2is_scopesrrcstdfdd }|S)NFcst|dkr||Sy |}Wn tk r>i}|<YnXy||Stk rn||}||<|SXdS)N)KeyError)Zparso_cache_noder- include_flowsZ for_moduler.)cachefuncr1r2wrappers  z(_get_parent_scope_cache..wrapper)F)r)rvrwr1)rurvr2_get_parent_scope_cachesrxcsj}|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 rrP:rZtfpdefrZif_stmtc3s,|]$}|jjko|jknVqdS)N)r;r<)rRn)r-r1r2rSsz#get_parent_scope..) r'rrr%r)rmr;r isinstancerZFlowanyZget_test_nodes)r-rtscopermr1)r-r2get_parent_scopes*    r~cCs 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)grammarpathr1r1r2get_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_hashed)rrr1r1r2rsrcCsjt|jddd|d|jd}|d}|j|dkrD||j8}|sLdS|dd||d<d|S)z3 Cuts of the value of the leaf at position T)keependsNrr5rDr3)rrFrccolumnrW)leafpositionrrr1r1r2cut_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()`. rrhrr$r5Fr awaitcss|]}|jddkVqdS)r.N)r))rRrfr1r1r2rS?sz!expr_is_dotted..Nr)r%lenr)expr_is_dottedall)r-r)r1r1r2r0s    rcsfdd}|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. r5TF)Zget_decoratorsr)rQ)Z function_noder Z dotted_name) method_namesr1r2rwDs   z&_function_is_x_method..wrapperr1)rrwr1)rr2_function_is_x_methodCs r staticmethod classmethodpropertycached_property)F)rONFF)F)&rYr\astrinspectrweakrefrZ parso.pythonrZ parso.cacherZparsorr,r@r*r8r:rCrGrIrErbrdrprrrxr~Zget_cached_parent_scoperrrrrZfunction_is_staticmethodZfunction_is_classmethodZfunction_is_propertyr1r1r1r2sF         %