o [h0@sddlZddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z zdd l mZWn eyEddl ZYnwd Zd Zd Zd ZeeedZedZddZddZddZddZddZddZddZddZd d!Zd"d#Z d$d%Z!d&d'Z"dS)(N)Argument) MultiCommand)Option)split_arg_string)Choice)echo)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 a #compdef %(script_names)s %(complete_func)s() { local -a completions local -a completions_with_descriptions local -a response (( ! $+commands[%(script_names)s] )) && return 1 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 fi if [ -n "$completions" ]; then compadd -U -V unsorted -a completions fi compstate[insert]="automenu" } compdef %(complete_func)s %(script_names)s zcomplete --no-files --command %(script_names)s --arguments "(env %(autocomplete_var)s=complete_fish COMP_WORDS=(commandline -cp) COMP_CWORD=(commandline -t) %(script_names)s)")bashzshfishz [^a-zA-Z0-9_]cCs>td|dd}t|t}|d|||ddS)N-_z_{}_completion)Z complete_funcZ script_namesZautocomplete_var;)_invalid_ident_char_resubreplace_completion_scriptsgetCOMPLETION_SCRIPT_BASHformatstrip) prog_name complete_varshellZcf_namescriptrz/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/click/_bashcomplete.pyget_completion_script_s r cCs|j||dd}|j|j}|rjt|jtre|jjs:|j||\}}}|dur*|S|j|||dd}|j|j}n.|r\|j||\}}}|durL|S|j|||dddd}|j}|s<|}|j|j}n |S|s|S)aParse 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)parentr!F)r"Zallow_extra_argsallow_interspersed_argsr!)Z make_contextprotected_argsargs isinstancecommandrchainZresolve_command)clirr%ctxcmd_namecmdZsub_ctxrrr resolve_ctxls@   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 "--") Nrrr)Z param_strrrrstart_of_optionsr.cCspt|tsdS|jr dSd}ttdd|DD]\}}|d|jkr&nt|r,|}q|r6||jvr6dSdS)aU :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  r:cCs^t|tsdS||j}|durdS|jdkrdSt|tjr-|jdkr-t||jkr-dSdS)a` :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&rnamer5r Iterablelen)Zcurrent_paramsr8Zcurrent_param_valuesrrris_incomplete_arguments     r?csVg}t|jtrfdd|jjD}|S|jdur)|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|r|dfqSN)str startswithr0c incompleterrr2s z,get_user_autocompletions..N)r*r%rFcSs"g|] }t|tr |n|dfqSr@)r&tuplerCrrrr2s)r&typerchoicesautocompletion)r*r%rFr8resultsZdynamic_completionsrrErget_user_autocompletionss   rLccs<|j|D]}||r|j||}|js|VqdS)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_commandsrBZ get_commandhidden)r*Z starts_withrDr'rrr"get_visible_commands_starting_withs rNcstjtr|ddt|DjdurBjtjtr;jjr;fddt|D}|dd|DjdusdSdS)NcSg|] }|j|fqSrr<Zget_short_help_strrCrrrr2s z.add_subcommand_completions..csg|] }|jjvr|qSr)r<r$rCr*rrr2s  cSrOrrPrCrrrr2 s)r&r'rextendrNr"r()r*rFZcompletions_outZremaining_commandsrrQradd_subcommand_completionss    rSc s:t|t|||}|durgSdv}tr.tvr.t}|d|dntkr4dg}|shtrh|jjD]%t t rej sefddj j D}|fdd|Dq@|S|jjD]tr|t|Sql|jjD]t|jrt|Sqt||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 Nz--rrcsg|] }|vs jr|qSr)Zmultiple)r0Z param_opt)r7paramrrr2.s zget_choices..cs g|] }|r|jfqSr)rBhelp)r0o)rFrUrrr24s )copydeepcopyr-r.r/ partitionappendr'paramsr&rrMr6Zsecondary_optsrRr:rLr?rSsorted) r)rr%rFr*Zhas_double_dashZpartition_incompleteZ completionsZ param_optsr)r7rFrUr get_choicessB             r^cCsttjd}ttjd}|d|}z||}Wn ty%d}Ynwt||||D]}t|d|rCt|dr@|dndq-dS)N COMP_WORDS COMP_CWORDrrrrT)rosenvironint IndexErrorr^r)r)rZinclude_descriptionscwordscwordr%rFitemrrr do_completeEs    rhcCslttjd}tjd}|dd}t||||D]}|dr-tdj|d|ddqt|dqdS)Nr_r`rz {arg} {desc}r)r1ZdescT)rrarbr^rr)r)rrerFr%rgrrrdo_complete_fishYs  ricCstd|vr |dd\}}n|}d}|dkrtt|||dS|dkr8|dkr,t||S|dvr8t|||d kSd S) Nrrr sourceTZcompleter >r r r F)splitrr rirh)r)rrZcomplete_instrr'rrrr bashcompletegs rl)#rXrarecorerrrparserrtypesrutilsr collectionsr ImportErrorr/rZCOMPLETION_SCRIPT_ZSHZCOMPLETION_SCRIPT_FISHrcompilerr r-r.r:r?rLrNrSr^rhrirlrrrrsH        $  +  5