diff --git a/pynode_next/node.py b/pynode_next/node.py index 54370c8..c6e879f 100644 --- a/pynode_next/node.py +++ b/pynode_next/node.py @@ -18,7 +18,7 @@ def __init__(self, id=None, value=None): self._incident_edges = [] self._color = Color.DARK_GREY - + self._size = 12 self._attrs = {} self._labels = {} @@ -36,6 +36,16 @@ def value(self): """Returns the node's value.""" return self._value + def set_size(self, size=12): + """Sets the size of a node""" + self._size = size + core.ax(lambda x: x.node(self._id).size(size)) + return self + + def size(self): + """Returns the node's size""" + return self._size + def set_color(self, color=Color.DARK_GREY): """Sets the node's color to the Color object specified.""" self._color = color