3 []+ @sddlZddlZddlZddlmZddlmZddlmZm Z m Z ddl m Z yddl mZWnek r|ddl ZYnXdZd Zd Zejd Zd d ZddZddZddZddZddZddZddZddZddZd d!ZdS)"N)echo)split_arg_string) MultiCommandOptionArgument)Choice)abc=a %(complete_func)s() { local IFS=$' ' COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \ COMP_CWORD=$COMP_CWORD \ %(autocomplete_var)s=complete $1 ) ) return 0 } %(complete_func)setup() { local COMPLETION_OPTIONS="" local BASH_VERSION_ARR=(${BASH_VERSION//./ }) # Only BASH version 4.4 and later have the nosort option. if [ ${BASH_VERSION_ARR[0]} -gt 4 ] || ([ ${BASH_VERSION_ARR[0]} -eq 4 ] && [ ${BASH_VERSION_ARR[1]} -ge 4 ]); then COMPLETION_OPTIONS="-o nosort" fi complete $COMPLETION_OPTIONS -F %(complete_func)s %(script_names)s } %(complete_func)setup a9 %(complete_func)s() { local -a completions local -a completions_with_descriptions local -a response response=("${(@f)$( env COMP_WORDS="${words[*]}" \ COMP_CWORD=$((CURRENT-1)) \ %(autocomplete_var)s="complete_zsh" \ %(script_names)s )}") for key descr in ${(kv)response}; do if [[ "$descr" == "_" ]]; then completions+=("$key") else completions_with_descriptions+=("$key":"$descr") fi done if [ -n "$completions_with_descriptions" ]; then _describe -V unsorted completions_with_descriptions -U -Q fi if [ -n "$completions" ]; then compadd -U -V unsorted -Q -a completions fi compstate[insert]="automenu" } compdef %(complete_func)s %(script_names)s z [^a-zA-Z0-9_]cCs@tjd|jdd}|dkr tnt}|d|||djdS)N-_zshz_%s_completion)Z complete_funcZ script_namesZautocomplete_var;)_invalid_ident_char_resubreplaceCOMPLETION_SCRIPT_ZSHCOMPLETION_SCRIPT_BASHstrip) prog_name complete_varshellZcf_namescriptr4/tmp/pip-build-w1as9q70/click/click/_bashcomplete.pyget_completion_scriptKs rcCs|j||dd}|j|j}x|rt|jtr|jjsv|jj||\}}}|dkrV|S|j|||dd}|j|j}qxD|r|jj||\}}}|dkr|S|j|||dddd}|j}qxW|}|j|j}qPqW|S)a Parse into a hierarchy of contexts. Contexts are connected through the parent variable. :param cli: command definition :param prog_name: the program that is running :param args: full list of args :return: the final context/command parsed T)resilient_parsingN)parentrF)rZallow_extra_argsallow_interspersed_argsr)Z make_contextprotected_argsargs isinstancecommandrchainZresolve_command)clirr!ctxcmd_namecmdZsub_ctxrrr resolve_ctxUs0     r)cCs|o|dddkS)z :param param_str: param_str to check :return: whether or not this is the start of an option declaration (i.e. starts "-" or "--") Nrr r)Z param_strrrrstart_of_optionzsr*cCsrt|tsdS|jrdSd}x>ttdd|DD]$\}}|d|jkrLPt|r4|}q4W|rn||jkrndSdS)aE :param all_args: the full original list of args supplied :param cmd_param: the current command paramter :return: whether or not the last option declaration (i.e. starts "-" or "--") is incomplete and corresponds to this cmd_param. In other words whether this cmd_param option can still accept values FNcSsg|]}|tkr|qSr) WORDBREAK).0argrrr sz(is_incomplete_option..rT)r"rZis_flag enumeratereversednargsr*opts)all_args cmd_paramZ last_optionindexZarg_strrrris_incomplete_options  r6cCs^t|tsdS||j}|dkr$dS|jdkr2dSt|tjrZ|jdkrZt||jkrZdSdS)aL :param current_params: the current params and values for this argument as already entered :param cmd_param: the current command parameter :return: whether or not the last argument is incomplete and corresponds to this cmd_param. In other words whether or not the this cmd_param argument can still accept values FNTr)r"rnamer1r Iterablelen)Zcurrent_paramsr4Zcurrent_param_valuesrrris_incomplete_arguments    r;csTg}t|jtr(fdd|jjD}n(|jdk rP|j||d}dd|D}|S)a :param ctx: context associated with the parsed command :param args: full list of args :param incomplete: the incomplete text to autocomplete :param cmd_param: command definition :return: all the possible user-specified completions for the param cs"g|]}t|jr|dfqS)N)str startswith)r,c) incompleterrr.sz,get_user_autocompletions..N)r&r!r?cSs"g|]}t|tr|n|dfqS)N)r"tuple)r,r>rrrr.s)r"typerchoicesZautocompletion)r&r!r?r4resultsZdynamic_completionsr)r?rget_user_autocompletionss   rDccs>x8|jj|D](}|j|r|jj||}|js|VqWdS)z :param ctx: context associated with the parsed command :starts_with: string that visible commands must start with. :return: all visible (not hidden) commands that start with starts_with. N)r#Z list_commandsr=Z get_commandhidden)r&Z starts_withr>r#rrr"get_visible_commands_starting_withs  rFcstjtr&|jddt|DxTjdk rzjtjtr(jjr(fddt|D}|jdd|Dq(WdS)NcSsg|]}|j|jfqSr)r8get_short_help_str)r,r>rrrr.sz.add_subcommand_completions..csg|]}|jjkr|qSr)r8r )r,r>)r&rrr.scSsg|]}|j|jfqSr)r8rG)r,r>rrrr.s)r"r#rextendrFrr$)r&r?Zcompletions_outZremaining_commandsr)r&radd_subcommand_completionss  rIcs4tj|t|||}|dkr"gStrTtkrTjt}j|d|dn tkr`dg}trxX|jjD]Lt t rvj rvfddj j D}|jfdd|DqvW|Sx(|jjD]trt|SqWx*|jjD]t|jrt|SqWt||t|S)z :param cli: command definition :param prog_name: the program that is running :param args: full list of args :param incomplete: the incomplete text to autocomplete :return: all the possible completions for the incomplete Nrr csg|]}|ksjr|qSr)multiple)r,Z param_opt)r3paramrrr.szget_choices..cs g|]}|jr|jfqSr)r=help)r,o)r?rLrrr.s)copydeepcopyr)r*r+ partitionappendr#paramsr"rrEr2Zsecondary_optsrHr6rDr;rIsorted)r%rr!r?r&Zpartition_incompleteZ completionsZ param_optsr)r3r?rLr get_choicess4        rUc Csttjd}ttjd}|d|}y ||}Wntk rLd}YnXx>t||||D],}t|d|r^t|dr|dndq^WdS)N COMP_WORDS COMP_CWORDrr rr T)rosenvironint IndexErrorrUr)r%rZinclude_descriptionscwordscwordr!r?itemrrr do_complete s    r_cCsR|jdr.|dkrdnd}tt|||dS|dks>|dkrNt|||dkSdS) NsourceZ source_zshrbashTZcompleteZ complete_zshF)r=rrr_)r%rrZcomplete_instrrrrr bashcompletes rb) rOrXreutilsrparserrcorerrrtypesr collectionsr ImportErrorr+rrcompilerrr)r*r6r;rDrFrIrUr_rbrrrrs2     % .