o ,@sxdZddlmZddlmZddlmZddlZddlZdZdZdZ d d d d d d d Z ddZ Gddde Z dS)z7A class that creates resource projection specification.)absolute_import)division)unicode_literalsNzglobal.restrictionz' projection: The parent ProjectionSpec.leftcC ||SN)ljustswr R/tmp/google-cloud-sdk/lib/googlecloudsdk/core/resource/resource_projection_spec.py" rcCrr)centerr r r r r#rcCrr)rjustr r r r r$r)rrrightcCsHi}i}|D]}|s q|jr||j|jr||jqt||dS)aFCombines a list of defaults into a new defaults object. Args: defaults: An ordered list of ProjectionSpec objects to combine. alias and symbol names from higher index objects in the list take precedence. Returns: A new ProjectionSpec object that is a combination of the objects in the defaults list. )symbolsaliases)rupdaterProjectionSpec)defaultsrrdefaultr r r CombineDefaults's    rc@seZdZdZdZdZdZGdddeZd:dd Z e d d Z e d d Z ddZ ddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Z e!j"fd6d7Z#d8d9Z$dS);raGCreates a resource projection specification. A resource projection is an expression string that contains a list of resource keys with optional attributes. A projector is a method that takes a projection specification and a resource object as input and produces a new JSON-serializable object containing only the values corresponding to the keys in the projection specification. Optional projection key attributes may transform the values in the output JSON-serializable object. Cloud SDK projection attributes are used for output formatting. A default or empty projection expression still produces a projector that converts a resource to a JSON-serializable object. This class is used by the resource projection expression parser to create a resource projection specification from a projection expression string. Attributes: aliases: Resource key alias dictionary. _active: The transform active level. Incremented each time Defaults() is called. Used to determine active transforms. attributes: Projection attributes dict indexed by attribute name. _columns: A list of (key,_Attribute) tuples used to project a resource to a list of columns. _compiler: The projection compiler method for nested projections. _empty: An empty projection _Tree used by Projector(). _name: The projection name from the expression string. _tree: The projection _Tree root, used by resource_projector.Evaluate() to efficiently project each resource. symbols: Default and caller-defined transform function dict indexed by function name. rc@seZdZdZddZdS)zProjectionSpec._ColumnzColumn key and transform attribute for self._columns. Attributes: key: The column key. attribute: The column key _Attribute. cCs||_||_dSr)key attributeselfrrr r r __init__ms zProjectionSpec._Column.__init__N)__name__ __module__ __qualname____doc__r r r r r _Columnes r%NcCs|pi|_i|_g|_||_d|_d|_i|_d|_|rN|j|_ t | |_ ||r3t |ni|_|jr@|j|j|jrL|j|jdSdSd|_ d|_ |pWi|_dS)a8Initializes a projection. Args: defaults: A list of resource_projection_spec.ProjectionSpec defaults. symbols: Transform function symbol table dict indexed by function name. aliases: Resource key alias dictionary. compiler: The projection compiler method for nested projections. Nr)r attributes_columns _compiler_empty_name_snake_headings _snake_reactive_activecopydeepcopyGetRoot_treeDefaultsrr)rrrrcompilerr r r r qs* zProjectionSpec.__init__cC|jS)z Gets the transform active level.)r.rr r r r-zProjectionSpec.activecCr5)z>Returns the projection compiler method for nested projections.)r(r6r r r r4r7zProjectionSpec.compilercCs(|j|j_|jD]}||q dS)zDefaults() helper -- converts a projection to a default projection. Args: projection: A node in the original projection _Tree. N)DEFAULTrflagtreevalues _Defaults)r projectionnoder r r r<s  zProjectionSpec._DefaultscCsNt|jD]}|djd|||j|jd||j|||dqdS)aPrint() helper -- prints projection node p and its children. Sorted by projection tree level for diff stability. Args: projection: A _Tree node in the original projection. out: The output stream. level: The nesting level counting from 1 at the root. z{indent} {key} : {attribute} z )indentrrrN)sortedr:writeformatr_Print)rr=outlevelrr r r rCs  zProjectionSpec._PrintcCs||j|<dS)zsAdds name=value to the attributes. Args: name: The attribute name. value: The attribute value Nr&)rnamevaluer r r AddAttributeszProjectionSpec.AddAttributecCs||jvr |j|=dSdS)zlDeletes name from the attributes if it is in the attributes. Args: name: The attribute name. NrFrrGr r r DelAttributes  zProjectionSpec.DelAttributecCs||f|j|<dS)zAdds name as an alias for key and attribute to the projection. Args: name: The short (no dots) alias name for key. key: The parsed key to add. attribute: The attribute for key. Nr)rrGrrr r r AddAliasszProjectionSpec.AddAliascCs|j|||dS)zAdds key and attribute to the projection. Args: key: The parsed key to add. attribute: Parsed _Attribute to add. N)r'appendr%rr r r AddKeyszProjectionSpec.AddKeycCs|jri|_||_dS)zSets the projection name. The projection name is the rightmost of the names in the expression. Args: name: The projection name. N)r*r&rJr r r SetNames zProjectionSpec.SetNamecCr5)ziReturns the projection root node. Returns: The resource_projector_parser._Tree root node. r2r6r r r r1zProjectionSpec.GetRootcC ||_dS)ziSets the projection root node. Args: root: The resource_projector_parser._Tree root node. NrQ)rrootr r r SetRoots zProjectionSpec.SetRootcCr5)zReturns the projector resource_projector_parser._Tree empty node. Returns: The projector resource_projector_parser._Tree empty node. r)r6r r r GetEmptyrRzProjectionSpec.GetEmptycCrS)zSets the projector resource_projector_parser._Tree empty node. The empty node is used by to apply [] empty slice projections. Args: node: The projector resource_projector_parser._Tree empty node. NrV)rr>r r r SetEmptys zProjectionSpec.SetEmptycCr5)zReturns the projection columns. Returns: The columns in the projection, None if the entire resource is projected. r'r6r r r Columns rRzProjectionSpec.ColumnscCs t|jS)zReturns the number of columns in the projection. Returns: The number of columns in the projection, 0 if the entire resource is projected. )lenr'r6r r r ColumnCounts zProjectionSpec.ColumnCountcCs*|jr ||jg|_|jd7_dS)zConverts the projection to a default projection. A default projection provides defaults for attribute values and function symbols. An explicit non-default projection value always overrides the corresponding default value. rN)r2r<r'r.r6r r r r3s zProjectionSpec.DefaultscCr5)zReturns the short key name alias dictionary. This dictionary maps short (no dots) names to parsed keys. Returns: The short key name alias dictionary. rLr6r r r Aliases&zProjectionSpec.AliasescCr5)zlReturns the projection _Attribute dictionary. Returns: The projection _Attribute dictionary. rFr6r r r Attributes0rRzProjectionSpec.AttributescCsdd|jDS)zReturns the projection column justfication list. Returns: The ordered list of alignment functions, where each function is one of ljust [default], center, or rjust. cSsg|]}t|jjqSr ) ALIGNMENTSralign.0colr r r ?z-ProjectionSpec.Alignments..rYr6r r r Alignments8szProjectionSpec.AlignmentscCs dd|jD}t|r|SdS)zReturns the ordered list of projection labels. Returns: The ordered list of projection label strings, None if all labels are empty. cSsg|]}|jjp dqS))rlabelrbr r r reHrfz)ProjectionSpec.Labels..N)r'any)rlabelsr r r LabelsAszProjectionSpec.LabelscCr5)zReturns the projection name. The projection name is the rightmost of the names in the expression. Returns: The projection name, None if none was specified. )r*r6r r r NameKr^zProjectionSpec.NamecCsXg}t|jD]\}}|jjs|jjr"||jjptj||jjfqddt|DS)aReturns the projection sort key order suitable for use by sorted(). Example: projection = resource_projector.Compile('...') order = projection.Order() if order: rows = sorted(rows, key=itemgetter(*order)) Returns: The list of (sort-key-index, reverse), [] if projection is None or if all sort order indices in the projection are None (unordered). cSsg|] \}}}||fqSr r )rc_ireverser r r regsz(ProjectionSpec.Order..) enumerater'rorderrprNsysmaxsizer@)rorderingrordr r r OrderUs zProjectionSpec.OrdercCs|jr ||j|ddSdS)znPrints the projection with indented nesting. Args: out: The output stream, sys.stdout if None. rN)r2rC)rrDr r r PrintiszProjectionSpec.PrintcCr5)zTReturns the projection tree root. Returns: The projection tree root. rQr6r r r TreerrRzProjectionSpec.Tree)NNNN)%r!r"r#r$r8INNERPROJECTobjectr%r propertyr-r4r<rCrIrKrMrOrPr1rUrWrXrZr\r3r]r_rgrlrmrvrsstdoutrwrxr r r r r>sB"                 r)r$ __future__rrrr/rsGLOBAL_RESTRICTION_NAMEPROJECTION_ARG_DOC ALIGN_DEFAULTr`rr{rr r r r s