B 7r\@sddlZddlZddlmZddlmZddlmZddlm Z m Z dddd d d d d dddddddddddhZ dZ d8ddZ ddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd9d*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd:d4d5Zd6d7ZdS);N)cleandoc)tree) parser_cache) literal_eval force_unicodefuncdefclassdefZ import_fromZ import_nameZtestZor_testZand_testZnot_testZ comparisonexprZxor_exprZand_exprZ shift_exprZ arith_exprZ atom_exprZtermZfactorZpowerZatom) 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=Z expr_stmtT) last_addedZ decorator)()type get_next_leafparentappendchildrenget_executable_nodesAttributeError_EXECUTE_NODES)noderresulttypZ next_leafchildrr'0lib/python3.7/site-packages/jedi/parser_utils.pyr s0       r ccsB|V|jd}x,|jdkr$|Vn |jdks0P|jd}qWdS)Ncomp_forZcomp_if)rr)r*Zlastr'r'r( get_comp_fors9s   r+cCs|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 r)rr)for_stmtr'r'r(for_stmt_defines_one_nameDsr.cCsj|j}|j|kr|jks(ntdd}x8t|jD]*\}}||jkrN|S|}|tkr8|}q8WdS)Nz!The node is not part of the flow.r) start_posend_pos ValueError enumeraterget_first_leaf_FLOW_KEYWORDS)Z flow_noder#r/keywordir&Z first_leafr'r'r(get_flow_branch_keywordNs r7c Cspxj|jD]`}|j|kr$|jkrnq|jdkrHt|tjtjfsH|Sy t||St k rfYqXqWdS)N)Z decoratedZ simple_stmtZsuiteZ async_stmtZ async_funcdef) rr/r0r isinstancerFlowZ ClassOrFuncget_statement_of_positionr!)r#poscr'r'r(r:]s    r:cCs*|}|dk r&tt|j}t|SdS)z3 Returns a cleaned version of the docstring token. N)Z get_doc_nodersafe_literal_evalvaluer) scope_noder#Zcleanedr'r'r(clean_scope_docstringls rAcCsJ|dd}|ddks$|dkr(dSyt|Stk rDdSXdS)Nrf)frZrfr=)lowerr SyntaxError)r?Z first_twor'r'r(r>{sr>HcCs|dkr |jdkrd}n|jj}|jdkrPdddd|Dd}n|jd }t d d |}|j rd |j }nd}|||}d t ||S)z Generate call signature of this 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 Nlambdefzrr=css|]}|VqdS)N)get_code).0rr'r'r( sz%get_call_signature..rrBz\s+ z -> ) rrr?joinZ get_paramsstriprrIresubZ annotationtextwrapZwrap)rwidth call_stringpZrtypecoder'r'r(get_call_signatures   & rWcCsxd}|jdkr>x@|D] }|jjdkrt||jjd}qWn|jdkrPt|}t|}|dkrd|S|sl|Sd||fS)z< Return a document string including call signature. Nr__init__)rT)rrHz%s %s)rZ iter_funcdefsrr?rWrA)r@Zcall_signaturerdocr'r'r(get_doc_with_call_signatures   rZcCsJy |j}Wn"tk r,|j|7_YnXx|D]}t||q4WdS)z$ Move the `Node` start_pos. N)rr!linemove)r#Z line_offsetrr<r'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 # r-Z with_stmtrN# rM) rrr3prefixrZ get_last_leafr!r1index)r#Z whitespacecommentr'r'r(get_following_comment_same_lines(   recCs |jdkS)N)Z file_inputrrrHr*)r)r#r'r'r(is_scopesrfcCs<|j}x0|dk r6|r$t|tjr$|St|r.P|j}qW|S)z' Returns the underlying scope. N)rr8rr9rf)r#Z include_flowsZscoper'r'r(get_parent_scopes  rgcCst|j|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. )rZ_hashedlines)Zgrammarpathr'r'r(get_cached_code_linessrj)F)rGN)F)rPrRinspectrZ parso.pythonrZ parso.cacherZjedi._compatibilityrrr"r4r r+r.r7r:rAr>rWrZr\rerfrgrjr'r'r'r(s0      %