retoor/dataset
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix infinite recursion

Browse Source
This commit is contained in:
Di Wu
2015-12-20 13:49:07 +08:00
committed by Di Wu
parent 0f7e7dc73c
commit b8980221cd
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dataset/persistence/util.py
+7 -6
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: chunk = self.result_proxy.fetchall()
self._iter = iter(self.result_proxy.fetchall()) else:
else: chunk = self.result_proxy.fetchmany(self.step)
self._iter = iter(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):