ndZddlmZddlmZddlmZddlmZddlmZddlm Z ddl Z Gd d e Z y) z,A module for walking the Cloud SDK CLI tree.)absolute_import)division)unicode_literals)Any) console_io)progress_trackerNc:eZdZdZ d dZdZd dZdZdZdZ y) Walkera=Base class for walking the Cloud SDK CLI tree. Attributes: _roots: The root elements of the CLI tree that will be walked. _num_nodes: The total number of nodes in the tree. _num_visited: The count of visited nodes so far. _progress_callback: The progress bar function to call to update progress. Nc N|j}|r8|Dcgc]}|j||}}|Dcgc]}|s| c}|_n|g|_d|_|rZt j d5|jD]+}|xjd|j d|zz c_- dddn:|jD]+}|xjd|j d|zz c_-d|_|xstj|_ ycc}wcc}w#1swY4xYw)aConstructor. Args: cli: The Cloud SDK CLI object. progress_callback: f(float), The function to call to update the progress bar or None for no progress bar. ignore_load_errors: bool, True to ignore command load failures. This should only be used when it is not critical that all data is returned, like for optimizations like static tab completion. restrict: Restricts the walk to the command/group dotted paths in this list. For example, restrict=['gcloud.alpha.test', 'gcloud.topic'] restricts the walk to the 'gcloud topic' and 'gcloud alpha test' commands/groups. When provided here, any groups above the restrictions in the tree will not be loaded or visited. rzLoading CLI Treeg?T) recursiveignore_load_errorsN) _TopElement_GetSubElement_roots _num_nodesrProgressTrackerLoadAllSubElements _num_visitedrDefaultProgressBarCallback_progress_callback) selfcliprogress_callbackr restricttoprrootsroots %lib/googlecloudsdk/calliope/walker.py__init__zWalker.__init__'s!$ // C4<=Hqt""3*He= %+1Q+dkEdkDO  + +,> ?KKD //S4#:#:1C$;$   /  @ ? ++$ 3!8!8/A"9"    DBZBB '>+ @ ?sDDD1;DD$cj|jddd}|}|D]}|j|}|ry|S)N.)splitLoadSubElement)r top_elementpathpartscurrentparts rrzWalker._GetSubElementRsB JJsOAB EG&&t,g  Ncdtdtfddfd fdd_d}jD] }|d} j |S) aCalls self.Visit() on each node in the CLI tree. The walk is DFS, ordered by command name for reproducability. Args: hidden: Include hidden groups and commands if True. universe_compatible: Exclusively include commands which are marked universe compatible. restrict: Restricts the walk to the command/group dotted paths in this list. For example, restrict=['gcloud.alpha.test', 'gcloud.topic'] restricts the walk to the 'gcloud topic' and 'gcloud alpha test' commands/groups. When provided here, parent groups will still be visited as the walk progresses down to these leaves, but only parent groups between the restrictions and the root. Returns: The return value of the top level Visit() call. commandreturncHt|t xr|jS)zDetermines if a command is universe compatible. Args: command: CommandCommon command node. Returns: True if command is universe compatible. ) isinstancedictIsUniverseCompatible)r-s r_IsUniverseCompatiblez*Walker.Walk.._IsUniverseCompatibleos"GT* * O0L0L0NOr+cs|jryr |sysydj|j}D]+}|j|ry|s|j|s+yy)a Determines if command should be included in the walk. Args: command: CommandCommon command node. traverse: If True then check traversal through group to subcommands. Returns: True if command should be included in the walk. FTr")IsHiddenjoinGetPath startswith)r-traverser'itemr3hiddenruniverse_compatibles r_IncludezWalker.Walk.._Includezsn((* %:7%C  XXgoo' (d$ ??4  -  r+c$|js j||d|S j||d}g}|jrDtj|jD]"\}}|s|j ||df$|j rFtj|j D]$\}}|ds|j ||df&t|D]&\}}}|r || j||d(|S)a Walk() helper that calls self.Visit() on each node in the CLI tree. Args: node: CommandCommon tree node. parent: The parent Visit() return value, None at the top level. Returns: The return value of the outer Visit() call. F)is_groupT)r9)r?_Visitcommandssix iteritemsappendgroupssorted) nodeparentcommands_and_groupsnamer-_r?r=_Walkrs rrLzWalker.Walk.._Walks]] D&5 1 {{4${7f  ]]4==9MD' g   & &gu'= >:  ]]4;;7MD' g -  & &gt'< =8#))<"= !Wh   ++gv+ 6 #> mr+rN)F)rboolrrDone) rr;r<rrHrr=r3rLs ```` @@@rWalkz Walker.Walk[s`( Ps Pt P0@D F T4 fIIK Mr+c|xjdz c_|j|j|jz|j|||S)Nr#)rrrVisitrrGrHr?s rr@z Walker._VisitsCD--@A ::dFH --r+cy)aVisits each node in the CLI command tree. Called preorder by WalkCLI() using DFS. Args: node: group/command CommandCommon info. parent: The parent Visit() return value, None at the top level. is_group: True if node is a group, otherwise its is a command. Returns: A new parent value for the node subtree. This value is the parent arg for the Visit() calls for the children of this node. NrRs rrQz Walker.Visits r+cy)zr^s- 3&'28 s Vs r+