,vdZddlmZddlmZddlmZddlZddlZdZdZdZ d d d d Z d Z Gdde Z y)z7A class that creates resource projection specification.)absolute_import)division)unicode_literalsNzglobal.restrictionz' projection: The parent ProjectionSpec.leftc$|j|SN)ljustsws r"s 1771:c$|j|Sr)centerr s r rr#s QXXa[rc$|j|Sr)rjustr s r rr$s AGGAJr)rrrightci}i}|D]T}|s|jr|j|j|js:|j|jVt||S)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)defaultsrrdefaults r CombineDefaultsr's[ ' 'g  nnW__% nnW__%   99rceZdZdZdZdZdZGddeZd!dZ e d Z e d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ e!jDfdZ#d Z$y)"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. rceZdZdZdZy)ProjectionSpec._ColumnzColumn key and transform attribute for self._columns. Attributes: key: The column key. attribute: The column key _Attribute. c ||_||_yr)key attributeselfr#r$s r __init__zProjectionSpec._Column.__init__msdh dnrN)__name__ __module__ __qualname____doc__r'rr _Columnr!es !rr-NcH|xsi|_i|_g|_||_d|_d|_i|_d|_|r|j|_ tj|j|_ |j|rtj|ni|_|jr%|jj!|j|jr&|jj!|jyyd|_ d|_ |xsi|_y)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)r&rrrcompilers r r'zProjectionSpec.__init__qs=bDLDODMDNDKDJDDN__dl==!1!1!34dj mmo/6T]]7+Bdl    H,,-    H,,- dldj]dlrc|jS)z Gets the transform active level.)r7r&s r r6zProjectionSpec.actives <<rc|jS)z>Returns the projection compiler method for nested projections.)r1r?s r r=zProjectionSpec.compilers >>rc|j|j_|jj D]}|j |y)zDefaults() helper -- converts a projection to a default projection. Args: projection: A node in the original projection _Tree. N)DEFAULTr$flagtreevalues _Defaults)r& projectionnodes r rFzProjectionSpec._Defaultss9 !% J&&( nnT)rc t|jD]b}|jdjd|z||j|j|j |j|||dzdy)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 )indentr#r$rN)sortedrDwriteformatr$_Print)r&rGoutlevelr#s r rNzProjectionSpec._Printsqjoo& ii077OOC(22845 kk*//#&UQY7 'rc"||j|<y)zsAdds name=value to the attributes. Args: name: The attribute name. value: The attribute value Nr/)r&namevalues r AddAttributezProjectionSpec.AddAttributes"DOODrc<||jvr|j|=yy)zlDeletes name from the attributes if it is in the attributes. Args: name: The attribute name. NrRr&rSs r DelAttributezProjectionSpec.DelAttributes!  t //$ rc&||f|j|<y)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)r&rSr#r$s r AddAliaszProjectionSpec.AddAliassy)DLLrcZ|jj|j||y)zAdds key and attribute to the projection. Args: key: The parsed key to add. attribute: Parsed _Attribute to add. N)r0appendr-r%s r AddKeyzProjectionSpec.AddKeys! MMc956rc8|jri|_||_y)zSets the projection name. The projection name is the rightmost of the names in the expression. Args: name: The projection name. N)r3r/rWs r SetNamezProjectionSpec.SetNames zzdoDJrc|jS)ziReturns the projection root node. Returns: The resource_projector_parser._Tree root node. r;r?s r r:zProjectionSpec.GetRoot ::rc||_y)ziSets the projection root node. Args: root: The resource_projector_parser._Tree root node. Nrb)r&roots r SetRootzProjectionSpec.SetRoots DJrc|jS)zReturns the projector resource_projector_parser._Tree empty node. Returns: The projector resource_projector_parser._Tree empty node. r2r?s r GetEmptyzProjectionSpec.GetEmptys ;;rc||_y)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. Nrh)r&rHs r SetEmptyzProjectionSpec.SetEmptys DKrc|jS)zReturns the projection columns. Returns: The columns in the projection, None if the entire resource is projected. )r0r?s r ColumnszProjectionSpec.Columns s ==rc,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. )lenr0r?s r ColumnCountzProjectionSpec.ColumnCounts t}} rc|jr|j|jg|_|xjdz c_y)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)r;rFr0r7r?s r r<zProjectionSpec.Defaultss0 zz nnTZZ DMLLALrc|jS)zReturns the short key name alias dictionary. This dictionary maps short (no dots) names to parsed keys. Returns: The short key name alias dictionary. rZr?s r AliaseszProjectionSpec.Aliases&s <<rc|jS)zlReturns the projection _Attribute dictionary. Returns: The projection _Attribute dictionary. rRr?s r AttributeszProjectionSpec.Attributes0s ??rcv|jDcgc]}t|jj!c}Scc}w)zReturns the projection column justfication list. Returns: The ordered list of alignment functions, where each function is one of ljust [default], center, or rjust. )r0 ALIGNMENTSr$align)r&cols r AlignmentszProjectionSpec.Alignments8s/8<}} E}Js}}** +} EE Es$6c|jDcgc]}|jjxsd}}t|r|SdScc}w)zReturns the ordered list of projection labels. Returns: The ordered list of projection label strings, None if all labels are empty. N)r0r$labelany)r&rylabelss r LabelszProjectionSpec.LabelsAsE48== A=Ccmm!!'R'=F A[6*d*Bs!Ac|jS)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. )r3r?s r NamezProjectionSpec.NameKs ::rcg}t|jD]\}}|jjs|jjs3|j |jjxst j||jjft|Dcgc] \}}}||f c}}}Scc}}}w)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). ) enumerater0r$orderreverser]sysmaxsizerK)r&orderingiry_rs r OrderzProjectionSpec.OrderUsHDMM*3    5 5 ]] /CKKCMM4I4I J L+/5X.> ?.>]Q7QL.> ?? ?s*B>cX|jr|j|j|dyy)znPrints the projection with indented nesting. Args: out: The output stream, sys.stdout if None. rN)r;rN)r&rOs r PrintzProjectionSpec.Printis$  zz kk$**c1%rc|jS)zTReturns the projection tree root. Returns: The projection tree root. rbr?s r TreezProjectionSpec.Treerrcr)NNNN)%r(r)r*r+rBINNERPROJECTobjectr-r'propertyr6r=rFrNrUrXr[r^r`r:rfrirkrmrpr<rsrurzrrrrstdoutrrr,rr rr>s D ' % ' ! !#>    8"" *7  F+@(jj&rr)r+ __future__rrrr8rGLOBAL_RESTRICTION_NAMEPROJECTION_ARG_DOC ALIGN_DEFAULTrwrrrr,rr rsO >&' /> -0.0 :.zVzr