add extras execute_batch

aio-libs · Dec 6, 2019 · ccd5a46 · ccd5a46
1 parent 17b5e4c
commit ccd5a46
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/aiopg/extras.py b/aiopg/extras.py
@@ -7,15 +7,10 @@ def _paginate(seq, page_size):
     count = len(seq)
     it = iter(seq)
     for s in range(count + 1):
-        try:
-            for i in range(page_size):
-                page.append(next(it))
-            yield page
-            page = []
-        except StopIteration:
-            if page:
-                yield page
-            return
+        for i in range(page_size):
+            page.append(next(it))
+        yield page
+        page = []
 
 
 async def execute_batch(cur, sql, argslist, page_size=100):