dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddlm Z ddl Z dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdZdZGddeZdZdZdZdZ dZ!dZ"dZ#d%d Z$d!Z%d"Z&d#Z'd$Z(y)&z3>rcg}|dd}|D]^}|j|stjjr|j d|zdzK|j d|z`|S)aReturns environment variables beginning with `word`. Args: word: The word that is being compared to environment variables. env_vars: The list of environment variables. Returns: []: No completions. [completions]: List, all possible sorted completions. r*Nr5r6) startswithrr7r8r/)wordenv_vars completionsprefixchilds r MatchEnvVarsrA]so+ 8&e   " " , , .3;,-3;'  rcHt|}|}|t}g}t}t}tj }|r|j } | jtr\d} |jti} | j|t| vr_| jtd\} } |j| n4| j|r d} |} t}nd} |jti} |rU| | vr| r| | }nt}| | }n|tk(r|tk7rt}| s|jtst} gS|t k(r t#d|t$k(rgS|tk(r|t'| | z }n|tk7r+|D]%} | j| s|j| 'na| s t#dt)j*| D]<\}} |j| s| r| tk7r |tz }|j|>|rt-|S)asTry to perform a completion based on the static CLI tree. Args: root: The root of the tree that will be traversed to find completions. cmd_line: [str], original command line. Raises: CannotHandleCompletionError: If FindCompletions cannot handle completion. Returns: []: No completions. [completions]: List, all possible sorted completions. Tr*Fz2Dynamic completions are not handled by this modulez5Positional completions are not handled by this module)r3 LOOKUP_FLAGS FLAG_BOOLEANr9r!r"popr; FLAG_PREFIXgetupdate _VALUE_SEPr-r/ENV_VARLOOKUP_COMMANDS FLAG_DYNAMICr FLAG_VALUErAsix iteritemssorted)rootr&wordsnode global_flagsr> flag_modeenv_var_prefixr=r< is_flag_word child_nodes flag_valuerr@s r_FindCompletionsrZss 8 $% $l#,+)"$. ZZ( 99;D {#lHH\2.k& t ::j!4j  Z  (lkilHH_b1k   !$')")T"$    $ DHH_$=!   l ") @B B  "  |D+66  $E   d #   u %) CE E MM+6LE5!!$'  e|3 Z E   U # 7w B  rctjjjtj tjjddS)z&Returns the SDK installation root dir.N)r!pathsepjoin__file__r-rrr_GetInstallationRootDirras5   (..5cr: ;;rcTtjjtddS)z/Returns the SDK static completion CLI tree dir.datacli)r!r]r_rarrr_GetCompletionCliTreeDirres -/ ??rcZtjj|xs tdS)z0Returns the SDK static completion CLI tree path.zgcloud_completions.py)r!r]r_re) directorys rCompletionCliTreePathrhs* -+-/F HHrc tjdd}tjjtddl}|j } |t_|S#t $r tdwxYw#t_wxYw)z1Loads and returns the static completion CLI tree.Nrz.Cannot find static completion CLI tree module.)sysr]r/regcloud_completionsSTATIC_COMPLETION_CLI_TREE ImportErrorr)sys_pathrktrees rLoadCompletionCliTreerpso xx{HHHOO,./  8 8D CH + : %8 :::CHsA AA//A22 A?c6tjtdS)z&Returns the completions output stream.wb)r!fdopenCOMPLETIONS_OUTPUT_FDrrr_OpenCompletionsOutputStreamrus ($ //rcBt}t}t||S)z7Returns the static completions, None if there are none.)rpr(rZ)rQr&s r_GetCompletionsrws  $  !( $ ))rcDt}|r|tjtjt t }d} t}|j|j|j|r|jyyy#|r|jwwxYw)z>Attempts completions and writes them to the completion stream.N) rwrr r!r" IFS_ENV_VARIFS_ENV_DEFAULTruwriter_encodeclose)r>ifsfs rCompleters!+  " "2::{O LC A & (aggchh{#**,-        s 8B B)N))r __future__rrrr!r,rjgooglecloudsdk.core.utilrrrNr#r%ryrzrtrFrDrLrMrJrKrCr0rIr. Exceptionrr(r3r9rArZrarerhrprurwrrrrrs C&' -.           ) (? ,Yx< @ H 0 *r