7SrSSKJr SSKJr SSKJr SSKJr SSKJr SSK J r SSK J r SS K Jr SS K Jr SS jrS rS rSrSrSrSrSrSrSr"SS\5rg)z5Common methods to display parts of SQL query results.)absolute_import)division)unicode_literals)partial)encoding)resource_printer)text)lexer)tokensc|UH5nURU:XdM[US5(aURs $ U$ U$)aGets 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_GetAdditionalPropertyrsBd xx< w  zz   c[XS5$)a Creates 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 za= 11rcSnUR(a2URVs/sHn[XUR5PM nn[X5$s snf)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 rrrBsR( __ OO-+DjT__*EF+ - d -s"Ac[USU5$)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 ,,rcU(aSnOSnUS:Xa"URURUS55 gURURUS55 g)a6Prints 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  % ` rcx[URRR5nUR U5 g)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!rc:[URS5(a8URRb![URRSU5 [URS5(a8URRb![URRSU5 UR R R(aUR R RVs/sHnUR=(d SPM nnSRS[U555nURVs/sH%nS [R"UR50PM' nn[R "US R#U5US 9 ggs snfs snf) 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),c3H# UHupSRX5v M g7f)z!row.slice({0}).join():label="{1}"N)r+).0ifs r &DisplayQueryResults..s(;(9@FFqLL(9s "r(z table({0})rM)rrBrYr0rZmetadatarowTypefieldsnamejoin enumerater)rMessageToPyValueentryrr*r+)rDr.fieldrd table_formatr(r)s rDisplayQueryResultsrlsZ V\\ "(,,"<"<"H !;!;T3G  ll & ? ? K !@!@%M __##__,,333E  %o%3  88;(1&(9;;L # x((3    4!4!4\!BL$ s .F>,FcP\rSrSrSrS SjrSrSrSrSr S r S r SS jr S r g)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. Nc0U=(d /UlXlgNr r)selfrr s r__init__ Node.__init__sNDM OrcSRX#URRURR5nUR U5 g)z@Prints the kind of the node (SCALAR or RELATIONAL) and its name.z {}{} {} {}N)r+rkind displayNamer*)rrr.prependstub kind_and_names r_DisplayKindAndNameNode._DisplayKindAndNames: ''t7K7K(,(C(CEMIImrc[URRRUS5nU(dg[URRUS5nU(a UR $g)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#)rr prop_namenested_prop_namer nested_props r_GetNestedStatPropertyNode._GetNestedStatPropertys[ " &&;;Y LD  ():):)E)E)92?K  % %% rcvURR(dg/nURSS5nU(aC[U5nSR U[ R "US55nURU5 URSS5nURSS5nURSS 5n U(a!URS R Xy55 O'U(a URS R X55 U(a3S R X#S RU55n URU 5 gg)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 {}{} ({}), ) rrrintr+r Pluralizeappendrfr*) rrr.rx beneath_stub stat_propsrexecutions_str mean_latency total_latencyrexecutions_stats_strs r_DisplayExecutionStatsNode._DisplayExecutionStats s ?? ) ) J001D1ACN>*n~~n&*nn^5@'BCn'..y&AL// 7CM  & &y& 9D/66|JK -44]IJ(//04 *0EG ii$%rc URR(a/nURRRHBnURSR UR UR R55 MD SR X#SR[U555nURU5 gg)aPrints 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) rrbrNrr+rr r#rfsortedr*)rrr.rxradditional_propsrrbs r_DisplayMetadataNode._DisplayMetadata3s //**??$ OODHHdjj&=&= > @@!!'"&))F3C,D"EGh ii rcURR(aBSRX#URRR5nUR U5 gg)Nr)rshortRepresentationr+ descriptionr*)rrr.rxr short_reps r_DisplayShortRepresentation Node._DisplayShortRepresentationFsH **""  // - - 9 9;i ii  +rcU(aSOSnXPR(aSOS- nSRX#U5nURUR55 g)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.  r~z |z{}{}{}N)r r+r*rstrip)rrr.rxris_root above_child break_lines r_DisplayBreakLineNode._DisplayBreakLineMsD"$rKMM5r1K DJIIj!"rcU=(d SnU(aSO U(aSOSnU(aSO U(aSOSnURXU5 URXU5 URXU5 URXU5 UR XXd5 [ UR 5HCupxU[UR 5S- :Hn Sn X#(aSOS -U -n URXU S S 9 ME g ) akPrints 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. r~z\-z+-rz| r'z  |F)rxis_lastrN) r{rrrrrgr lenrU) rrr.rxrrryridxchild is_last_childindent child_prepends rrUNode.PrettyPrintbsmG2uTD!2wtDLS40l;# 5$$S<@3? . S/!33mfS9FBm  mUL/rrqrp)NTT)__name__ __module__ __qualname____firstlineno____doc__rsr{rrrrrrU__static_attributes__rrrrs1! 0(&T&#*#LrrN)Unknown)r __future__rrr functoolsrapitools.base.pyrgooglecloudsdk.core.resourcergooglecloudsdk.core.utilr sqlparser r r9rrrr%r0r@rErQrWrlobjectrrrrrsj<&'%9) . 2& -4(* Q2"MDgL6gLr