Fix incomplete processing for write queries

vertica · May 2, 2015 · a04d4ec · vrbala · May 3, 2015 · zer0n
1 parent 89b60a6
commit a04d4ec
Show file tree

Hide file tree

Showing 3 changed files with 2 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