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

13
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: 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):