From dbc836d74a11fa355c0ae0038f9d4ffdcea3edd8 Mon Sep 17 00:00:00 2001 From: Darcy LF Date: Mon, 9 Aug 2021 17:25:02 +1000 Subject: [PATCH] :sparkles: Add node.set_size() and node.size() --- pynode_next/node.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) 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