fix infinite recursion

Browse Source
This commit is contained in:
Di Wu 2015-12-20 13:32:09 +08:00 committed by Di Wu
parent 0f7e7dc73c
commit b8980221cd
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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