U d/e0@sdZddlZddlZddlmZddlmZddlmZddd gZeed d d Z eed ddZ eed ddZ eeeedddZ eeeedddZ GdddZGdddZeed ddZeed ddZeeedddZGd d d ejejZdS)!z"Create DOT code with method-calls.N)_tools)base)quoting GraphSyntax DigraphSyntaxDot)linereturncCs d|dS)zReturn comment header line.z//  )r r r 0/tmp/pip-unpacked-wheel-n8ok7rre/graphviz/dot.pycomment srnamer cCs d|dS)zReturn DOT graph head line.zgraph { r rr r r graph_headsrcCs d|dS)zReturn DOT digraph head line.zdigraph rr rr r r digraph_headsrtailheadattrr cCsd|d||dS)z%Return DOT graph edge statement line. z -- r r rrrr r r graph_edgesrcCsd|d||dS)z'Return DOT digraph edge statement line.rz -> r r rr r r digraph_edge!src@s eZdZdZeeZeeZdS)rzDOT graph head and edge syntax.N) __name__ __module__ __qualname____doc__ staticmethodr_headr_edger r r r r&sc@s eZdZdZeeZeeZdS)rz!DOT digraph head and edge syntax.N) rrrr r!rr"rr#r r r r r.scCs d|dS)zReturn DOT subgraph head line.z subgraph rr rr r r subgraph6sr$cCs |dS)z$Return plain DOT subgraph head line.rr rr r r subgraph_plain;sr%)leftrightr cCsd||dS)zReturn DOT node statement line.rr r )r&r'r r r node@sr(cseZdZUdZeed<eeZee e dddZ e e e dddZ dZ eeZeeZeeZZe e e d d d Zee e e e d d dZe e e e dddZddddddddeje eje eddfddZfddZejddd4eddddZejddd5eeje dd d!Z ejd"dd6e eje dd#d$d%Zejd&dd7e e eje dd'd(d)Z!dd*d+d,Z"ejd-dd8eje dd.d/d0Z#ejd-dd9eje eje d1d2d3ZZ$S):rzAssemble DOT source code.directedrcCs tddS)zReturn DOT head line.*must be implemented by concrete subclassesNNotImplementedErrorrr r r r"Lsz Dot._headcCsd||S)zReturn DOT strict head line.zstrict )r")clsrr r r _head_strictQszDot._head_strictz} )r&r cCs ||dS)N)_attr)r-r&r r r _attr_plain^szDot._attr_plainrcCs tddS)zReturn DOT edge statement line.r*Nr+rr r r r#bsz Dot._edge)rrr cCs|j||ddS)z%Return plain DOT edge statement line.r/r)r#)r-rrr r r _edge_plaingszDot._edge_plainNFrr graph_attr node_attr edge_attrbodystrict)rrr8r c  s|tjf|||_||_|dk r*t|ni|_|dk r@t|ni|_|dk rVt|ni|_|dk rlt|ng|_ ||_ dSN) super__init__rrdictr4r5r6listr7r8) selfrrr4r5r6r7r8kwargs __class__r r r;lsz Dot.__init__c s8tj|j|jt|jt|jt|jt|j |j dS)z3Return the kwargs to create a copy of the instance.r3) r: _copy_kwargsrrr<r4r5r6r=r7r8)r>r?r@r r rBs zDot._copy_kwargsr)Zsupported_number) keep_attrsr cCs0|s"|j|j|jfD] }|q|jdS)zReset content to an empty body, clear graph/node/egde_attr mappings. Args: keep_attrs (bool): preserve graph/node/egde_attr mappings N)r4r5r6clearr7)r>rCar r r rDs z Dot.clear)r$r ccs|jr||jV|r:|jr&td|jr2|jn|j}n|jrF|jn|j}||jrd| |jdndVdD]0}t ||d}|rp| ||j d|dVqp|j EdH|jVdS)zYield the DOT source code line by line (as graph or subgraph). Yields: Line ending with a newline (``'\n'``). zsubgraphs cannot be strict r/graphr(edger0N)r?)r_commentr8 ValueErrorr _subgraph_subgraph_plainr.r"_quotegetattrr0 _attr_listr7_tail)r>r$rkwattrsr r r __iter__s  z Dot.__iter__)rlabelr cKs6||}|j|||d}|||}|j|dS)zCreate 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). r? attributesN)rNrP_noder7append)r>rrV _attributesrS attr_listr r r r r(s  zDot.node) tail_name head_namerVr cKsD||}||}|j|||d}|j|||d}|j|dS)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 `. rWrN) _quote_edgerPr#r7rZ)r>r^r_rVr[rSr\r r r r rIs   zDot.edge)r cs.|j|j|jfdd|D7_dS)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 `. cs$g|]\}}||dqS))rrr ).0thrIquoter r szDot.edges..N)r2r`r7)r>Ztail_head_iterr rdr edgess z Dot.edges)rRr cKsz|dk r"|dkr"td||s*|rv|dkrN|jd||d}||}n|jd||d}|||}|j|dS)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 `. NrGz1attr statement must target graph, node, or edge: rW)lowerrKZ_a_listr1rPr0r7rZ)r>rRr[rSa_listr r\r r r rs   zDot.attr)rrc s|dkr\|}|dd|j||||||dd|jf|tjfdd} | |dS||||||g} tdd| Dstd |j|jkrt|d ||j d d |j d dD7_ dS)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. Nfilenamer3c3sV|dS)z*Return subgraph and add to parent on exit.Nr$parentrlr r subgraph_contextmanager<sz-Dot.subgraph..subgraph_contextmanagerrmcss|]}|dkVqdSr9r )rarEr r r EszDot.subgraph..z)graph must be sole argument of subgraph()z( cannot add subgraph of different kind: cSsg|]}d|qS)rr )rar r r r rfLsz Dot.subgraph..Trl) rBpopupdaterA contextlibcontextmanagerallrKr)r7rT) r>rHrrr4r5r6r7r?roargsr rlr r$ s('    z Dot.subgraph)F)F)NN)NN)NN)NNNNNNN)%rrrr bool__annotations__r!rrJstrr" classmethodr.rQr$rLr%rMr(rYr0r1r#r2typingOptionalr;rBrZdeprecate_positional_argsrDIteratorrTrIrgr __classcell__r r r@r rEs         )r rsr{r/rrr__all__ryrrrrrrrr$r%r(ZQuoteZBaserr r r r s"