From c5df534b7acd4681ff675ab74c0d469c125d79e5 Mon Sep 17 00:00:00 2001 From: Fredrik Unger Date: Wed, 6 Apr 2011 22:51:03 +0200 Subject: [PATCH] Adding a simple iterator to the Trie class, now the set operations on the links in the tree works again. --- src/tree-cutter.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/tree-cutter.py b/src/tree-cutter.py index 4c0f019..0b72e6b 100755 --- a/src/tree-cutter.py +++ b/src/tree-cutter.py @@ -193,6 +193,15 @@ class Trie(): def __init__(self): self._root = [] + def __iter__(self): + return self.inorder(self._root) + + def inorder(self,t): + for l in t: + yield l.value() + for ch in l.children(): + self.inorder(ch) + def _add(self,trie, key, content): # is the key a leaf k = key.pop(0) -- 2.30.2