lddZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZddlZd Zed z ZdZdZee_dZdZdZGddeZ GddeZ!GddeZ"dZ#GddeZ$GddejJZ&y)z(The gcloud interactive shell completion.)absolute_import)division)unicode_literalsN)parser_completer)parser)generate_cli_trees) module_util) console_attr) completionz://c&|j}|js|jjddyt |jj dk(r)|j |jj dy|jy)zFcompletion.generate_completions override that auto selects singletons.TF)insert_common_part select_firstr rN)current_buffercomplete_stateclistart_completionlencurrent_completionsapply_completion complete_next)eventbs 7lib/googlecloudsdk/command_lib/interactive/completer.py_GenerateCompletionsr,ss !   II$UK 1   / /0A5q'';;A>?OOc"tj}|jd|D]E}|jdj|jxsd|j j G|jd|jS)z0Pretty prints args into a string and returns it.[z({},{})z""])ioStringIOwriteformatvalue token_typenamegetvalue)argsbufargs r _PrettyArgsr,>se #))C. cIIisyy0D#..2E2EFG))C. rct}|j|}|dk\r0|ttz}|d|dz g||djdzS|jdS)zFReturns the list of component names in path, treating foo:// as a dir.rNr /)_URI_SEPfindrsplit)pathurisep uri_indexns r_Splitr6Hs^ &ii)!^CM!A !A#J<$qr(..- -- Crc<djt|ddS)z,Returns the dirname of path, '' if it's '.'.r.Nr )joinr6)r2s r_Dirnamer9Rs &,s# $$rc"eZdZdZdZdZdZy)CacheArgzA completion cache arg.c.||_||_i|_yN)prefix completionsdirs)selfr>r?s r__init__zCacheArg.__init__ZsDK"DDIrc|jduSr=)r?rAs rIsValidzCacheArg.IsValid_s   4 ''rc6t|_d|_i|_yr=)_INVALID_ARG_COMMAND_COUNT command_countr?r@rDs r InvalidatezCacheArg.Invalidatebs3DDDIrN)__name__ __module__ __qualname____doc__rBrErIrrr;r;Ws (rr;c.eZdZdZdZdZdZdZdZy)CompletionCachea-A per-arg cache of completions for the command line under construction. Since we have no details on the compeleted values this cache is only for the current command line. This means that async activities by other commands (creating files, instances, resources) may not be seen until the current command under construction is executed. Attributes: args: The list of CacheArg args holding the completion state for each arg. completer: The completer object. command_count: The completer.cli.command_count value for the current cache. c6g|_||_t|_yr=)r) completer_INVALID_COMMAND_COUNTrH)rArRs rrBzCompletionCache.__init__vsDIDN/Drc\|j|jjjk(Sr=)rHrRrrDs rrEzCompletionCache.IsValid{s#   !3!3!A!A AArc|j|jsy||jj|j|jS)z?Returns True if args[index] matches the cache prefix for index.T)r)rEr% startswithr>)rAr)indexs rArgMatchzCompletionCache.ArgMatch~sD 99U  # # %  ;   ' ' %(8(?(? @@rcj|r|jsyt|t|jkDryt|dz }t|D]}|j ||ry|j|jsy||j }|j dr|dd}|jjjj||j|jj|d\}}|sy||j|_ ||j|_n||j|jvr|jjjjt||j|jjt|d\}}|r(||j|_ ||j|_|j ||sy|j|jDcgc]}|j!|s|c}Scc}w)z@Returns the cached completions for the last arg in args or None.Nr r.r NN)rErr)rangerXr%endswithrRdebugdirtextr@getr>r?r9rV) rAr)last_arg_indexiaparentr>r?cs rLookupzCompletionCache.Lookups t||~  4y3tyy>! Y]N > " ]]4 ## 99^ $ , , .  ^""Azz#"vf nn##F+ IIn5::>>v?KMfk )/dii&.9dii+ dii',, , nn##HQK0 IIn5::>>x{?KMfk +1 .!(0; .!- ==~ . yy0<< Pr\r@r?r6r8six iteritems) rAr)r?rarbrcr@compmarkpartsdremarkeds rUpdatezCompletionCache.Updates++99DY]N > " c$))n  $q'--67}}T1% ! ! # ^""ATYY' iix;/0 IIn % - - / LL>299 : 3 Cyy(--ii'..ii'3305$))N # ( ( ,34[1A .!&&q"v.sk!nSb11cl d$ == cr$$$t   )t )s5z"AhhuRay!! d]DG%Q-($ ]]4 CKD a GNN4 #$--%$!Q ??4 C-&,2AI .!&&q) & *+dii&.9dii+>A%s499~ 6 iil7rN) rJrKrLrMrBrErXrfrqrNrrrPrPhs% 0 BA3QjA rrPc@eZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) SpinneraA Spinner to show when completer takes too long to respond. Some completer calls take too long, specially those that fetch remote resources. An instance of this class can be used as a context manager wrapping slow completers to get spinmarks while the completer fetches. Attributes: _done_loading: Boolean flag indicating whether ticker thread is working. _set_spinner: Function reference to InteractiveCliCompleter's spinner setter. _spin_marks: List of unicode spinmarks to be cycled while loading. _ticker: Thread instance that handles displaying the spinner. _ticker_index: Integer specifying the last iteration index in _spin_marks. _TICKER_INTERVAL: Float specifying time between ticker rotation in milliseconds. _ticker_length: Integer spcifying length of _spin_marks. _TICKER_WAIT: Float specifying the wait time before ticking in milliseconds. _TICKER_WAIT_CHECK_INTERVAL: Float specifying interval time to break wait in milliseconds. d cd|_tjjj|_d|_d|_t|j |_ ||_ y)NFr) _done_loadingr GetConsoleAttrGetProgressTrackerSymbols spin_marks _spin_marks_ticker _ticker_indexr_ticker_length _set_spinner)rA set_spinners rrBzSpinner.__init__sUD#224 " " $ZZ DLDd../D#Drctjj|dztjjy)z0Marks spin_mark on stdout and moves cursor back.N)sysstdoutr#flush)rA spin_marks r_Markz Spinner._Mark!s)JJY%&JJrc|jdd|_|jr|jjyy)z2Erases last spin_mark and joins the ticker thread. TN)rrxr}r8rDs rStopz Spinner.Stop&s1JJsOD || llrcttjtjzD]6}t j tjdz |j s6n|j s|j|j}|j||jdz|jz|_t j tjdz |j s~yy)z