o O@sdZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddlm Z dd lm Z dd lm Z ddl Z Gd d d eZ  dddZdS)z5A class for parsing a resource projection expression.)absolute_import)division)unicode_literalsN)resource_exceptions)resource_filter) resource_lex)resource_projection_spec)resource_transformc@seZdZdZddgZdgZdddZGdd d eZGd d d eZ d d Z ddZ ddZ ddZ ddZddZddZdddZdS)ParseraResource projection expression parser. A projection is an expression string that contains a list of resource keys with optional attributes. This class parses a projection expression into resource key attributes and a tree data structure that is used by a projector. A projector is a method that takes a JSON-serializable object and a projection as input and produces a new JSON-serializable object containing only the values corresponding to the keys in the projection. Optional projection key attributes may transform the values in the resulting JSON-serializable object. In the 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. Attributes: __key_attributes_only: Parse projection key list for attributes only. _projection: The resource_projection_spec.ProjectionSpec to parse into. _root: The projection _Tree tree root node. _snake_headings: Dict used to disambiguate key attribute labels. _snake_re: Compiled re for converting key names to angry snake case. optionalreversewrapNcCs*d|_tj||||d|_i|_d|_dS)a!Constructor. Args: defaults: 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. FdefaultssymbolsaliasescompilerN)_Parser__key_attributes_onlyrProjectionSpec _projection_snake_headings _snake_re)selfrrrrrT/tmp/google-cloud-sdk/lib/googlecloudsdk/core/resource/resource_projection_parser.py__init__?s   zParser.__init__c@seZdZdZddZdS)z Parser._TreezDefines a Projection tree node. Attributes: tree: Projection _Tree node indexed by key path. attribute: Key _Attribute. cCsi|_||_dSN)tree attribute)rrrrrrVs zParser._Tree.__init__N)__name__ __module__ __qualname____doc__rrrrr_TreeNs r#c@s eZdZdZddZddZdS)zParser._AttributeaDefines a projection key attribute. Attribute semantics, except transform, are caller defined. e.g., the table formatter uses the label attribute for the column heading for the key. Attributes: align: The column alignment name: left, center, or right. flag: The projection algorithm flag, one of DEFAULT, INNER, PROJECT. hidden: Attribute is projected but not displayed. label: A string associated with each projection key. optional: Column data is optional if True. order: The column sort order, None if not ordered. Lower values have higher sort precedence. reverse: Reverse column sort if True. skip_reorder: Don't reorder this attribute in the next _Reorder(). subformat: Sub-format string. transform: obj = func(obj,...) function applied during projection. width: Fixed column width. wrap: Column can be wrapped if True. cCsNtj|_||_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ dS)NF)r ALIGN_DEFAULTalignflaghiddenlabelr orderr skip_reorder subformat transformwidthr )rr&rrrrps zParser._Attribute.__init__c Csg}|jr |d|jr|d|jr|d|jr"|d|r-dd|}nd}dj|j|jdur:d nt |j|j durG|j nd |j d |j |j rV|j jnd|j|j |d S) Nr'r r r+z, [{0}]|zK({flag}, {order}, {label}, {align}, {active}, {wrap}, {transform}{options}) UNORDERED')r&r)r(r%activer r,options)r'appendr r r+formatjoinr&r)six text_typer(r%r,r2r )roptionr3rrr__str__~s:      zParser._Attribute.__str__N)rr r!r"rr:rrrr _AttributeZs r;cCsz|jdur td|_d}t|D])}t|tjr:|jd|}|r+|d|}n|}||j vr:d|j |<|Sq|S)aReturns an ANGRY_SNAKE_CASE string representation of a parsed key. For key input [A, B, C] the headings [C, C_B, C_B_A] are generated. The first heading not in self._snake_headings is added to self._snake_headings and returned. Args: key: A parsed resource key and/or list of strings. Returns: The ANGRY_SNAKE_CASE string representation of key, adding components from right to left to disambiguate from previous ANGRY_SNAKE_CASE strings. Nz)((?<=[a-z0-9])[A-Z]+|(?!^)[A-Z](?=[a-z]))r/z_\1_) rrecompilereversed isinstancer7 string_typessubupperr)rkeyr(index key_snakerrr_AngrySnakeCases      zParser._AngrySnakeCasecs|j}ddD]+}|j}||vr$||j}|j|jjkr#|jj|_n |||jj||<||}q |j}r>dnd}||v}|rk||j}|j sat fdd|j Drat |}|j rg|j sjd|_n0t|tjrd|vrt |d||<||j}n|}|j r|j rd|_s|jr||||<|j dur|j |_ |j r|jd7_d|_|jdur|j|_n |jdur||_|jtjkr|j|_|jdur|j|_n|jdurd|_|jdur|j|_n|jdurd|_|jr|j|_|jr|j|_|jdur|j|_n |jdurd|_|jdur"|j|_n |jdur+d|_|j |j||j r<|jrJ|jj|_|j!|dS|sT|jj"|_dSdS) zPropagates default attribute values and adds key to the projection. Args: key: The parsed key to add. attribute_add: Parsed _Attribute to add. Nr/csg|] }|jkr|qSrrE).0colrJrr sz"Parser._AddKey..FTr=)#_rootrrr&rPROJECTINNERr#r;ranyColumnscopyr)r'rAr7 integer_typesdeepcopyr,_Parser__key_order_offsetr*r(rHr%rr$r r r+r-r AddAliasAddKeyDEFAULT)rrE attribute_add projectionnamerr name_in_treerrJr_AddKeys                            zParser._AddKeycs.fddjrjjd_dSdS)zRecursively adds self.__key_order_offset to non-zero attribute order. This slides established attribute.order out of the way so new attribute.order in projection composition take precedence. csH|D]}|jjr|jjrd|j_n |jjj7_|jqdS)a;Adds self.__key_order_offset to unmarked attribute.order. A DFS search that visits each attribute once. The search clears skip_reorder attributes marked skip_reorder, otherwise it adds self.__key_order_offset to attribute.order. Args: tree: The attribute subtree to reorder. FN)valuesrr)r*rVr)rnode_AddOffsetToOrderrrrrbs   z*Parser._Reorder.._AddOffsetToOrderrN)rVrNrrrrar_Reorders   zParser._ReordercCs |jjddd}|j}|jjdddr"d}|jjdddd}nd}|d r2|d d }d}nd}||jvrH|sGtd |j |n|r[||j vr[td |j ||dkrv|smtd|j ||j |||na|dkr|t jvrtd|j |||_nH|dkr|pd|_n>|dkr|pd|_n4|dkr||_n,|dkr||_n$|dkr||_n|dkr||_n|dkr||_n td|j ||jdsd Sq)aSParses one or more key attributes and adds them to attribute. The initial ':' has been consumed by the caller. Args: key: The parsed key name of the attributes. attribute: Add the parsed transform to this resource_projector._Attribute. Raises: ExpressionSyntaxError: The expression has a syntax error. Tz=:,)Fspace=eoi_okz:,)rfconvertno-Nzvalue not expected [{0}].zvalue expected [{0}].aliaszCannot unset alias [{0}].r%zUnknown alignment [{0}].r5r/r(r r sortr-r zUnknown key attribute [{0}].:)_lexToken GetPosition IsCharacter startswith_BOOLEAN_ATTRIBUTESrExpressionSyntaxErrorr5Annotate_OPTIONAL_BOOLEAN_ATTRIBUTESrrWr ALIGNMENTSr%r+r(r r r)r-r )rrErr\here boolean_valuevaluerrr_ParseKeyAttributes2sj         zParser._ParseKeyAttributescsP|j\}}|jjdddr|j||jj}nd}|js"|s)|jr/|r/|s/t|}n| |jj }|j s=||_ n |rG|j j |j |j|jdrX||||j r|j jr|jjtdfdd}tjtj|id }tj|j j|d sdS|jdur|s|j r||j jg|j j d j|_|||dS) aParses a key and optional attributes from the expression. The parsed key is appended to the ordered list of keys via _AddKey(). Transform functions and key attributes are also handled here. Raises: ExpressionSyntaxError: The expression has a syntax error. (TrhNrp conditionalscs t|dSr)getattr) unused_obj restrictionunused_patternrrrEvalGlobalRestrictions z/Parser._ParseKey..EvalGlobalRestriction)r)rr) rqKeyWithAttributert Transformpoprr2rrSr;rOr, _transformsextend SkipSpacer~ conditionalrgetr GetTypeDataNamerrGLOBAL_RESTRICTION_NAMErCompileEvaluater(rHr\argsr^)rrEr add_transformrrrrr _ParseKeyrs\       zParser._ParseKeycCsX|jdrdS ||j|jdrdS|jds+td|jq )zParses a comma separated list of keys. The initial '(' has already been consumed by the caller. Raises: ExpressionSyntaxError: The expression has a syntax error. )NT,z*Expected ) in projection expression [{0}].)rqrtrrrrwr5rxrcrrr _ParseKeyss    zParser._ParseKeyscCs |jjddd}|rL|jdr|jjdddd}nd}t|tjr-|d d d d }|j||| d rD|j |ddn|j d ||jdrTdS|jdset d |jq)zParses a comma separated [no-]name[=value] projection attribute list. The initial '[' has already been consumed by the caller. Raises: ExpressionSyntaxError: The expression has a syntax error. Tz=,])Frergz,])rjr=z\n z\t rlrmN]rz#Expected ] in attribute list [{0}].)rqrrrtrAr7rBreplacer AddAttributeru DelAttributerrwr5rx)rr\r}rrr_ParseAttributess&     zParser._ParseAttributescCsR|j|_|js|||jj|_|j|j|j|||jj|rt ||j|_ d}d|_ |j r|j dr\|j sMd}|j||j r[d|_ |n;|j drg|n0|j drtd|_ d|_n#|j }|j d}|std|j ||j|d}|j s=d |_ |r|j|jS) aParse a projection expression. An empty projection is OK. Args: expression: The resource projection expression string. Raises: ExpressionSyntaxError: The expression has a syntax error. Returns: A ProjectionSpec for the expression. Fr[rpTrz:([zName expected [{0}].N)rGetRootrNr#r;rYSetRootSetEmptyrOrLexerrqrrrtDefaultsrrdrrVrsrrisalpharrwr5rxSetName)r expressionrr{r\rrrParsesN            z Parser.Parse)NNNNr)rr r!r"rvryrobjectr#r;rHr^rdr~rrrrrrrrr "s  ?^@5r r/cCst||||d|S)aParses a resource projector expression. Args: expression: The resource projection expression string. defaults: 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. Returns: A ProjectionSpec for the expression. r)r r)rrrrrrrrr s r)r/NNNN)r" __future__rrrrSr>googlecloudsdk.core.resourcerrrrr r7rr rrrrrs&        m