ci54@dZddlZddlZddlmZddlmZddlmZgdZded efd Z d ed efd Z d ed efd Z dededed efdZ dededed efdZ GddZGddZd ed efdZd ed efdZdeded efdZGddej&ej(Zy)z"Create DOT code with method-calls.N)_tools)base)quoting) GraphSyntax DigraphSyntaxDotlinereturncd|dS)zReturn comment header line.z//  )r s 3/tmp/pip-target-z3e9_cxr/lib/python/graphviz/dot.pycommentr s b>namecd|dS)zReturn DOT graph head line.zgraph { rrs r graph_headrs D6 rcd|dS)zReturn DOT digraph head line.zdigraph rrrs r digraph_headrs dV4  rtailheadattrcd|d||dS)z%Return DOT graph edge statement line. z -- r rrrrs r graph_edger vT$vR ((rcd|d||dS)z'Return DOT digraph edge statement line.rz -> r rrs r digraph_edger"!r rc0eZdZdZeeZeeZy)rzDOT graph head and edge syntax.N) __name__ __module__ __qualname____doc__ staticmethodr_headr_edgerrrrr&s)  $E  $Errc0eZdZdZeeZeeZy)rz!DOT digraph head and edge syntax.N) r$r%r&r'r(rr)r"r*rrrrr.s+  &E  &Errcd|dS)zReturn DOT subgraph head line.z subgraph rrrs rsubgraphr-6s tfD !!rc |dS)z$Return plain DOT subgraph head line.rrrs rsubgraph_plainr/;sV4=rleftrightcd||dS)zReturn DOT node statement line.rr r)r0r1s rnoder3@s veWB rc eZdZUdZeed<eeZede de fdZ e de de fdZ dZ eeZeeZeexZZe de de fd Zed e d e d e de fd Ze d e d e de fdZdddddddddej2e dej2e deddffdZfdZej:ddd*deddfdZej:ddd*dedej>e fdZ ej:dde! d+de dej2e ddfd Zej:d!de! d+d"e d#e dej2e ddfd$Z"d,d%Z#ej:d&de! d+d'ej2e ddfd(Z$ej:d&d d-dej2e dej2e fd)ZxZ%S).r zAssemble DOT source code.directedrr ctd)zReturn DOT head line.*must be implemented by concrete subclassesNotImplementedErrorrs rr)z Dot._headL""NOOrc*d|j|S)zReturn DOT strict head line.zstrict )r))clsrs r _head_strictzDot._head_strictQs4)**rz} r0c&|j|dS)N)_attr)r<r0s r _attr_plainzDot._attr_plain^syyr""rrrrctd)zReturn DOT edge statement line.r7r8rs rr*z Dot._edgebr:rc*|j||dS)z%Return plain DOT edge statement line.r?r)r*)r<rrs r _edge_plainzDot._edge_plaingsyydBy77rNFrr graph_attr node_attr edge_attrbodystrictrrJc t |di|||_ ||_ | t |ni|_ | t |ni|_ | t |ni|_ | t|ng|_ ||_ y)Nr) super__init__rrdictrFrGrHlistrIrJ) selfrrrFrGrHrIrJkwargs __class__s rrMz Dot.__init__ls "6" T @.8.D$z*"R,5,AirR,5,AirR"&"2DJ S 7rc t||j|jt |j t |j t |jt|j|jS)z3Return the kwargs to create a copy of the instance.rE) rL _copy_kwargsrrrNrFrGrHrOrIrJ)rPrQrRs rrTzDot._copy_kwargss]w#,0LL/3DOO/D.24>>.B.24>>.B)-dii+/;; $8 8rrrP)supported_number ignore_arg keep_attrsc|s8|j|j|jfD]}|j|jjy)zReset content to an empty body, clear graph/node/egde_attr mappings. Args: keep_attrs (bool): preserve graph/node/egde_attr mappings N)rFrGrHclearrI)rPrWas rrYz Dot.clears@oot~~t~~F    rr-c#NK|jr|j|j|r<|jr td|jr |j n |j }n$|jr |jn |j}||jr|j|jdznddD]9}t||d}|s|j||jd|;|jEd{|jy7w)zYield the DOT source code line by line (as graph or subgraph). Yields: Line ending with a newline (``'\n'``). zsubgraphs cannot be strict r?graphr3edger@N)rQ)r_commentrJ ValueErrorr _subgraph_subgraph_plainr=r)_quotegetattrr@ _attr_listrI_tail)rPr-rkwattrss r__iter__z Dot.__iter__s <<-- - - {{ !=>>%)YY4>>D4H4HD(, 4$$D4994;;tyy)C/"EE+ JBDRD,/EjjT__T%_%HII J 99jj sCD%6D%D#D%)rUrVcategorylabelc |j|}|j|||}|j||}|jj |y)aCreate a node. Args: name: Unique identifier for the node inside the source. label: Caption to be displayed (defaults to the node ``name``). attrs: Any additional node attributes (must be strings). Attention: When rendering ``label``, backslash-escapes and strings of the form ``<...>`` have a special meaning. See the sections :ref:`backslash-escapes` and :ref:`quoting-and-html-like-labels` in the user guide for details. rQ attributesN)rdrf_noderIappend)rPrrm _attributesri attr_listr s rr3zDot.nodesH${{4 OOE%KOP zz$ * r tail_name head_namec |j|}|j|}|j|||}|j|||}|jj |y)aCreate an edge between two nodes. Args: tail_name: Start node identifier (format: ``node[:port[:compass]]``). head_name: End node identifier (format: ``node[:port[:compass]]``). label: Caption to be displayed near the edge. attrs: Any additional edge attributes (must be strings). Note: The ``tail_name`` and ``head_name`` strings are separated by (optional) colon(s) into ``node`` name, ``port`` name, and ``compass`` (e.g. ``sw``). See :ref:`details in the User Guide `. Attention: When rendering ``label``, backslash-escapes and strings of the form ``<...>`` have a special meaning. See the sections :ref:`backslash-escapes` and :ref:`quoting-and-html-like-labels` in the user guide for details. rorN) _quote_edgerfr*rIrr)rPrvrwrmrsrirtr s rr_zDot.edges^6$$Y/ $$Y/ OOE%KOP zzyyyzI rc |j}|j}|xj|Dcgc]\}}|||||c}}z c_ycc}}w)aCreate a bunch of edges. Args: tail_head_iter: Iterable of ``(tail_name, head_name)`` pairs (format:``node[:port[:compass]]``). Note: The ``tail_name`` and ``head_name`` strings are separated by (optional) colon(s) into ``node`` name, ``port`` name, and ``compass`` (e.g. ``sw``). See :ref:`details in the User Guide `. )rrN)rDryrI)rPtail_head_iterr_quoteths redgesz Dot.edgessV   "02!QauQx82 2 2s A rrhc $| |jdvrtd||s|rj|&|jd||}|j|}n&|j d||}|j ||}|j j|yy)aBAdd a general or graph/node/edge attribute statement. Args: kw: Attributes target (``None`` or ``'graph'``, ``'node'``, ``'edge'``). attrs: Attributes to be set (must be strings, may be empty). See the :ref:`usage examples in the User Guide `. Nr]z1attr statement must target graph, node, or edge: ro)lowerra_a_listrArfr@rIrr)rPrhrsria_listr rts rrzDot.attrs >bhhj0II!!#() ) Kzd5[Q''/ OOD;OW zz"i0 II  T " rc  |m|j}|jdd|j||||||d|jd i| tj fd} | |S||||||g} t d| Ds td|j|jk7rt|d||xj|jd D cgc]} d |  c} z c_ ycc} w) aAdd the current content of the given sole ``graph`` argument as subgraph or return a context manager returning a new graph instance created with the given (``name``, ``comment``, etc.) arguments whose content is added as subgraph when leaving the context manager's ``with``-block. Args: graph: An instance of the same kind (:class:`.Graph`, :class:`.Digraph`) as the current graph (sole argument in non-with-block use). name: Subgraph name (``with``-block use). comment: Subgraph comment (``with``-block use). graph_attr: Subgraph-level attribute-value mapping (``with``-block use). node_attr: Node-level attribute-value mapping (``with``-block use). edge_attr: Edge-level attribute-value mapping (``with``-block use). body: Verbatim lines to add to the subgraph ``body`` (``with``-block use). See the :ref:`usage examples in the User Guide `. When used as a context manager, the returned new graph instance uses ``strict=None`` and the parent graph's values for ``directory``, ``format``, ``engine``, and ``encoding`` by default. Note: If the ``name`` of the subgraph begins with ``'cluster'`` (all lowercase) the layout engine will treat it as a special cluster subgraph. NfilenamerEc38K|jyw)z*Return subgraph and add to parent on exit.Nr-)parentr-s rsubgraph_contextmanagerz-Dot.subgraph..subgraph_contextmanagerKs)s)rc3$K|]}|du yw)Nr).0rZs r zDot.subgraph..Ts+19+sz)graph must be sole argument of subgraph()z( cannot add subgraph of different kind: Trrr) rTpopupdaterR contextlibcontextmanagerallrar5rIrj) rPr^rrrFrGrHrIrQrargsr r-s @rr-z Dot.subgraphsN =&&(F JJz4 ( MMtW%/9PY#D  2&t~~//H  & & *' * +$7 7gz9iF+d++HI I >>T]] *x(!!& +, , enndn.KLd4&kLL Ls C6)F)NN)r N)NNNNNNN)&r$r%r&r'bool__annotations__r(rr`strr) classmethodr=rgr-rbr/rcr3rqr@rAr*rDtypingOptionalrMrTrdeprecate_positional_argsrYIteratorrjDeprecationWarningr_rr __classcell__)rRs@rr r Es#NG$HPCPCPP++++ EX&I">2O &&EE#s#s##PsP#PSPSPP8#8S8S88 /315 DDt % 8s+8 //#.8 859 888&V%%qVL   M &V%%qVL&//#2FM2&V%%qV/AC,0OOC(+/C*&V%%qV/AC,0ccOOC(+/C>2&&V%%qV/AC.2#vs+#+/#C#.&V%%qVL!.215<@ >Ms+>M //#.>MM>Mrr )r'rrr?rrr__all__rrrrrr"rrr-r/r3QuoteBaser rrrrs(  1## SS !s!s! ))3)c)c) )#)S))) %%''"3"3"   s 3 3 VM'--VMr