Turn ResultIter class into Python 3 iterator

Browse Source
This commit is contained in:
Stefan Wehrmeyer 2013-12-18 03:34:03 +01:00
parent 23e484cbed
commit 5d3e8b90bf
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dataset/persistence/util.py
View File

@ -31,14 +31,14 @@ class ResultIter(object):
def _next_rp(self):
try:
self.rp = self.result_proxies.next()
self.rp = next(self.result_proxies)
self.count += self.rp.rowcount
self.keys = self.rp.keys()
return True
except StopIteration:
return False
def next(self):
def __next__(self):
row = self.rp.fetchone()
if row is None:
if self._next_rp():
@ -48,5 +48,7 @@ class ResultIter(object):
raise StopIteration
return dict(zip(self.keys, row))
next = __next__
def __iter__(self):
return self