o 2@sdZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z Gd d d e Z Gd ddeZGddde ZGdddeZdS)aCloud resource filter expression rewrite backend classes. These classes are alternate resource_filter.Compile backends that rewrite expressions instead of evaluating them. To rewrite a filter expression string: rewriter = resource_expr_rewrite.Backend() frontend_expr, backend_expr = rewriter.Rewrite(filter_expression_string) It is possible for a rewritten expression to collapse to None. This means that there is no equivalent server-side expression, i.e., no server-side pruning is possible. These rewrites can only prune expressions that will be False client-side. In this sense a rewrite => None means "the client side will figure it out". This results in a backend expression that can be applied server-side to prune the resources passed back to the client-side, where the full filter expression is applied. The result will be the same whether or not the backend filter is applied. The only difference would be the number of resources transmitted from the server back to the client. None is the value for keys and operators not supported by the backend. ExprTRUE, ExprAND, ExprOR and ExprNOT do expression rewrites based on None: TRUE => None None AND x => x x AND None => x x OR None => None None OR x => None NOT None => None )absolute_import)division)unicode_literals)resource_exceptions)resource_filter) resource_lex)resource_projection_spec)resource_property)resource_transformc@s eZdZdZddZddZdS)_ExprzHAn expression rewrite object that evaluates to the rewritten expression.cCs ||_dSNexprselfrrO/tmp/google-cloud-sdk/lib/googlecloudsdk/core/resource/resource_expr_rewrite.py__init__>s z_Expr.__init__cCs|jS)z@Returns the server side string rewrite of the filter expression.r rrrrRewriteAsz _Expr.RewriteN)__name__ __module__ __qualname____doc__rrrrrrr ;s r c@seZdZdZdddZdS) _BelieveMez3A symbols dict with nothing that claims everything.NcCs ~~|jSr )get)robjtyperrrrIsz_BelieveMe.getr )rrrrrrrrrrFsrc@seZdZdZd>ddZd?ddZddZd d Zd d Zd dZ ddZ ddZ d?ddZ ddZ d@ddZd@ddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd>d*d+Zd>d,d-Zd>d.d/Zd>d0d1Zd>d2d3Zd>d4d5Zd>d6d7Zd>d8d9Zd>d:d;Zd.Nrr unused_left unused_rightrrr RewriteANDzBackendBase.RewriteANDcCr1)Rewrites .Nrr3rrr RewriteORr7zBackendBase.RewriteORcCr1)zRewrites .Nr)r unused_exprrrr RewriteNOTr7zBackendBase.RewriteNOTcCr1)z#Rewrites global restriction .Nr)r unused_callrrr RewriteGlobalr7zBackendBase.RewriteGlobalcCr1)Rewrites an operand.Nr)runused_operandrrrRewriteOperandr7zBackendBase.RewriteOperandcCr1)Rewrites .Nr)r unused_key unused_opr?unused_key_typerrr RewriteTermszBackendBase.RewriteTermcCs:t|} |jddd}|s |S|dvrdj|dSq)z;Returns expression enclosed in (...) if it contains AND/OR.Tz ())balance_parens)ANDORz({expression})r,)rLexerTokenformat)rr,lextokrrr Parenthesizes  zBackendBase.ParenthesizeFcCsztt|WStyYnwztt|WSty!Ynwg}|}d}|D]9}|r1d}n-|dkrD||||d}d}n|dkrP|dd}n|s\|dks\|dkr^d}||q*d|}|rqdj|d S|S) a(Returns value or value "..." quoted with C-style escapes if needed. Args: value: The string value to quote if needed. always: Always quote non-numeric value if True. Returns: A string: value or value "..." quoted with C-style escapes if needed or requested. F\T"':z "{string}")string)strint ValueErrorfloatappendr+joinrL)rvaluealwayscharsencloseescapedcrUrrrQuotes:        zBackendBase.QuotecsHt|trfdd|D}dddd|DdSj|dS)aReturns operand enclosed in "..." if necessary. Args: operand: A string operand or list of string operands. If a list then each list item is quoted. always: Always quote if True. Returns: A string: operand enclosed in "..." if necessary. csg|] }j|dqS)r])rb.0xr]rrr sz,BackendBase.QuoteOperand..(,cSsg|]}|dur|qSr rrdrrrrhs)rc) isinstancelistr[rb)roperandr]operandsrrgr QuoteOperands zBackendBase.QuoteOperandcCs|s|r |dSt|}|jrEz t||j\}}Wn"ty>|jdur7t||js7t d ||dYSwt|}nd}|| ||||S)aReturns the rewritten backend term expression. Args: key: The parsed key. op: The operator name. operand: The operand. transform: The transform object if a transform was specified. args: The transform args if a transform was specified. Raises: UnknownFieldError: If key is not supported on the frontend and backend. Returns: The rewritten backend term expression. Nz!Unknown field [{}] in expression.) r0r GetKeyNamer r GetMessageFieldTypeKeyErrorr LookupFieldrUnknownFieldErrorrLrE)rkeyoprn transformargskey_namekey_typerrrTerms(      zBackendBase.TermcCstdSr )r rrrrExprTRUEszBackendBase.ExprTRUEcCsZ|r|}|r |}|sd|_|r||SdS|s$d|_||S||||S)zReturns an AND expression node.TNF)rr!r0completer6rleftrightrrrExprANDs zBackendBase.ExprANDcCsF|r|}|s |dS|r|}|s|dS||||S)zReturns an OR expression node.N)rr0r9rrrrExprOR%s  zBackendBase.ExprORcCs*|r|}|s |dS|||Sr )rr0r;rrrrExprNOT3s  zBackendBase.ExprNOTcCs|||Sr )r0r=)rcallrrr ExprGlobal:szBackendBase.ExprGlobalcCs|Sr r)rr\rrr ExprOperand=zBackendBase.ExprOperandcC||d|||S)N<r|rrvrnrxryrrrExprLT@zBackendBase.ExprLTcCr)Nz<=rrrrrExprLECrzBackendBase.ExprLEcCr)NrSrrrrrExprHASFrzBackendBase.ExprHAScCr)N=rrrrrExprEQIrzBackendBase.ExprEQcCr)Nz!=rrrrrExprNELrzBackendBase.ExprNEcCr)Nz>=rrrrrExprGEOrzBackendBase.ExprGEcCr)N>rrrrrExprGTRrzBackendBase.ExprGTcCr)N~rrrrrExprREUrzBackendBase.ExprREcCr)N!~rrrrr ExprNotREXrzBackendBase.ExprNotREcCr1r/rrrrr IsRewriter[rzBackendBase.IsRewriter)NNr )F)!rrrrrrr0r6r9r;r=r@rErOrbrpr|r}rrrrrrrrrrrrrrrrrrrrNs@    ++          rc@s8eZdZdZddZddZddZdd Zd d Zd S) Backenda[Rewrites for OnePlatform server side filter expressions. This class rewrites client side expressions to OnePlatform server side expressions. The only difference is the server side does not support the regular expression ~ and !~ operators. Use this class for target expressions that implement a large subset of OnePlatform expressions. cCdj||||dS)r2z{left} AND {right}rrrLrOrrrrr6h zBackend.RewriteANDcCr)r8z{left} OR {right}rrrrrrr9mrzBackend.RewriteORcCsdj||dS)zRewrites .zNOT {expression}rIr)rr,rrrr;rszBackend.RewriteNOTcCs ||S)r>)rp)rrnrrrr@vs zBackend.RewriteOperandcCs4~|dvrdS||}|durdSdj|||dS)rA)rrNz{key}{op}{operand})rvrwrn)r@rL)rrvrwrnr{argrrrrEzs zBackend.RewriteTermN) rrrrr6r9r;r@rErrrrr_s rN)r __future__rrrgooglecloudsdk.core.resourcerrrrr r objectr dictrrrrrrrs