makes tree iterable

1 files changed, 3 insertions, 0 deletions

diff --git a/tree.py b/tree.py
index 4fcb6c6..d8cfffe 100644
--- a/tree.py
+++ b/tree.py
@@ -43,6 +43,9 @@ class Node(object):
         self.parent = None
         self.children = []
 
+    def __iter__(self):
+        return iter(self.children)
+
     def __str__(self, indent=0):
         s = ['%s%s %s' % (' '*indent, self.__class__.__name__, self.id)]
         indent += 2