Merge pull request #57 from uber/bugfix

Fix incomplete processing for write queries
vertica · May 4, 2015 · 6b98001 · 6b98001
2 parents 89b60a6 + c573588
commit 6b98001
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -43,7 +43,6 @@ connection = connect({
     'user': 'some_user',
     'password': 'some_password',
     'database': 'a_database'
-
     })
 
 cur = connection.cursor()

diff --git a/vertica_python/vertica/connection.py b/vertica_python/vertica/connection.py
@@ -186,6 +186,8 @@ def process_message(self, message):
             self.parameters[message.name] = message.value
         elif isinstance(message, messages.ReadyForQuery):
             self.transaction_status = message.transaction_status
+        elif isinstance(message, messages.CommandComplete):
+            pass
         else:
             raise errors.MessageError("Unhandled message: {0}".format(message))
 

diff --git a/vertica_python/vertica/cursor.py b/vertica_python/vertica/cursor.py
@@ -70,7 +70,6 @@ def execute(self, operation, parameters=None):
             elif isinstance(message, messages.RowDescription):
                 self.description = map(lambda fd: Column(fd), message.fields)
             elif 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
@@ -83,6 +82,8 @@ def fetchone(self):
             row = self.row_formatter(self._message)
             self._message = self.connection.read_message()
             return row
+        else:
+            self.connection.process_message(self._message)
 
     def iterate(self):
         row = self.fetchone()