Merge pull request #52 from uber/iterator

fix condition when the query returns 0 row
vertica · Apr 28, 2015 · 2e19d49 · 2e19d49 · zer0n · Apr 29, 2015
2 parents ff05b8f + 9c62aa4
commit 2e19d49
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vertica_python/vertica/cursor.py b/vertica_python/vertica/cursor.py
@@ -76,7 +76,9 @@ def execute(self, operation, parameters=None):
         # Read until data rows or end of stream
         while True:
             message = self.connection.read_message()
-            if isinstance(message, messages.DataRow) or isinstance(message, messages.ReadyForQuery):
+            if isinstance(message, messages.DataRow) \
+                    or isinstance(message, messages.CommandComplete) \
+                    or isinstance(message, messages.ReadyForQuery):
                 self._message = message  # cache the message because there's no way to undo the read
                 break
             else: