From b8980221cd43159e78a5f4453fae5452cc4cbf8f Mon Sep 17 00:00:00 2001 From: Di Wu Date: Sun, 20 Dec 2015 13:32:09 +0800 Subject: [PATCH] fix infinite recursion --- dataset/persistence/util.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/dataset/persistence/util.py b/dataset/persistence/util.py index 924b49c..c1a89c0 100644 --- a/dataset/persistence/util.py +++ b/dataset/persistence/util.py @@ -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):