dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd Z Gd d e ZGd dejZddZy )zACoverage tree generator. Used for flag coverage kokoro presubmit.)absolute_import)division)unicode_literals)walker)progress_tracker)resource_printer)resource_projectorNctjd5tjt |||j cdddS#1swYyxYw)aGenerates and returns the static completion CLI tree. Args: cli: The CLI. branch: The path of the CLI subtree to generate. restrict: The paths in the tree that we are allowing the tree to walk under. Returns: Returns the serialized static completion CLI tree. z&Generating the flag coverage CLI tree.)restrictN)rProgressTrackerr MakeSerializableCoverageTreeGeneratorWalkclibranchr s 8lib/googlecloudsdk/command_lib/flag_coverage/generate.pyGenerateCoverageTreersI''.0  . .c6H=BBD F000s /AAc(eZdZdZfdZdZxZS)CoverageCommandNodezCommand/group info. Attributes: commands: {str:_Command}, The subcommands in a command group. flags: [str], Command flag list. Global flags, available to all commands, are in the root command flags list. cZtt| ||_|!|jj dd}|||<|j }|jD]S}|jD]B}|jr|jds"|j|r4|j||<DUy)N_-z--) superr__init___parentnamereplaceai flag_argsoption_strings is_hidden startswithIsAncestorFlagrequire_coverage_in_tests)selfcommandparentrargsarg __class__s rrzCoverageCommandNode.__init__6s t-/DL  \\ ! !#s +dfTl ::D~~$$$ == t$    t $ 22T %cH|j}|r||vry|j}|ry)aODetermines if flag is provided by an ancestor command. NOTE: This function is used to allow for global flags to be added in at the top level but not in subgroups/commands Args: flag: str, The flag name (no leading '-'). Returns: bool, True if flag provided by an ancestor command, false if not. TF)r)r&flagr's rr$z"CoverageCommandNode.IsAncestorFlagJs-llG  g  r,)__name__ __module__ __qualname____doc__rr$ __classcell__r+s@rrr-s3(r,rc*eZdZdZdfd ZdZxZS)rz0Generates the gcloud static completion CLI tree.c>tt| ||||_y)z:branch is the command path of the CLI subtree to generate.)rr N)rrr_branch)r&rrr r+s rrzCoverageTreeGenerator.__init__`s /C(/KDLr,ct||S)aWVisits each node in the CLI command tree to construct the external rep. Args: node: group/command CommandCommon info. parent: The parent Visit() return value, None at the top level. is_group: True if node is a command group. Returns: The subtree parent value, used here to construct an external rep node. )r)r&noder(is_groups rVisitzCoverageTreeGenerator.Visites tV ,,r,NNN)r/r0r1r2rr;r3r4s@rrr]s8 -r,rcRt|||}tj|d||S)aVLists the flag coverage CLI tree as a Python module file. Args: cli: The CLI. branch: The path of the CLI subtree to generate. out: The output stream to write to, sys.stdout by default. restrict: The paths in the tree that we are allowing the tree to walk under. Returns: Returns the serialized coverage CLI tree. rjson) print_formatout)rrPrint)rrr@r trees rOutputCoverageTreerCss) #fx H$F< +r,)NNr<)r2 __future__rrrgooglecloudsdk.calliopergooglecloudsdk.core.consolergooglecloudsdk.core.resourcerr rdictrWalkerrrCr,rrKsH H&'*89;F"-$-`-FMM-,r,