7dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd l mZdd Zd Zd ZdZdZdZdZdZdZdZGddeZy)z5Common methods to display parts of SQL query results.)absolute_import)division)unicode_literals)partial)encoding)resource_printer)text)lexer)tokenscl|D].}|j|k(st|dr|jcS|S|S)a Gets the value for the given key in a list of properties. Looks through a list of properties and tries to find the value for the given key. If it's not found, not_found_value is returned. Args: properties: A dictionary of key string, value string pairs. property_key: The key string for which we want to get the value. not_found_value: The string value to return if the key is not found. Returns: A string containing the value for the given key, or `not_found_value` if the key is not found. value)keyhasattrr ) properties property_keynot_found_valueprops -lib/googlecloudsdk/command_lib/spanner/sql.py_GetAdditionalPropertyrs?d xx< w zz   c t||dS)aCreates tree of Node objects from the plan_nodes in server response. Args: plan_nodes (spanner_v1_messages.PlanNode[]): The plan_nodes from the server response. Plan nodes are topologically sorted. Returns: A Node, root of a tree built from `plan_nodes`. r) _BuildSubTree) plan_nodess r_ConvertToTreer4s z:a= 11rcd}|jr0|jDcgc]}t|||j}}t||Scc}w)a>Helper for building the subtree of a query plan node. Args: plan_nodes (spanner_v1_messages.PlanNode[]): The plan_nodes from the server response. Plan nodes are topologically sorted. node (spanner_v1_messages.PlanNode): The root node of the subtree to be built. Returns: A Node object. N) childLinksr childIndexNode)rnodechildrenlinks rrrBsV( __ OO-+Dj*T__*EF+ - dH -s A ct|d|S)zConverts the prop to a string if it exists. Args: prop (object_value): The value returned from _GetAdditionalProperty. Returns: A string value for the given prop, or the `not_found_value` if the prop does not exist. string_value)getattr)rs r_ConvertToStringValuer%Us ~t ,,rc|rd}nd}|dk(r"|j|j|dy|j|j|dy)aBPrints number of rows modified by a DML statement. Args: row_count: Either the exact number of rows modified by statement or the lower bound of rows modified by a Partitioned DML statement. is_exact_count: Boolean stating whether the number is the exact count. out: Output stream to which we print. zStatement modified {} {}z)Statement modified a lower bound of {} {}rowrowsN)Printformat) row_countis_exact_countout output_strs r_DisplayNumberOfRowsModifiedr0bsJ+J2JK'(<=,Xo-FG+H^,DE0:M1NO  % ` rcxt|jjj}|j |y)zDisplays a graphical query plan for a query. Args: result (spanner_v1_messages.ResultSet): The server response to a query. out: Output stream to which we print. N)rrB queryPlan planNodes PrettyPrint)rDr.node_tree_roots rDisplayQueryPlanrWs,"&,,"8"8"B"BC.S!rct|jdr7|jj!t|jjd|t|jdr7|jj!t|jjd||j j jr|j j jDcgc]}|jxsd}}djdt|D}|jDcgc]#}d tj|ji%}}tj |d j#|| yycc}wcc}w) zPrints the result rows for a query. Args: result (spanner_v1_messages.ResultSet): The server response to a query. out: Output stream to which we print. rowCountExactNTrowCountLowerBoundFz (Unspecified),c3FK|]\}}dj||yw)z!row.slice({0}).join():label="{1}"N)r+).0ifs r z&DisplayQueryResults..s*;(91@FFq!L(9s!r(z table({0})rM)rrBrYr0rZmetadatarowTypefieldsnamejoin enumerater)rMessageToPyValueentryrr*r+)rDr.fieldrc table_formatr(r)s rDisplayQueryResultsrksP V\\ "(,,"<"<"H !;!;T3G  ll & ? ? K !@!@%M __##__,,333E  %o%3  88;(1&(9;;L # x((3    4!4!4\!BL$ s E8%(E=cDeZdZdZd dZdZdZdZdZdZ d Z d d Z y) rzRepresents a single node in a Spanner query plan. Attributes: properties (spanner_v1_messages.PlanNode): The details about a given node as returned from the server. children: A list of children in the query plan of type Node. Nc(|xsg|_||_yN)r r)selfrr s r__init__z Node.__init__sNDM DOrcdj|||jj|jj}|j |y)z@Prints the kind of the node (SCALAR or RELATIONAL) and its name.z {}{} {} {}N)r+rkind displayNamer*)ror.prependstub kind_and_names r_DisplayKindAndNamezNode._DisplayKindAndNames< ''t7K7K(,(C(CEMIImrct|jjj|d}|syt|jj|d}|r |j Sy)avGets a nested property name on this object's executionStats. Args: prop_name: A string of the key name for the outer property on executionStats. nested_prop_name: A string of the key name of the nested property. Returns: The string value of the nested property, or None if the outermost property or nested property don't exist. N)rrexecutionStatsrN object_valuer#)ro prop_namenested_prop_namer nested_props r_GetNestedStatPropertyzNode._GetNestedStatPropertys[ " &&;;Y LD  ():):)E)E)92?K  % %% rcH|jjsyg}|jdd}|rBt|}dj |t j |d}|j||jdd}|jdd}|jdd } |r"|jd j || n#|r!|jd j || |r4d j ||d j|} |j| yy)aPrints the relevant execution statistics for a node. More specifically, print out latency information and the number of executions. This information only exists when query is run in 'PROFILE' mode. Args: out: Output stream to which we print. prepend: String that precedes any information about this node to maintain a visible hierarchy. beneath_stub: String that preserves the indentation of the vertical lines. Nexecution_summarynum_executionsz{} {} executionlatencymeantotalunitz{} {} average latencyz{} {} total latencyz {}{} ({}), ) rrzrintr+r Pluralizeappendrer*) ror.rt beneath_stub stat_propsrexecutions_str mean_latency total_latencyrexecutions_stats_strs r_DisplayExecutionStatszNode._DisplayExecutionStats s ?? ) ) J001D1ACN>*n~~n&*nn^5@'BCn'..y&AL// 7CM  & &y& 9D/66|TJK -44]DIJ(//04 *0EG ii$%rc v|jjrg}|jjjD]A}|jdj |j |j jCdj ||djt|}|j|yy)a0Prints the keys and values of the metadata for a node. Args: out: Output stream to which we print. prepend: String that precedes any information about this node to maintain a visible hierarchy. beneath_stub: String that preserves the indentation of the vertical lines. z{}: {}{}{} {}rN) rrarNrr+rr r#resortedr*)ror.rtradditional_propsrras r_DisplayMetadatazNode._DisplayMetadata3s //**??$ OODHHdjj&=&= > @@!!'<"&))F3C,D"EGh ii rc|jjrCdj|||jjj}|j |yy)Nr)rshortRepresentationr+ descriptionr*)ror.rtr short_reps r_DisplayShortRepresentationz Node._DisplayShortRepresentationFsJ **"" < // - - 9 9;i ii  +rc|rdnd}||jrdndz }dj|||}|j|jy)aDisplays an empty line between nodes for visual breathing room. Keeps in tact the vertical lines connecting all immediate children of a node to each other. Args: out: Output stream to which we print. prepend: String that precedes any information about this node to maintain a visible hierarchy. beneath_stub: String that preserves the indentation of the vertical lines. is_root: Boolean indicating whether this node is the root of the tree.  ryz |z{}{}{}N)r r+r*rstrip)ror.rtris_root above_child break_lines r_DisplayBreakLinezNode._DisplayBreakLineMsH"$rKDMM5r1K, DJIIj!"rc|xsd}|rdn|rdnd}|rdn|rdnd}|j||||j||||j||||j||||j ||||t |j D]C\}}|t|j dz k(} d} ||rdnd z| z} |j|| | d Ey ) aPrints a string representation of this node in the tree. Args: out: Output stream to which we print. prepend: String that precedes any information about this node to maintain a visible hierarchy. is_last: Boolean indicating whether this node is the last child of its parent. is_root: Boolean indicating whether this node is the root of the tree. ryz\-z+-rz| r'z  |F)rtis_lastrN) rwrrrrrfr lenrU) ror.rtrrruridxchild is_last_childindent child_prepends rrUzNode.PrettyPrintbsmG2uTD!2wtDLS'40Wl;#w 5$$S'<@3w? . US/!33mfS9FBm  }mUL/rrn)NTT) __name__ __module__ __qualname____doc__rprwrrrrrrUrrrrs1! 0(&T&#*#LrrN)Unknown)r __future__rrr functoolsrapitools.base.pyrgooglecloudsdk.core.resourcergooglecloudsdk.core.utilr sqlparser r r9rrrr%r0r@rErQrWrkobjectrrrrrsj<&'%9) . 2& -4(* Q2"MDgL6gLr