o ooi54@sdZddlZddlZddlmZddlmZddlmZgdZded efd d Z d ed efd dZ d ed efddZ dededed efddZ dededed efddZ GdddZGdddZd ed efddZd ed efddZd ed!ed efd"d#ZGd$d%d%ejejZdS)&z"Create DOT code with method-calls.N)_tools)base)quoting) GraphSyntax DigraphSyntaxDotlinereturncC d|dS)zReturn comment header line.z//  )r r r @/home/ubuntu/.local/lib/python3.10/site-packages/graphviz/dot.pycomment  rnamecCr )zReturn DOT graph head line.zgraph { r rr r r graph_headrrcCr )zReturn DOT digraph head line.zdigraph rr rr r r digraph_headrrtailheadattrcCd|d||dS)z%Return DOT graph edge statement line. z -- r r rrrr r r graph_edgercCr)z'Return DOT digraph edge statement line.rz -> r r rr r r digraph_edge!rrc@ eZdZdZeeZeeZdS)rzDOT graph head and edge syntax.N) __name__ __module__ __qualname____doc__ staticmethodr_headr_edger r r rr& rc@r)rz!DOT digraph head and edge syntax.N) r r!r"r#r$rr%rr&r r r rr.r'rcCr )zReturn DOT subgraph head line.z subgraph rr rr r rsubgraph6rr(cCs |dS)z$Return plain DOT subgraph head line.rr rr r rsubgraph_plain;s r)leftrightcCsd||dS)zReturn DOT node statement line.rr r )r*r+r r rnode@sr,c s*eZdZUdZeed<eeZede de fddZ e de de fddZ d Z eeZeeZeeZZe d e de fd d Zed e de de de fddZe d e de de fddZdddddddddeje deje deddffddZfddZejdddd8d eddfd!d"Zejdddd8d#edeje fd$d%Z ejd&de!d'  d9de d(eje ddfd)d*Zejd+de!d'  d9d,e d-e d(eje ddfd.d/Z"d:d0d1Z#ejd2de!d'  d9d3eje ddfd4d5Z$ejd2dd     d;deje deje fd6d7ZZ%S)r?) selfrrr;r<r=r>r?kwargs __class__r rrAlsz Dot.__init__c s8tj|j|jt|jt|jt|jt|j |j dS)z3Return the kwargs to create a copy of the instance.r:) r@ _copy_kwargsrrrBr;r<r=rCr>r?)rDrErFr rrHs zDot._copy_kwargsrrD)supported_number ignore_arg keep_attrscCs0|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)r;r<r=clearr>)rDrKar r rrLs z Dot.clearr(ccs|jr ||jV|r|jrtd|jr|jn|j}n |jr$|jn|j}||jr3| |jdndVdD]}t ||d}|rQ| ||j d|dVq9|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 r6graphr,edger7N)rE)r_commentr? ValueErrorr _subgraph_subgraph_plainr4r%_quotegetattrr7 _attr_listr>_tail)rDr(rkwattrsr r r__iter__s    z Dot.__iter__)rIrJcategorylabelcKs6||}|j|||d}|||}|j|dS)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. rE attributesN)rVrX_noder>append)rDrr_ _attributesr[ attr_listr r r rr,s  zDot.node tail_name head_namecKsD||}||}|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 `. 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. r`rN) _quote_edgerXr&r>rc)rDrgrhr_rdr[rer r r rrQs  zDot.edgecs.|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 ).0thrQquoter r szDot.edges..N)r9rir>)rDtail_head_iterr rmredgess z Dot.edgesrrZcKs~|dur|dvrtd||s|r=|dur'|jd||d}||}n|jd||d}|||}|j|dSdS)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 `. NrOz1attr statement must target graph, node, or edge: r`)lowerrS_a_listr8rXr7r>rc)rDrZrdr[a_listr rer r rrs   zDot.attrc s|dur0|}|dd|j||||||dd|jdi|tjfdd} | |dS||||||g} tdd| DsEtd |j|jkrTt|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. Nfilenamer:c3sV|dS)z*Return subgraph and add to parent on exit.Nr(parentrvr rsubgraph_contextmanagerKsz-Dot.subgraph..subgraph_contextmanagerrwcss|]}|duVqdS)Nr )rjrMr r r TszDot.subgraph..z)graph must be sole argument of subgraph()z( cannot add subgraph of different kind: cSsg|]}d|qS)rr )rjr r r rro[sz Dot.subgraph..Trvr ) rHpopupdaterG contextlibcontextmanagerallrSr-r>r\) rDrPrrr;r<r=r>rEryargsr rvrr(s&'   $z Dot.subgraph)F)NN)r N)NNNNNNN)&r r!r"r#bool__annotations__r$rrRstrr% classmethodr4rYr(rTr)rUr,rbr7r8r&r9typingOptionalrArHrdeprecate_positional_argsrLIteratorr\DeprecationWarningrQrqr __classcell__r r rFrrEs         r)r#r}rr6rrr__all__rrrrrrrrr(r)r,QuoteBaserr r r rs$