B 7r\<$@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZed d ddgZddZddZddZGdddeZddZddZddZeddd d!gZd"d#Zd$d%Zd&d'Z ed(d)d*Z!dS)+z Helpers for the API N) namedtuple)dedent)Parser)tree)u) eval_atom)evaluate_call_of_leaf)get_string_context_set)call_signature_time_cacheCompletionPartspathZhas_dotnamecCst|dddS)NcSs|jpd|jpd|jpdfS)Nr) module_pathlinecolumn)xr/lib/python3.7/site-packages/jedi/api/helpers.pysz$sorted_definitions..)key)sorted)Zdefsrrrsorted_definitionssrcCst||}|dks|jdkrJ||dd}td|d|ddS|jdkrXdS|jd|d|jdS)N)string error_leafrz (?!\d)\w+$|$)r keywordr)get_leaf_for_positiontyperesearchgroupvalue start_pos) module_nodelinespositionleafrrrrget_on_completion_names  r(cCsR||dd|d}|dd|d|d<|d|dd|d<d|S)Nrrr)join) code_linesr#end_posr%rrr _get_code(sr-c@seZdZeddZdS) OnErrorLeafcCs |jdS)Nr)args)selfrrrr3szOnErrorLeaf.error_leafN)__name__ __module__ __qualname__propertyrrrrrr.2sr.cCs|j|dd}|j|kr0|}|dkr0tdS|jdk}x&|jdkr`|}|dkr.EndMarkerReachedN)r1r2r3rrrrEndMarkerReachedesr:c3sb|}xR|D]J}|jkr&q|jr:q|jrT|Vq|VqWdS)N) _tokenizerprefixendswith)codetokenstoken)r:grammarsafewordrrtokenize_without_endmarkerhs     z9get_stack_at_position..tokenize_without_endmarkerZ)ZZZ_USER_WANTS_TO_COMPLETE_HERE_WITH_JEDI T)Zerror_recovery)r?z7This really shouldn't happen. There's a bug in Jedi: %sN) Exceptionrr9rZ _pgen_grammarparsestack SystemErrorlist)rAr+r$posrCr>pr)r:rArBrget_stack_at_positionas rLcCsr|jdkr|||S|j}|jdkr2||jS|jdkrFt||St|tjr\t||S|jdkrnt |SgS)Nr atomtrailer)Zfstring_stringZ fstring_startZ fstring_end) rZgoto_definitionsr8Z eval_noder isinstancerZLiteralrr ) evaluatorcontextr'r8rrrevaluate_goto_definitions         rRCallSignatureDetails bracket_leafZ call_indexZkeyword_name_strcsfdd|D}|djdkr8fdd|djD}d}|r|d}|jdkr~|jdd kr~|jdjkr~|jd j}n|d kr|d j}|d |fS) zG Returns the amount of commas and the keyword argument string. csg|]}|jkr|qSr)r#).0c)r&rr sz&_get_index_and_key..r)Zarglistcsg|]}|jkr|qSr)r#)rUrV)r&rrrWsNZargumentr=r,)rchildrenr,r"count)Znodesr&Z nodes_beforeZkey_strZlastr)r&r_get_index_and_keys r]cCsxttt|jD]l\}}|dkr|j|kr|dkr|j|d}|}|dkrVq|jdksl|jjdkrt|ft ||SqWdS)N(rr )rNrM) reversedrI enumerater[r,r7rr8rSr])noder&indexelementr[r rrr+_get_call_signature_details_from_error_nodesrdcCs|j|dd}|j|kr,|}|dkr,dS|dkrF|j|krF|}|j}x|dk r|jdkrddSxB|jdddD].}|j|krv|jdkrvt||}|dk rv|SqvW|jdkr|jdd kr|}|dkrdSt |jdft |j|S|j}qNWdS) NT)r5))ZfuncdefZclassdefr)Z error_noderNrr^) rr#r7r,Z get_next_leafr8rr[rdrSr])moduler&r'ranresultrrrget_call_signature_detailss2      riZcall_signatures_validityc cs|dd}||d|d}||jd|}d||g}td|tj} |} | dkrndVn| | |jfVt|||VdS)z'This function calculates the cache key.rrNrz.*\() r#r*rmatchDOTALLZget_root_contextZ py__file__rRr7) rPrQrTr+Zuser_posZ line_indexZ before_cursorZ other_linesZwholeZbefore_bracketrrrrcache_call_signaturess  rl)"__doc__r collectionsrtextwraprZparso.python.parserrZ parso.pythonrZjedi._compatibilityrZjedi.evaluate.syntax_treerZjedi.evaluate.helpersrZjedi.evaluate.compiledr Z jedi.cacher r rr(r-rEr.r9rLrRrSr]rdrirlrrrrs2          )* '