adding iterator to FilterRowset class

ntt · Jan 25, 2012 · fe04062 · fe04062
1 parent 5092296
commit fe04062
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/util.py b/src/util.py
@@ -15,6 +15,7 @@
 import time
 import math
 import cPickle
+import _weakref
 
 from . import _os
 
@@ -295,6 +296,10 @@ def Sort(self, colname):
 		ret = Rowset(self.header, self.items.values(), self.RowClass)
 		return ret.Sort(colname)
 
+	def __iter__(self):
+		return (self[key] for key in self.iterkeys())
+
+
 
 class IndexedRowLists(dict):
 	__guid__ = 'util.IndexedRowLists'