"SrSSKJr SSKJr SSKJr SSKJr SSKr"SS\5r S r S r S r S r SS jrSrSSjrSrSSjrg)zLCode to transform the (cleaned-up) description of a dataflow into Graphviz. )absolute_import)division)unicode_literals) exceptionsNcR\rSrSrSrSrSrSrSrSr Sr SS jr S r S r S rg )_Clustera Encapsulation of a single cluster in the final Step-Graph. The cluster hierarchy represents pieces of the user_name. A given cluster is either a leaf (it contains a single step and no sub-clusters) or a transform (it contains no step and one or more sub-clusters). c80UlXlX lSUlgN)_Cluster__children_Cluster__parent_Cluster__name_in_parent_Cluster__step)selfparentname_in_parents 1lib/googlecloudsdk/api_lib/dataflow/step_graph.py__init___Cluster.__init__"sDOM*DKc$UR(+$)zVA leaf cluster contains no sub-clusters. Returns: True iff this is a leaf cluster. )r rs rIsLeaf_Cluster.IsLeaf(s  rc2[UR5S:H$)zkA singleton is any cluster that contains a single child. Returns: True iff this is a singleton cluster. )lenr rs r IsSingleton_Cluster.IsSingleton0s t 1 $$rc$UR(+$)zVDetermine if this cluster is the root. Returns: True iff this is the root cluster. )r rs rIsRoot_Cluster.IsRoot8s }} rcUR$)zlReturn the step for this cluster. Returns: The step for this cluster. May be None if this is not a leaf. )rrs rGetStep_Cluster.GetStep@s ;;rc\UR(aeUR(aeXlg)zSet the step for this cluster. Can only be called on leaf nodes that have not yet had their step set. Args: step: The step that this cluster represents. N)r r)rsteps rSetStep_Cluster.SetStepHs" {{?KrNcUR(aURU:Xa UR$URRU5nU(aUS-UR-$UR$)zReturn the name of this sub-cluster relative to the given ancestor. Args: relative_to: The ancestor to output the name relative to. Returns: The string representing the user_name component for this cluster. /)r rName)r relative_to parent_names rr, _Cluster.NameTs[ MMt}} ;  " ""--$$[1K 3 !6!6 66  " ""rcUR(aeXR;a[X5URU'URU$)zReturn the cluster representing the given piece_name within this cluster. Args: piece_name: String representing the piece name of the desired child. Returns: Cluster representing the child. )rr r)r piece_names r GetOrAddChild_Cluster.GetOrAddChildfs<{{?($,T$>dooj! ??: &&rcT[[R"UR55$)z\Return the sub-clusters. Returns: Sorted list of pairs for the children in this cluster. )sortedsix iteritemsr rs rChildren_Cluster.Childrenss #--0 11r) __children__name_in_parent__parent__stepr )__name__ __module__ __qualname____firstlineno____doc__rrrr!r$r(r,r2r8__static_attributes__rrrrs4 % #$ '2rrcZSn/n/nURS5HwnXRS5URS5- - nURU5 US:Xa'URSRU55 USS2 MfURS5 My U(aURU5 U$)aAGiven a user name for a step, split it into the individual pieces. Examples: _SplitStep('Transform/Step') = ['Transform', 'Step'] _SplitStep('Read(gs://Foo)/Bar') = ['Read(gs://Foo)', 'Bar'] Args: user_name: The full user_name of the step. Returns: A list representing the individual pieces of the step name. rr+()N)splitcountappendjoin) user_nameparensaccum step_partspieces r _SplitSteprR|s & %*s#e kk#S!111F LL {' ( ll3$ e rc[SS5nUHQn[USRSUS55nUnUHnURU5nM UR U5 MS U$)a Extract a hierarchy from the steps provided. The `step graph' is constructed as follows: 1. Every node has a `name'. This is flat, something like "s1", "s100". 2. Each node can depend on others. These edges are specified by "name". 3. Each node can also have a user_name, like "Foo/Bar". This name creates a hierarchy of subgraphs (eg., Foo/Bar and Foo/Baz are in the same cluster). Args: steps: A list of steps from the Job message. Returns: A Cluster representing the root of the step hierarchy. NrH propertiesrMname)rrRgetr2r()stepsrootr' step_pathnoderQs r_UnflattenStepsToClustersr[sl $ $d4 -11+tF|LMI D    &dLL  +rcURS5(a%[R"SRU55eSRUR SS55$)aEscape a string for use as in Graphviz. Args: name: The string to escape. Returns: The `name', with double-quotes escaped, and quotes around it. Raises: exceptions.UnsupportedNameException: If the name is incompatible with Graphviz ID escaping. \z0Unsupported name for Graphviz ID escaping: {0!r}z"{0}""z\")endswithrUnsupportedNameExceptionformatreplace)rUs r_EscapeGraphvizIdrcsP ]]4  - -:AA$G II  S%0 11rzO{name} [label={user_name}, tooltip={full_name}, style=filled, fillcolor=white];c ## UR5(aaUR5n[R[ US5[ UR 55[ UR US95S9v gUR 5(dUR5(a.UR5Hup4[XAS9HnUv M M gUR 5nSR[ SU-55v Sv Sv S v S R[ U55v S R[ UR U555v UR5Hup7[XpS9HnUv M M S v g7f) NrU)r-)rU full_namerM)rzsubgraph {0} {{zcluster z style=filled;zbgcolor=white;zlabeljust=left;z tooltip={0};z label={0};}) rr$ _NODE_FORMATrarcr,rr!r8_YieldGraphvizClusters)clusterrr' unused_key subclusterlineresubgroups rrhrhsQ ^^ ?? D    tF| ,#GLLN3#GLLVL$DE  GG 0 0")"2"2"4 (C$ D#5 I  " "#4Z)5K#L MM       1) < ==   / V0DE FF ' 0 0 2 (B$ C!3 IsFFzF{edge_source} -> {edge_dest} [taillabel={edge_output}, style={style}];cp[R[US5[U5[US5US9$)a-Returns an edge from the output referred to by output_ref to step_name. Args: step_name: String identifying the step with the dependency. output_ref: Output-reference dictionary to start the edge at. style: The style for the edge. Returns: A string representing the edge in Graphviz format. step_name output_name) edge_source edge_dest edge_outputstyle) _EDGE_FORMATrarc)ro output_refrts r _GraphvizEdgerwsA   #J{$;<!),#J}$=>   rc#&# USnUSRSS5nU(a [X5v USRS/5Hn[X5v M USRS05R5Hn[XSS9v M g7f) zOutput Graphviz edges for the given step. Args: step: Step to write edges for. Yields: The Graphviz edge lines for the given step. rUrTparallel_inputNinputsnon_parallel_inputsdashed)rt)rVrwvalues)r'ro par_input other_input side_inputs r_YieldGraphvizEdgesrs6l)< $$%5t<)  --,'++Hb9k  //:&**+@"ELLNj  X >>OsBBc## SR[U=(d S5S9v [U5n[U5HnUv M Sv UHn[ U5HnUv M M Sv g7f)aGiven a root cluster produce the Graphviz DOT format. No attempt is made to produce `pretty' output. Args: steps: A list of steps from the Job message. graph_name: The name of the graph to output. Yields: The lines representing the step-graph in Graphviz format. zstrict digraph {graph_name} {{G) graph_namerHrfN)rarcr[rhr)rWrrXrlr's r YieldGraphvizrsx )//":#45 0 77 #5 )$$T*d J+ (d#D) j* )sA-A/r )solid)rB __future__rrrgooglecloudsdk.api_lib.dataflowrr6objectrrRr[rcrgrhrurwrrrDrrrs^''6 _2v_2D>42(( 2< $?, r