OdZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddlm Z dd lm Z dd lm Z ddl Z Gd d eZ dd Zy)z5A class for parsing a resource projection expression.)absolute_import)division)unicode_literalsN)resource_exceptions)resource_filter) resource_lex)resource_projection_spec)resource_transformceZdZdZddgZdgZddZGddeZGd d eZ d Z d Z d Z dZ dZdZdZddZy)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. optionalreversewrapNcjd|_tj|||||_i|_d|_y)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_onlyr ProjectionSpec _projection_snake_headings _snake_re)selfrrrrs >lib/googlecloudsdk/core/resource/resource_projection_parser.py__init__zParser.__init__?s8"'D/>>7GhPDDDNceZdZdZdZy) Parser._TreezDefines a Projection tree node. Attributes: tree: Projection _Tree node indexed by key path. attribute: Key _Attribute. c i|_||_yN)tree attribute)rr$s rrzParser._Tree.__init__Vsdi dnrN)__name__ __module__ __qualname____doc__rrr_Treer Ns !rr*ceZdZdZdZdZy)Parser._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. ctj|_||_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ y)NF)r ALIGN_DEFAULTalignflaghiddenlabelr orderr skip_reorder subformat transformwidthr)rr0s rrzParser._Attribute.__init__ps]+99djdidkdjdmdjdlddndndjdirc g}|jr|jd|jr|jd|jr|jd|jr|jd|r!dj dj |}nd}dj |j|jd ntj|j|j |jnd |jzd z|j|jr|jjnd|j|j| S) Nr1r rr5z, [{0}]|zK({flag}, {order}, {label}, {align}, {active}, {wrap}, {transform}{options}) UNORDERED')r0r3r2r/activerr6options)r1appendr rr5formatjoinr0r3six text_typer2r/r6r=r)roptionr>s r__str__zParser._Attribute.__str__~sf  h  j!  i  k" ""388F#34''-vyy JJ.%47MM$**4M'+zz'9s@$&)@* 26....dyy NN!(. (# $rN)r%r&r'r(rrEr)rr _Attributer,Zs* $rrFcZ|jtjd|_d}t|D]t}t |t j s|jjd|j}|r |dz|z}n|}||jvsdd|j|<|S|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. z)((?<=[a-z0-9])[A-Z]+|(?!^)[A-Z](?=[a-z]))r:z_\1_) rrecompilereversed isinstancerB string_typessubupperr)rkeyr2index key_snakes r_AngrySnakeCasezParser._AngrySnakeCases ~~zz"MNdn E# E3++ ,NN&&vu5;;= c/E)%% ,, ,()$  u %  L Lrcj|j}|ddD]}|j}||vrN||j}|j|jj k7rS|jj |_n7|j|j|jj ||<||}|j}|r|dnd}||v}|r||j}|jsVt|jjDcgc]}|j|k(s|c}rtj|}|jr |jsd|_nt!|t"j$r/d|vr+tj&|d||<||j}nC|}|jr|jrd|_|s |j(r|j|||<|j9|j|_|jr|xj*dz c_d|_|j.|j.|_n"|j.|j1||_|j2t4j6k7r|j2|_|j8|j8|_n|j8d|_|j:|j:|_n|j:d|_|j(r|j(|_|j<r|j<|_|j>|j>|_n|j>d|_|j@|j@|_ n|j@d|_ |jjC|j.|||jr |jr8|jj |_|jjE||y|s|jjF|_yycc}w)zPropagates default attribute values and adds key to the projection. Args: key: The parsed key to add. attribute_add: Parsed _Attribute to add. Nr:FTrI)$_rootr#r$r0rPROJECTINNERr*rFranyColumnsrQcopyr3r1rMrB integer_typesdeepcopyr6_Parser__key_order_offsetr4r2rTr/r r.r rr5r7rAddAliasAddKeyDEFAULT) rrQ attribute_add projectionnamer#r$ name_in_treecols r_AddKeyzParser._AddKeysJCR __d J(( >>T--55 5++11).ZZ0@0@0F0F GHT :j ??D3r7rD4  % %)9)9''//in c9- ''//in{Ls P0P0ctfdjr$jjd_yy)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. c"|jD]{}|jjrQ|jjrd|j_n)|jxjjz c_|j }y)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)valuesr$r3r4r_r#)r#node_AddOffsetToOrderrs rrmz*Parser._Reorder.._AddOffsetToOrdersb++-$ >>   ^^ ( (*/DNN ' NN D$;$; ; $))$  rrN)r_rWr#)rrms`@r_ReorderzParser._Reorders0%$  ( !drcT |jjdd}|jj}|jjddr!d}|jjddd}nd}|j d r|d d }d}nd}||j vr?|st jd j|jj||rK||jvr=t jd j|jj||dk(r]|s=t jdj|jj||jj|||n|dk(rW|tjvr=t jdj|jj|||_n|dk(r |xsd|_n|dk(r |xsd|_n~|dk(r||_nq|dk(r||_nd|dk(r||_nW|dk(r||_nJ|dk(r||_n=t jdj|jj||jjdsy )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:,)rqconvertno-Nzvalue not expected [{0}].zvalue expected [{0}].aliaszCannot unset alias [{0}].r/zUnknown alignment [{0}].r@r:r2r rsortr7rzUnknown key attribute [{0}].:)_lexToken GetPosition IsCharacter startswith_BOOLEAN_ATTRIBUTESrExpressionSyntaxErrorr@Annotate_OPTIONAL_BOOLEAN_ATTRIBUTESrr`r ALIGNMENTSr/r5r2r rr3r7r)rrQr$rehere boolean_valuevalues r_ParseKeyAttributeszParser._ParseKeyAttributes2sn  YY__V5_ 1d YY " " $d   s4  0  UDA ??5 !ab$%% )) )#99)001C1CD1IJL L T)J)JJ!77 # * *499+=+=d+C DF F #99)001C1CD1IJL L !!%i8 7? 0;; ;#99(// 0B0B40HIK K  8 #kr  7?+2  : "  9 !  6>  7?  6> !77 * 1 1$))2D2DT2J KM M YY " "3 ' e rc<|jj\}}|jjddr?|jj|j |j j }nd}|js|s|jr|r|stj|}n%|j|j j}|js||_ n1|r/|jjj|j|jj|jjdr|j|||jr|jj r|j j"j%t'j(dfd}t+j,t*j.|i}t1j2|jj | j5sy|j6]|s[|jrO|j9|jj:g|jjd j<z|_|j?||y) 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. (TrsNr{ conditionalsct|dSr")getattr) unused_obj restrictionunused_patternrs rEvalGlobalRestrictionz/Parser._ParseKey..EvalGlobalRestrictions|[$77r)r)rr) r|KeyWithAttributer Transformpoprr=rr\rFrXr6 _transformsextend SkipSpacer conditionalrgetr GetTypeDataNamer rGLOBAL_RESTRICTION_NAMErCompileEvaluater2rTreargsrh)rrQr$ add_transformrrrs @r _ParseKeyzParser._ParseKeyrs YY//1NC yyS.ii))#'')T5E5E5L5LMmm  & &t11i ))I&i//$"2"2":":;i   )i %%,,]-F-FGII yyS! sI.y22>>%%--11  , ,^ <>l8*88+CC(*+h $ $    ) )%Xl34 sy/B/B,,    # # $    ) )! , 1 1 23io LLi rcx|jjdry |j|jj|jjdry|jjds.>.F.FGHdj tzz* 4??4#3#3#;#;<=?$$Z1A1ABdih#(d II   ! 99  %++H    % % ' //   ' ').D & MMO YY " "3 '    ! YY " "3 ''+$ $$%$ !&&($'$%;;&--dii.@.@.FGI I    " "4 ((+ II   !,di  !!#   r)NNNNr")r%r&r'r(rrrobjectr*rFrTrhrnrrrrrr)rrr r "sg2$Y/"(  !f !=$6=$~<\0|":>@3!j'*P81rr c>t||||j|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)rrrrrs rrr s$ 7G! ##(5#45r)r:NNNN)r( __future__rrrr\rJgooglecloudsdk.core.resourcerrrr r rBrr rr)rrrsH <&' <85A; gVgT?C5r