ability to add icons to the cluster label

pull/438/head
Georgi Ivanov 6 years ago committed by tessier
parent d891a7ecb4
commit f318707537

@ -53,7 +53,7 @@ class _Cluster:
except LookupError: except LookupError:
self._parent = None self._parent = None
def __enter__(self): def __enter__(self):
setcluster(self) setcluster(self)
return self return self
@ -77,14 +77,14 @@ class _Cluster:
def node(self, node: "Node") -> None: def node(self, node: "Node") -> None:
"""Create a new node.""" """Create a new node."""
self.nodes[node.nodeid] = node self.nodes[node.nodeid] = node
def remove_node(self, nodeid: str) -> None: def remove_node(self, nodeid: str) -> None:
del self.nodes[nodeid] del self.nodes[nodeid]
def subgraph(self, subgraph: "_Cluster") -> None: def subgraph(self, subgraph: "_Cluster") -> None:
"""Create a subgraph for clustering""" """Create a subgraph for clustering"""
self.subgraphs.append(subgraph) self.subgraphs.append(subgraph)
@property @property
def nodes_iter(self): def nodes_iter(self):
if self.nodes: if self.nodes:
@ -212,7 +212,7 @@ class Diagram(_Cluster):
setdiagram(self) setdiagram(self)
super().__enter__() super().__enter__()
return self return self
def __exit__(self, *args): def __exit__(self, *args):
super().__exit__(*args) super().__exit__(*args)
setdiagram(None) setdiagram(None)
@ -456,7 +456,7 @@ class Node(_Cluster):
@property @property
def nodeid(self): def nodeid(self):
return self._id return self._id
# TODO: option for adding flow description to the connection edge # TODO: option for adding flow description to the connection edge
def connect(self, node: "Node", edge: "Edge"): def connect(self, node: "Node", edge: "Edge"):
"""Connect to other node. """Connect to other node.

Loading…
Cancel
Save