File size: 4,425 Bytes
d015b2a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
r"""Assemble DOT source code objects.
Example:
>>> doctest_mark_exe()
>>> import graphviz
>>> dot = graphviz.Graph(comment='Mønti Pythøn ik den Hølie Grailen')
>>> dot.node('Møøse')
>>> dot.node('trained_by', 'trained by')
>>> dot.node('tutte', 'TUTTE HERMSGERVORDENBROTBORDA')
>>> dot.edge('Møøse', 'trained_by')
>>> dot.edge('trained_by', 'tutte')
>>> dot.node_attr['shape'] = 'rectangle'
>>> print(dot.source) #doctest: +NORMALIZE_WHITESPACE
// Mønti Pythøn ik den Hølie Grailen
graph {
node [shape=rectangle]
"Møøse"
trained_by [label="trained by"]
tutte [label="TUTTE HERMSGERVORDENBROTBORDA"]
"Møøse" -- trained_by
trained_by -- tutte
}
>>> dot.render('doctest-output/m00se.gv').replace('\\', '/')
'doctest-output/m00se.gv.pdf'
"""
import typing
from .encoding import DEFAULT_ENCODING
from . import _tools
from . import dot
from . import jupyter_integration
from . import piping
from . import rendering
from . import unflattening
__all__ = ['Graph', 'Digraph']
class BaseGraph(dot.Dot,
rendering.Render,
jupyter_integration.JupyterIntegration, piping.Pipe,
unflattening.Unflatten):
"""Dot language creation and source code rendering."""
@_tools.deprecate_positional_args(supported_number=2)
def __init__(self, name: typing.Optional[str] = None,
comment: typing.Optional[str] = None,
filename=None, directory=None,
format: typing.Optional[str] = None,
engine: typing.Optional[str] = None,
encoding: typing.Optional[str] = DEFAULT_ENCODING,
graph_attr=None, node_attr=None, edge_attr=None,
body=None,
strict: bool = False, *,
renderer: typing.Optional[str] = None,
formatter: typing.Optional[str] = None) -> None:
if filename is None and name is not None:
filename = f'{name}.{self._default_extension}'
super().__init__(name=name, comment=comment,
graph_attr=graph_attr,
node_attr=node_attr, edge_attr=edge_attr,
body=body, strict=strict,
filename=filename, directory=directory,
encoding=encoding,
format=format, engine=engine,
renderer=renderer, formatter=formatter)
@property
def source(self) -> str:
"""The generated DOT source code as string."""
return ''.join(self)
class Graph(dot.GraphSyntax, BaseGraph):
"""Graph source code in the DOT language.
Args:
name: Graph name used in the source code.
comment: Comment added to the first line of the source.
filename: Filename for saving the source
(defaults to ``name`` + ``'.gv'``).
directory: (Sub)directory for source saving and rendering.
format: Rendering output format (``'pdf'``, ``'png'``, ...).
engine: Layout command used (``'dot'``, ``'neato'``, ...).
renderer: Output renderer used (``'cairo'``, ``'gd'``, ...).
formatter: Output formatter used (``'cairo'``, ``'gd'``, ...).
encoding: Encoding for saving the source.
graph_attr: Mapping of ``(attribute, value)`` pairs for the graph.
node_attr: Mapping of ``(attribute, value)`` pairs set for all nodes.
edge_attr: Mapping of ``(attribute, value)`` pairs set for all edges.
body: Iterable of verbatim lines (including their final newline)
to add to the graph ``body``.
strict (bool): Rendering should merge multi-edges.
Note:
All parameters are `optional` and can be changed under their
corresponding attribute name after instance creation.
"""
@property
def directed(self) -> bool:
"""``False``"""
return False
class Digraph(dot.DigraphSyntax, BaseGraph):
"""Directed graph source code in the DOT language."""
if Graph.__doc__ is not None:
__doc__ += Graph.__doc__.partition('.')[2]
@property
def directed(self) -> bool:
"""``True``"""
return True
|