B LJbbF@sUddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZeejeejfeeeed ddZGdddZdZdZdZGdddZGdddeZGdddeZGdddeZeeedZejeej efe!d<d4ej eej"eddd d!Z#eej"ej ed"d#d$Z$e e e%d%d&d'Z&e ee%d(d)d*Z'e ej(ee e%d+d,d-Z)eejeejfeej(ee d.d/d0Z*e ej(eeej+ej,ee fefd1d2d3Z-dS)5N)gettext)Argument) BaseCommand)Context) MultiCommand)Option) Parameter)ParameterSource)split_arg_string)echo)clictx_args prog_name complete_var instructionreturnc Csf|d\}}}t|}|dkr$dS|||||}|dkrJt|dS|dkrbt|dSdS)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_clscomprj/var/www/downstreamdata.science/rtclock/rtclock-venv/lib/python3.7/site-packages/click/shell_completion.pyshell_completes  rc@sHeZdZdZdZd ejeejeejddddZ eejdd 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 |j|S)N)r!get)r$r&rrr __getattr__XszCompletionItem.__getattr__)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@seZdZUdZejeed<ejeed<eej eej feeddddZ e edd d Z ej eej fdd d Zedd dZejejeefdddZejeeejedddZeedddZedddZdS) 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 r&source_templateN)r rrrrcCs||_||_||_||_dS)N)r rrr)r$r rrrrrrr%szShellComplete.__init__)rcCs(tdd|jddtj}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|S)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`. )r3r<)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|||\}}|||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_completioncs4\}}||}fdd|D}d|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|]}|qSr)rG).0rF)r$rr $sz*ShellComplete.complete.. )r>rEjoin)r$r?r@Z completionsoutr)r$rrs  zShellComplete.complete)r)r*r+r,r.ClassVarr0__annotations__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}td|j}|dk rj|\}}|dks\|dkrv|dkrvtt dn tt d dS) NrrTz-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.) subprocessrunPIPEr6searchrUdecodegroups RuntimeErrorr)r$rWoutputmatchmajorminorrrr_check_version.s  zBashComplete._check_versioncs|tS)N)rbsuperr)r$) __class__rrrEszBashComplete.sourcecCsVttjd}ttjd}|d|}y ||}Wntk rLd}YnX||fS)N COMP_WORDS COMP_CWORDrr4)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_BASHr3rbr0rr.rQrRr>rrG __classcell__rr)rdrrS(s rSc@sFeZdZdZdZeZejej e e fdddZ e e dddZ d S) ZshCompletezShell completion for Zsh.zsh)rcCsVttjd}ttjd}|d|}y ||}Wntk rLd}YnX||fS)Nrerfrr4)r rgrhrirj)r$rkrlr?r@rrrr>_s   zZshComplete.get_completion_args)rFrcCs$|jd|jd|jr|jndS)NrJr)rrr )r$rFrrrrGkszZshComplete.format_completionN)r)r*r+r,r& _SOURCE_ZSHr3r.rQrRr0r>rrGrrrrrpYs  rpc@sFeZdZdZdZeZejej e e fdddZ e e dddZ d S) FishCompletezShell completion for Fish.fish)rcCsHttjd}tjd}|dd}|r@|r@|d|kr@|||fS)Nrerfr)r rgrhpop)r$rkr@r?rrrr>us   z FishComplete.get_completion_args)rFrcCs2|jr |jd|jd|jS|jd|jS)Nrm )r rr)r$rFrrrrGszFishComplete.format_completionN)r)r*r+r,r& _SOURCE_FISHr3r.rQrRr0r>rrGrrrrrsos  rs)rTrtrq_available_shells)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&ry)rzr&rrradd_completion_classs r{)rrcCs t|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. )ryr')rrrrrsr)rCparamrcCsjt|tsdS|jdk st|j|j}|jdkph||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. FNrur) isinstancerr&AssertionErrorparamsnargsZget_parameter_sourcer Z COMMANDLINEtuplelistlen)rCr|rrrr_is_incomplete_arguments    r)rCrrcCs|sdS|d}||jkS)z5Check if the value looks like the start of an option.Fr)Z _opt_prefixes)rCrcrrr_start_of_optionsr)rCr?r|rcCslt|tsdS|js|jrdSd}x6tt|D]&\}}|d|jkrHPt||r0|}q0W|dk oj||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_flagcount enumeratereversedrropts)rCr?r|Z last_optionindexargrrr_is_incomplete_options   r)r rrr?rc Csd|d<|j||f|}|j|j}x|r|j}t|tr|js|||\}}}|dkrb|S|j|||dd}|j|j}qxB|r|||\}}}|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|kr:t||r:|d\}}}||d|krVt||rV|j|fS|j|}x |D]}t|||rh||fSqhWx|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. =r4z--)rrappendr get_paramsrr)rCr?r@r&rrr|rrrrBs         rB)N).rgr6typingr.rrcorerrrrrr r parserr utilsr rOr0r/rirrrnrrrxr2rSrprsryTyperNr1r{rboolrrrRrrArQUnionrBrrrrsP           D&j1   3