o XAr@s>ddlmZmZmZmZddlZddlZddlZddlZddl m Z m Z ddl mZmZmZmZmZddl mZddlmZmZmZmZdejvZejad d Zd Zd Zej ej!ej"ej#ej$ej%ej&fZ'ej(d dZ)ej(ddZ*Gddde+Z,dddZ-ddZ.Gddde/Z0Gddde0Z1e0Z2de2_3ddZ4dS))absolute_importdivisionprint_functionunicode_literalsN) completersmy_shlex) USING_PYTHON2str sys_encoding ensure_str ensure_bytes)FilesCompleter)IntrospectiveArgumentParseraction_is_satisfiedaction_is_openaction_is_greedy _ARC_DEBUGcGstr t|dtidSdS)Nfile)_DEBUGprint debug_streamargsr6/usr/lib/python3/dist-packages/argcomplete/__init__.pydebugsrOPccs2tj}ttjdt_z dVW|t_dS|t_wNw)sysstdoutopenosdevnull)r"rrr mute_stdouts r&ccsFtj}ttjdt_zdVWtj|t_dStj|t_wr)r!stderrr#r$r%close)r'rrr mute_stderr(s   r)c@s eZdZdS)ArgcompleteExceptionN)__name__ __module__ __qualname__rrrrr*2sr*c sdurt|tj|ddd_tjdd_gfdd} z/}|jkr7ddddfWSj krNt d|dj j d ||WS|Wn%tyyt djd j j d j kru|jYStd wq%) NT)posix_ARGCOMPLETE_COMP_WORDBREAKScst|j}tjttfrjjvr|d7}|t|kr.td |d}|d|||d}}d}jdurKjj vrKj}|||j fS)NrzIn trailing whitespacer0) leninstreamtell isinstancestater bytes whitespacerappendquoteslast_wordbreak_pos)word point_in_wordprefixsuffixprequotelexerpointwordsrr split_word=s  zsplit_line..split_wordr;zsplit, lexer state: '{s}'sz#split (lexer stopped, state: '{s}')zcUnexpected internal state. Please report this bug at https://github.com/kislyuk/argcomplete/issues.)r1shlexwhitespace_splitr$environget wordbreaks get_tokeneofr2r3rformatr5r8 ValueErrortokenr*)linerBrDr;rr@r split_line5s.   rRcCs ||SN startswith) completionr=rrrdefault_validatorcs rWc@seZdZdZdddddedfddZdejdddddefddZd d Z d d Z d dZ ddZ ddZ eddZddZddZddZddZddZdd Zd!d"ZdS)#CompletionFindera Inherit from this class if you wish to override any of the stages below. Otherwise, use ``argcomplete.autocomplete()`` directly (it's a convenience instance of this class). It has the same signature as :meth:`CompletionFinder.__call__()`. NTFcCs^||_||_||_|durt}||_||_d|_i|_||_|dur*t j ddk}||_ dS)NF_ARGCOMPLETE_SUPPRESS_SPACE1) _parseralways_complete_optionsexcluderW validatorprint_suppressed completing_display_completionsdefault_completerr$rIrJ append_space)selfargument_parserr\r]r^r_rbrcrrr__init__ls zCompletionFinder.__init__c  Cs|j||||||| ddtjvrdSztddaWntjaY|durz5CompletionFinder._get_completions..zinvoking parser withr) namespacez parsed args:z exceptionzwhile parsing argsF)_patch_argument_parserargparse Namespacer`r rr)r[parse_known_args BaseExceptiontyper collect_completionsfilter_completionsquote_completions) rdrrrr:active_parsers parsed_argsaerrrrrys(" z!CompletionFinder._get_completionscsFg|_g|_|fdd|jtd|jtd|j|jS)a Since argparse doesn't support much introspection, we monkey-patch it to replace the parse_known_args method and all actions with hooks that tell us which action was last taken or about to be taken, and let us have the parser figure out which subparsers need to be activated (then recursively monkey-patch those). We save all active ArgumentParsers to extract all their possible option names later. csj|j|t|trdSd}trt|}t|t|jfi|_|j D]}t |dr0q(Gfddd|j}|j|_ |j |_ ||_q(dS)N(MonkeyPatchedIntrospectiveArgumentParser _orig_classcseZdZdfdd ZdS)zPCompletionFinder._patch_argument_parser..patch..IntrospectActionNcstd|td||||td|jtd|jjs'|j||||ddSt|jtjrFtd|j|d|j||||ddS|jtvr_|j sTj ||j||||ddSdS)NzAction stub called onz args:z orig class:z orig callable:) option_stringz:orig class is a subparsers action: patching and running itr) rr_orig_callabler` issubclassr_SubParsersAction_name_parser_map safe_actionsoption_stringsvisited_positionalsr8)rdparserrvaluesr completerpatchrrrs     zYCompletionFinder._patch_argument_parser..patch..IntrospectAction.__call__rS)r+r,r-rrrrrIntrospectActionsr)rr8rr4rr r6r __class___actionshasattrrrr)r classnameactionrrrrrs     z6CompletionFinder._patch_argument_parser..patchzActive parsers:zVisited positionals:)rrr[rrdrrrrs )  z'CompletionFinder._patch_argument_parsercsRdd}|D]}||j|j}|r|j|j|<qfdd|jD}|S)NcsB|s |r |r |SdS|dd}dfdd|DS)Nr0, c3s|] }|r|VqdSrSrTrxr=rr <szVCompletionFinder._get_subparser_completions..filter_aliases..)rUreplacesplitr{)metavardestr=rrrrfilter_aliases6szCCompletionFinder._get_subparser_completions..filter_aliasescsg|] }|r|qSrrT)rsubcmdrrrrCz?CompletionFinder._get_subparser_completions..)_get_subactionsrrhelprachoiceskeys)rdrrrrsubcmd_with_aliasesrrrr_get_subparser_completions5s  z+CompletionFinder._get_subparser_completionscsztdks |jdurfdd|jDSdd|jD}dd|jD}|jdkr0|r.|S|S|jdkr;|r9|S|SgS) NrTcs"g|] }t|rt|qSrr rUroptrrrrH"z5CompletionFinder._include_options..cSs g|] }t|dkrt|qSr1r rrrrrI cSs g|] }t|dkrt|qSrrrrrrrJrlongshort)r1r\r)rdrr long_opts short_optsrrr_include_optionsFs    z!CompletionFinder._include_optionscsn|jfdd|jDg}|jD]!}|jtjkr|jsq|||s&qt|tj s4|| |7}q|S)Ncs2g|]}|jrdfdd|jD|jgqS)rc3s&|]}t|rt|VqdSrSrrrrrrSs$zFCompletionFinder._get_option_completions...)rr{r)rrrrrrSs&z.) raupdaterrrSUPPRESSr__action_allowedr4rr)rdrroption_completionsrrrr_get_option_completionsQs    z(CompletionFinder._get_option_completionscCs(|j|gD] }||jvrdSqdSNFT)_action_conflictsrJ_seen_non_default_actions)rrconflict_actionrrrras  z CompletionFinder._action_allowedc stdjt|j|jdod|jvfdd|jD}|r:t|dks,Jdtd|dd i_g}nr>|Sd }|pD|jD]js`trPd }|s`tr`ts`td qEtd j t dd}|durj durt t jstj }n t t jsj}|rt|rLJfdd||dD} | r|| 7}t |tjrjfdd| Dn6jdd| Dn*tdtdD]!} || } | durn| rj| di|| qtd|qE|S)NzActive actions (L={l}): {a})rqrrcsg|] }t|r|qSr)rr) isoptionalrrrnrz.rzexpect at most 1 greedy actionzResetting completions becausezmust consume the next argumentFTSkippingzActivating completion forrcsg|] }|r|qSr)r^rc)rrdrrrs  )r=rrcsg|]}|jgqSr)rr) active_actionrrrscSsg|]}|dgqS)r0rrrrrrrzICompleter is not callable, trying the readline completer protocol insteadi'r0z Completions:)rrNr1active_actions prefix_charsrarrrrgetattrrr4rrrChoicesCompleterrbcallablerrangecompleter^r8) rdrnext_positionalrrrgreedy_actionscomplete_remaining_positionalsrcompletions_from_callableinext_completionr)rrrrdr_complete_active_optionjsj           z(CompletionFinder._complete_active_optioncCsg}|d||d}|d||js t|dkr(|d|jvr(||||7}|d||}|d|t|tjrD||||7}| |||||}|d||d|j |S) a0 Visits the active parsers and their actions, executes their completers or introspects them to collect their option strings. Returns the resulting completions as a list of strings. This method is exposed for overriding in subclasses; there is no need to use it directly. zall active parsers:zactive_parser:rzoptional options:znext_positional:zactive options:zdisplay completions:) r\r1rr_get_next_positionalr4rrrrra)rdrrrrr active_parserrrrrrs"         z$CompletionFinder.collect_completionscCsz|jd}|jd}|}|sdS||kr|dSd}tt|D] }|||kr,nq"|dt|kr;||dSdS)z> Get the next positional action if it exists. rNrr)rr_get_positional_actionsrr1)rdrlast_positionalall_positionalsrrrrrs    z%CompletionFinder._get_next_positionalcs<dd|D}|jdurt|_t|jfdd|DS)a Ensures collected completions are Unicode text, de-duplicates them, and excludes those specified by ``exclude``. Returns the filtered completions as an iterable. This method is exposed for overriding in subclasses; there is no need to use it directly. cSrr)r rrrrrrz7CompletionFinder.filter_completions..Ncs"g|] }|vr|s|qSr)addrseenrrrr)r]set)rdrrrrrs   z#CompletionFinder.filter_completionscsd}|dkrrfdd|D}|d7}n|dkr|d7}tjdd kr)d}n |d kr6d}d d|D}|D] fd d|D}q8|jrcd }t|dkrc|dd|vrc|dkrc|dd7<|S)a If the word under the cursor started with a quote (as indicated by a nonempty ``cword_prequote``), escapes occurrences of that quote character in the completions, and adds the quote to the beginning of each completion. Otherwise, escapes all characters that bash splits words on (``COMP_WORDBREAKS``), and removes portions of completions before the first colon if (``COMP_WORDBREAKS``) contains a colon. If there is only one completion, and it doesn't end with a **continuation character** (``/``, ``:``, or ``=``), adds a space after the completion. This method is exposed for overriding in subclasses; there is no need to use it directly. \r0csg|] }|ddqS)rNrr)r:rrr z6CompletionFinder.quote_completions..z();<>|&!`$* "'"z"`$!_ARGCOMPLETE_SHELLtcsh'cSsg|]}|ddqS)rz'\''rrrrrrscsg|] }|dqS)rrr)charrrrrz=/:rrrr)r$rIrJrcr1)rdrrr: special_charscontinuation_charsr)rr:rrs(  z"CompletionFinder.quote_completionscsn|dkr)t\}}}}|dtjd||||}fdd|D|_|t|jkr5|j|SdS)a Alternate entry point for using the argcomplete completer in a readline-based REPL. See also `rlcompleter `_. Usage: .. code-block:: python import argcomplete, argparse, readline parser = argparse.ArgumentParser() ... completer = argcomplete.CompletionFinder(parser) readline.set_completer_delims("") readline.set_completer(completer.rl_complete) readline.parse_and_bind("tab: complete") result = input("prompt> ") (Use ``raw_input`` instead of ``input`` on Python 2, or use `eight `_). rcs g|] }|tdqSrS)r1)rmatchrtextrrr<rz0CompletionFinder.rl_complete..N)rRinsertr!argvry _rl_matchesr1)rdrr5rrrfirst_colon_posmatchesrrr rl_complete%s zCompletionFinder.rl_completecCs|jS)a This function returns a mapping of option names to their help strings for displaying to the user Usage: .. code-block:: python def display_completions(substitution, matches, longest_match_length): _display_completions = argcomplete.autocomplete.get_display_completions() print("") if _display_completions: help_len = [len(x) for x in _display_completions.values() if x] if help_len: maxlen = max([len(x) for x in _display_completions]) print("\n".join("{0:{2}} -- {1}".format(k, v, maxlen) for k, v in sorted(_display_completions.items()))) else: print(" ".join(k for k in sorted(_display_completions))) else: print(" ".join(x for x in sorted(matches))) import readline print("cli /> {0}".format(readline.get_line_buffer()), end="") readline.redisplay() ... readline.set_completion_display_matches_hook(display_completions) )rarrrrget_display_completionsCsz(CompletionFinder.get_display_completions)r+r,r-__doc__rrfr$_exitrryrrrr staticmethodrrrrrrr r rrrrrXfs.  b<  D0 rXc@seZdZeddZdS)ExclusiveCompletionFindercCs<t||sdStjtjf}|j|vrdS||jvrdSdSr)rXrr _AppendAction_AppendConstActionrr)rrappend_classesrrrres    z)ExclusiveCompletionFinder._action_allowedN)r+r,r-r rrrrrrdsrzV Use this to access argcomplete. See :meth:`argcomplete.CompletionFinder.__call__()`. cGstdg|RdtidS)z Prints **args** to standard error when running completions. This will interrupt the user's command line interaction; use it to indicate an error condition that is preventing your completer from working.  rN)rrrrrrwarnvsrrS)5 __future__rrrrr$r!r contextlibr0rrrGcompatr r r r r r my_argparserrrrrIrr'rrBASH_FILE_COMPLETION_FALLBACKBASH_DIR_COMPLETION_FALLBACK _StoreAction_StoreConstAction_StoreTrueAction_StoreFalseActionrr _CountActionrcontextmanagerr&r) Exceptionr*rRrWobjectrXr autocompleter rrrrrsF      .