Merge pull request #145 from twds/master

Browse Source 
fix infinite recursion
This commit is contained in:
Friedrich Lindenberg 2015-12-20 10:43:12 +01:00
commit 9d0b6503d7
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
dataset/persistence/util.py
View File

@ -50,13 +50,14 @@ class ResultIter(object):
self._iter = None
def _next_chunk(self):
try:
if not self.step:
self._iter = iter(self.result_proxy.fetchall())
chunk = self.result_proxy.fetchall()
else:
self._iter = iter(self.result_proxy.fetchmany(self.step))
chunk = self.result_proxy.fetchmany(self.step)
if chunk:
self._iter = iter(chunk)
return True
except StopIteration:
else:
return False
def __next__(self):