o Ďix>@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z d dd Z d!d d Z d!d d Z d"ddZ d"ddZddZd#ddZddZd$ddZddZddZdS)%z?Provides tab completion functionality for CLIs built with Fire.)absolute_import)division)print_functionN) inspectutilsbashcCs(|dkr t|t||St|t||S)Nfish) _FishScript _Commands _BashScript)name componentdefault_optionsshellrK/home/jeff/fluffinator/venv/lib/python3.10/site-packages/fire/completion.pyScriptsrc s|pt}t||\}}}d}d}d}ddfdd} g} t} | | t|} | t|} | D] } | | jdt|| || d } | |j| | d q>d | }|j|d }|j|d| d d dd ddd|dS)aXReturns a Bash script registering a completion function for the commands. Args: name: The first token in the commands, also the name of the command. commands: A list of all possible commands that tab completion can complete to. Each command is a list or tuple of the string tokens that make up that command. default_options: A dict of options that can be used with any command. Use this if there are flags that can always be appended to a command. Returns: A string which is the Bash script. Source the bash script to enable tab completion in Bash. a# bash completion support for {name} # DO NOT EDIT. # This script is autogenerated by fire/completion.py. _complete-{identifier}() {{ local cur prev opts lastcommand COMPREPLY=() prev="${{COMP_WORDS[COMP_CWORD-1]}}" cur="${{COMP_WORDS[COMP_CWORD]}}" lastcommand=$(get_lastcommand) opts="{default_options}" GLOBAL_OPTIONS="{global_options}" {checks} COMPREPLY=( $(compgen -W "${{opts}}" -- ${{cur}}) ) return 0 }} get_lastcommand() {{ local lastcommand i lastcommand= for ((i=0; i < ${{#COMP_WORDS[@]}}; ++i)); do if [[ ${{COMP_WORDS[i]}} != -* ]] && [[ -n ${{COMP_WORDS[i]}} ]] && [[ ${{COMP_WORDS[i]}} != $cur ]]; then lastcommand=${{COMP_WORDS[i]}} fi done echo $lastcommand }} filter_options() {{ local opts opts="" for opt in "$@" do if ! option_already_entered $opt; then opts="$opts $opt" fi done echo $opts }} option_already_entered() {{ local opt for opt in ${{COMP_WORDS[@]:0:$COMP_CWORD}} do if [ $1 == $opt ]; then return 0 fi done return 1 }} is_prev_global() {{ local opt for opt in $GLOBAL_OPTIONS do if [ $opt == $prev ]; then return 0 fi done return 1 }} complete -F _complete-{identifier} {command} z; case "${{lastcommand}}" in {lastcommand_checks} esaczQ {command}) {opts_assignment} opts=$(filter_options $opts) ;;z if is_prev_global; then opts="${{GLOBAL_OPTIONS}}" else opts="{options} ${{GLOBAL_OPTIONS}}" fiz, opts="{options} ${{GLOBAL_OPTIONS}}" cs|krSSNr)commandr %opts_assignment_main_command_template#opts_assignment_subcommand_templaterr_GetOptsAssignmentTemplatesz/_BashScript.._GetOptsAssignmentTemplate )options)ropts_assignment )lastcommand_checks/.,)r rchecksr identifierglobal_options) set_GetMapsaddunionkeysformatjoinsortedappendreplace)r commandsr r# options_mapsubcommands_mapbash_completion_template check_wrapperlastcommand_check_templaterlines commands_setrrrr!rrrr "sP  M  r c Cs|pt}t|||\}}}d}d}d}d} t|t|D]3} || D] } ||j|| | d7}q)|| |D]} | |k} ||j|| | rL| nd| dd7}q=q#|jd d d |Dd S) aXReturns a Fish script registering a completion function for the commands. Args: name: The first token in the commands, also the name of the command. commands: A list of all possible commands that tab completion can complete to. Each command is a list or tuple of the string tokens that make up that command. default_options: A dict of options that can be used with any command. Use this if there are flags that can always be appended to a command. Returns: A string which is the Fish script. Source the fish script to enable tab completion in Fish. aLfunction __fish_using_command set cmd (commandline -opc) for i in (seq (count $cmd) 1) switch $cmd[$i] case "-*" case "*" if [ $cmd[$i] = $argv[1] ] return 0 else return 1 end end end return 1 end function __option_entered_check set cmd (commandline -opc) for i in (seq (count $cmd)) switch $cmd[$i] case "-*" if [ $cmd[$i] = $argv[1] ] return 1 end end end return 0 end function __is_prev_global set cmd (commandline -opc) set global_options {global_options} set prev (count $cmd) for opt in $global_options if [ "--$opt" = $cmd[$prev] ] echo $prev return 0 end end return 1 end zJcomplete -c {name} -n '__fish_using_command {command}' -f -a {subcommand} z}complete -c {name} -n '__fish_using_command {command};{prev_global_check} and __option_entered_check --{option}' -l {option} z and __is_prev_global;)r r subcommandr--)r rprev_global_checkoptionrcss|] }d|dVqdS)"Nr).0r9rrr sz_FishScript..)r#)r$r%r(r'r)lstripr*)r r.r r#r/r0 fish_sourcesubcommand_template flag_templater8rr6r9 check_neededrrrrs6  -     rFcCst|tr |dr dS|rdS|tus|tus|turdSt|ttr'dSg}t|r4||vr4dSt |rb|durDt |pCi}| |}|rb|j dvrRdSttdtd}t|j|rbdSt|trm|d SdS)aReturns whether a member should be included in auto-completion or help. Determines whether a member of an object with the specified name should be included in auto-completion or help text(both usage and detailed help). If the member name starts with '__', it will always be excluded. If it starts with only one '_', it will be included for all non-string types. If verbose is True, the members, including the private members, are included. When not in verbose mode, some modules and functions are excluded as well. Args: component: The component containing the member. name: The name of the member. member: The member itself. class_attrs: (optional) If component is a class, provide this as: inspectutils.GetClassAttrsDict(component). If not provided, it will be computed. verbose: Whether to include private members. Returns A boolean value indicating whether the member should be included. __FTN)methodproperty _tuplegetter_) isinstancestr startswithrrrtypeinspectismoduleisclassrGetClassAttrsDictgetkindgetattr collectionsobject)r r member class_attrsverbosemodules_to_hide class_attr tuplegetterrrr MemberVisibles4      rZcsFttr }nt}durtfdd|DS)aReturns a list of the members of the given component. If verbose is True, then members starting with _ (normally ignored) are included. Args: component: The component whose members to list. class_attrs: (optional) If component is a class, you may provide this as: inspectutils.GetClassAttrsDict(component). If not provided, it will be computed. If provided, this determines how class members will be treated for visibility. In particular, methods are generally hidden for non-instantiated classes, but if you wish them to be shown (e.g. for completion scripts) then pass in a different class_attr for them. verbose: Whether to include private members. Returns: A list of tuples (member_name, member) of all members of the component. Nc s*g|]\}}t||dr||fqS)rUrV)rZ)r; member_namerTrUr rVrr os z"VisibleMembers..)rGdictitemsrK getmembersrrN)r rUrVmembersrr]rVisibleMembersUs    rccCs.g}|D]}|dd}|d|q|S)zTakes a list of fn args and returns a list of the fn's completion strings. Args: fn_args: A list of the args accepted by a function. Returns: A list of possible completion strings for that function. rF-r7)r-r,)fn_args completionsargrrr_CompletionsFromArgsvs  rhcCsvt|s t|rt|}t|j|jSt|t t fr)ddt t |DSt |r0gSddt||dDS)aGives possible Fire command completions for the component. A completion is a string that can be appended to a command to continue that command. These are used for TAB-completions in Bash for Fire CLIs. Args: component: The component whose completions to list. verbose: Whether to include all completions, even private members. Returns: A list of completions for a command that would so far return the component. cSsg|]}t|qSr)rH)r;indexrrrr^szCompletions..cSsg|]\}}t|qSr)_FormatForCommand)r;r\rFrrrr^srV)rK isroutinerMrGetFullArgSpecrhargs kwonlyargsrGtuplelistrangelen isgeneratorrc)r rVspecrrr Completionss   rvcCs,t|ts t|}|dr|S|ddS)aReplaces underscores with hyphens, unless the token starts with a token. This is because we typically prefer hyphens to underscores at the command line, but we reserve hyphens at the start of a token for flags. This becomes relevant when --verbose is activated, so that things like __str__ don't get transformed into --str--, which would get confused for a flag. Args: token: The token to transform. Returns: The transformed token. rFrd)rGrHrIr-)tokenrrrrjs   rjccst|s t|rt|ddD]}|fVqt|rdS|dkr%dSt|iddD]\}}t|}|fVt||dD]}|f|Vq?q,dS)a Yields tuples representing commands. To use the command from Python, insert '.' between each element of the tuple. To use the command from the command line, insert ' ' between each element of the tuple. Args: component: The component considered to be the root of the yielded commands. depth: The maximum depth with which to traverse the member DAG for commands. Yields: Tuples, each tuple representing one possible command for this CLI. Only traverses the member DAG up to a depth of depth. FrkNr[)rKrlrMrvrcrjr )r depth completionr\rTrrrrr s"  r cCs |dS)Nrd)rI)rgrrr _IsOptions r|c st}tfdd}tt}|D]G}t|dkr5t|dr+||dq|||dq|r\|d}t|d}t|rH|} n|} | ||| |dd|q|||fS) aReturns sets of subcommands and options for each command. Args: name: The first token in the commands, also the name of the command. commands: A list of all possible commands that tab completion can complete to. Each command is a list or tuple of the string tokens that make up that command. default_options: A dict of options that can be used with any command. Use this if there are flags that can always be appended to a command. Returns: global_options: A set of all options of the first token of the command. subcommands_map: A dict storing set of subcommands for each command/subcommand. options_map: A dict storing set of options for each subcommand. cs tSr)copyrr rrs z_GetMaps..ryrrFrd) r}rR defaultdictr$rsr|r&rjr-) r r.r r#r/r0rr6rgargs_maprr~rr%s$      r%)Nrr)NF)F)rx)__doc__ __future__rrrrRr}rKfirerrr rrZrcrhrvrjr r|r%rrrrs(        ] 9!  #