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 self._iter = None
def _next_chunk(self): def _next_chunk(self):
try:
if not self.step: if not self.step:
self._iter = iter(self.result_proxy.fetchall()) chunk = self.result_proxy.fetchall()
else: 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 return True
except StopIteration: else:
return False return False
def __next__(self): def __next__(self):