3 YiSF@sUddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZeejeejfeeeed ddZGdddZdZdZdZGdddZGdddeZGdddeZGdddeZeeedZejeej efd3ej eej!edddd Z"eej!ej ed!d"d#Z#e e e$d$d%d&Z%ee$d'd(d)Z&ej'ee e$d*d+d,Z(eejeejfeej'ee d-d.d/Z)e ej'eeej*ej+ee fefd0d1d2Z,dS)4N)gettext)Argument) BaseCommand)Context) MultiCommand)Option) Parameter)ParameterSource)split_arg_string)echo)clictx_args prog_name complete_var instructionreturnc Csf|jd\}}}t|}|dkr$dS|||||}|dkrJt|jdS|dkrbt|jdSdS)aPerform shell completion for the given CLI program. :param cli: Command being called. :param ctx_args: Extra arguments to pass to ``cli.make_context``. :param prog_name: Name of the executable in the shell. :param complete_var: Name of the environment variable that holds the completion instruction. :param instruction: Value of ``complete_var`` with the completion instruction and shell, in the form ``instruction_shell``. :return: Status code to exit with. _Nrsourcercomplete) partitionget_completion_classr rr) r rrrrshellrZcomp_clscompr7/tmp/pip-build-8om4l6ph/click/click/shell_completion.pyshell_completes  rc@sHeZdZdZdZdejeejeejddd d Z eejd d d Z dS)CompletionItema)Represents a completion value and metadata about the value. The default metadata is ``type`` to indicate special shell handling, and ``help`` if a shell supports showing a help string next to the value. Arbitrary parameters can be passed when creating the object, and accessed using ``item.attr``. If an attribute wasn't passed, accessing it returns ``None``. :param value: The completion suggestion. :param type: Tells the shell script to provide special completion support for the type. Click uses ``"dir"`` and ``"file"``. :param help: String shown next to the value if supported. :param kwargs: Arbitrary metadata. The built-in implementations don't use this, but custom type completions paired with custom shell support could use it. valuetypehelp_infoplainN)rrr kwargsrcKs||_||_||_||_dS)N)rrr r!)selfrrr r#rrr__init__LszCompletionItem.__init__)namercCs |jj|S)N)r!get)r$r&rrr __getattr__XszCompletionItem.__getattr__)rrr r!)r"N) __name__ __module__ __qualname____doc__ __slots__tAnystrOptionalr%r(rrrrr7sra%(complete_func)s() { local IFS=$'\n' local response response=$(env COMP_WORDS="${COMP_WORDS[*]}" COMP_CWORD=$COMP_CWORD %(complete_var)s=bash_complete $1) for completion in $response; do IFS=',' read type value <<< "$completion" if [[ $type == 'dir' ]]; then COMPREPLY=() compopt -o dirnames elif [[ $type == 'file' ]]; then COMPREPLY=() compopt -o default elif [[ $type == 'plain' ]]; then COMPREPLY+=($value) fi done return 0 } %(complete_func)s_setup() { complete -o nosort -F %(complete_func)s %(prog_name)s } %(complete_func)s_setup; a#compdef %(prog_name)s %(complete_func)s() { local -a completions local -a completions_with_descriptions local -a response (( ! $+commands[%(prog_name)s] )) && return 1 response=("${(@f)$(env COMP_WORDS="${words[*]}" COMP_CWORD=$((CURRENT-1)) %(complete_var)s=zsh_complete %(prog_name)s)}") for type key descr in ${response}; do if [[ "$type" == "plain" ]]; then if [[ "$descr" == "_" ]]; then completions+=("$key") else completions_with_descriptions+=("$key":"$descr") fi elif [[ "$type" == "dir" ]]; then _path_files -/ elif [[ "$type" == "file" ]]; then _path_files -f fi done if [ -n "$completions_with_descriptions" ]; then _describe -V unsorted completions_with_descriptions -U fi if [ -n "$completions" ]; then compadd -U -V unsorted -a completions fi } compdef %(complete_func)s %(prog_name)s; afunction %(complete_func)s; set -l response; for value in (env %(complete_var)s=fish_complete COMP_WORDS=(commandline -cp) COMP_CWORD=(commandline -t) %(prog_name)s); set response $response $value; end; for completion in $response; set -l metadata (string split "," $completion); if test $metadata[1] = "dir"; __fish_complete_directories $metadata[2]; else if test $metadata[1] = "file"; __fish_complete_path $metadata[2]; else if test $metadata[1] = "plain"; echo $metadata[2]; end; end; end; complete --no-files --command %(prog_name)s --arguments "(%(complete_func)s)"; c@seZdZUdZejeejee ej eej feeddddZ e edddZej eej fdd d Zedd d Zejejeefdd dZejeeejedddZeedddZedddZdS) ShellCompleteaBase class for providing shell completion support. A subclass for a given shell will override attributes and methods to implement the completion instructions (``source`` and ``complete``). :param cli: Command being called. :param prog_name: Name of the executable in the shell. :param complete_var: Name of the environment variable that holds the completion instruction. .. versionadded:: 8.0 N)r rrrrcCs||_||_||_||_dS)N)r rrr)r$r rrrrrrr%szShellComplete.__init__)rcCs(tjdd|jjddtj}d|dS)zQThe name of the shell function defined by the completion script. z\W*-rZ _completion)resubrreplaceASCII)r$ safe_namerrr func_nameszShellComplete.func_namecCs|j|j|jdS)zVars for formatting :attr:`source_template`. By default this provides ``complete_func``, ``complete_var``, and ``prog_name``. )Z complete_funcrr)r:rr)r$rrr source_varsszShellComplete.source_varscCs|j|jS)zProduce the shell script that defines the completion function. By default this ``%``-style formats :attr:`source_template` with the dict returned by :meth:`source_vars`. )source_templater;)r$rrrrszShellComplete.sourcecCstdS)zUse the env vars defined by the shell script to return a tuple of ``args, incomplete``. This must be implemented by subclasses. N)NotImplementedError)r$rrrget_completion_argssz!ShellComplete.get_completion_args)args incompletercCs0t|j|j|j|}t|||\}}|j||S)aTDetermine the context and last complete command or parameter from the complete args. Call that object's ``shell_complete`` method to get the completions for the incomplete value. :param args: List of complete args before the incomplete value. :param incomplete: Value being completed. May be empty. )_resolve_contextr rr_resolve_incompleter)r$r?r@ctxobjrrrget_completionss zShellComplete.get_completions)itemrcCstdS)zFormat a completion item into the form recognized by the shell script. This must be implemented by subclasses. :param item: Completion item to format. N)r=)r$rFrrrformat_completionszShellComplete.format_completioncs4j\}}j||}fdd|D}dj|S)zProduce the completion data to send back to the shell. By default this calls :meth:`get_completion_args`, gets the completions, then calls :meth:`format_completion` for each completion. csg|]}j|qSr)rG).0rF)r$rr $sz*ShellComplete.complete.. )r>rEjoin)r$r?r@Z completionsoutr)r$rrs  zShellComplete.complete)r)r*r+r,r.ZClassVarr0r&r<rDictr/r%propertyr:r;rTupleListr>rrErGrrrrrr2s"      r2csjeZdZdZdZeZddddZedfdd Z e j e j eefdd d Z eed d d ZZS) BashCompletezShell completion for Bash.bashN)rcCszddl}|jdddg|jd}tjd|jj}|dk rj|j\}}|dks\|dkrv|dkrvtt dn tt d dS) NrrRz-czecho ${BASH_VERSION})stdoutz^(\d+)\.(\d+)\.\d+4zCShell completion is not supported for Bash versions older than 4.4.z@Couldn't detect Bash version, shell completion is not supported.) subprocessrunPIPEr5searchrSdecodegroups RuntimeErrorr)r$rUoutputmatchmajorminorrrr_check_version.s  zBashComplete._check_versioncs|jtjS)N)r`superr)r$) __class__rrrEszBashComplete.sourcec CsVttjd}ttjd}|d|}y ||}Wntk rLd}YnX||fS)N COMP_WORDS COMP_CWORDrr3)r osenvironint IndexError)r$cwordscwordr?r@rrrr>Is   z BashComplete.get_completion_args)rFrcCs|jd|jS)N,)rr)r$rFrrrrGUszBashComplete.format_completion)r)r*r+r,r& _SOURCE_BASHr<r`r0rr.rOrPr>rrG __classcell__rr)rbrrQ(s rQc@sFeZdZdZdZeZejej e e fdddZ e e dddZ d S) ZshCompletezShell completion for Zsh.zsh)rc CsVttjd}ttjd}|d|}y ||}Wntk rLd}YnX||fS)Nrcrdrr3)r rerfrgrh)r$rirjr?r@rrrr>_s   zZshComplete.get_completion_args)rFrcCs$|jd|jd|jr|jndS)NrJr)rrr )r$rFrrrrGkszZshComplete.format_completionN)r)r*r+r,r& _SOURCE_ZSHr<r.rOrPr0r>rrGrrrrrnYs  rnc@sFeZdZdZdZeZejej e e fdddZ e e dddZ d S) FishCompletezShell completion for Fish.fish)rcCsHttjd}tjd}|dd}|r@|r@|d|kr@|j||fS)Nrcrdr)r rerfpop)r$rir@r?rrrr>us   z FishComplete.get_completion_args)rFrcCs2|jr |jd|jd|jS|jd|jS)Nrk )r rr)r$rFrrrrGszFishComplete.format_completionN)r)r*r+r,r& _SOURCE_FISHr<r.rOrPr0r>rrGrrrrrqos  rq)rRrrro)clsr&rcCs|dkr|j}|t|<dS)amRegister a :class:`ShellComplete` subclass under the given name. The name will be provided by the completion instruction environment variable during completion. :param cls: The completion class that will handle completion for the shell. :param name: Name to register the class under. Defaults to the class's ``name`` attribute. N)r&_available_shells)rwr&rrradd_completion_classs ry)rrcCs tj|S)zLook up a registered :class:`ShellComplete` subclass by the name provided by the completion instruction environment variable. If the name isn't registered, returns ``None``. :param shell: Name the class is registered under. )rxr')rrrrrsr)rCparamrcCsjt|tsdS|jdk st|j|j}|jdkph|j|jtjk ph|jdkoht|t t foht ||jkS)zDetermine if the given parameter is an argument that can still accept values. :param ctx: Invocation context for the command represented by the parsed complete args. :param param: Argument object being checked. FNrrs) isinstancerr&AssertionErrorparamsnargsZget_parameter_sourcer Z COMMANDLINEtuplelistlen)rCrzrrrr_is_incomplete_arguments    r)rrcCs"|sdS|d}|j o |dkS)z5Check if the value looks like the start of an option.Fr/)isalnum)rcrrr_start_of_optionsr)r?rzrcCsdt|tsdS|jrdSd}x4tt|D]$\}}|d|jkrBPt|r*|}q*W|dk ob||jkS)zDetermine if the given parameter is an option that needs a value. :param args: List of complete args before the incomplete value. :param param: Option object being checked. FNr)r{rZis_flag enumeratereversedr~ropts)r?rzZ last_optionindexargrrr_is_incomplete_options r)r rrr?rc Csd|d<|j||jf|}|j|j}x|r|j}t|tr|js|j||\}}}|dkrb|S|j|||dd}|j|j}qxB|r|j||\}}}|dkr|S|j|||dddd}|j}qW|}|j|j}q*Pq*W|S)a`Produce the context hierarchy starting with the command and traversing the complete arguments. This only follows the commands, it doesn't trigger input prompts or callbacks. :param cli: Command being called. :param prog_name: Name of the executable in the shell. :param args: List of complete args before the incomplete value. Tresilient_parsingN)parentrF)rZallow_extra_argsallow_interspersed_argsr) Z make_contextcopyZprotected_argsr?commandr{rchainZresolve_command) r rrr?rCrr&cmdZsub_ctxrrrrAs8    rA)rCr?r@rcCs|dkrd}n*d|kr8t|r8|jd\}}}|j|d|krRt|rR|j|fS|jj|}x|D]}t||rd||fSqdWx|D]}t||r||fSqW|j|fS)ahFind the Click object that will handle the completion of the incomplete value. Return the object and the incomplete value. :param ctx: Invocation context for the command represented by the parsed complete args. :param args: List of complete args before the incomplete value. :param incomplete: Value being completed. May be empty. =r3z--)rrappendr get_paramsrr)rCr?r@r&rr}rzrrrrBs         rB)N)-rer5typingr.rrcorerrrrrr r parserr utilsr rMr0r/rgrrrlrprvr2rQrnrqrxTyper1ryrboolrrrPrrArOUnionrBrrrrsP           D&j1  3