introduce PEP 479 support to native driver

xzkostyan · Aug 29, 2018 · f3bb4da · f3bb4da
1 parent 2c27d20
commit f3bb4da
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/clickhouse_sqlalchemy/drivers/native/connector.py b/clickhouse_sqlalchemy/drivers/native/connector.py
@@ -217,16 +217,12 @@ def setoutputsize(self, size, column=None):
         pass
 
     # Iteration support.
-    def __next__(self):
-        one = self.fetchone()
-        if one is None:
-            raise StopIteration
-        return one
-
-    next = __next__
-
     def __iter__(self):
-        return self
+        while True:
+            one = self.fetchone()
+            if one is None:
+                return
+            yield one
 
     def _process_response(self, response, context):
         if not response: