Revert "Update cqlsh driver for new driver execution API"

This reverts commit acf67d559c.

bin/cqlsh.py

@@ -111,13 +111,13 @@ except ImportError, e:
              'Error: %s\n' % (sys.executable, sys.path, e))
 
 from cassandra.auth import PlainTextAuthProvider
-from cassandra.cluster import Cluster
+from cassandra.cluster import Cluster, PagedResult
 from cassandra.metadata import (ColumnMetadata, KeyspaceMetadata,
                                 TableMetadata, protect_name, protect_names,
                                 protect_value)
 from cassandra.policies import WhiteListRoundRobinPolicy
 from cassandra.protocol import QueryMessage, ResultMessage
-from cassandra.query import SimpleStatement, ordered_dict_factory, TraceUnavailable
+from cassandra.query import SimpleStatement, ordered_dict_factory
 
 # cqlsh should run correctly when run out of a Cassandra source tree,
 # out of an unpacked Cassandra tarball, and after a proper package install.
@@ -675,7 +675,6 @@ class Shell(cmd.Cmd):
 
         self.session.default_timeout = client_timeout
         self.session.row_factory = ordered_dict_factory
-        self.session.default_consistency_level = cassandra.ConsistencyLevel.ONE
         self.get_connection_versions()
 
         self.current_keyspace = keyspace
@@ -1122,8 +1121,8 @@ class Shell(cmd.Cmd):
 
     def do_use(self, parsed):
         ksname = parsed.get_binding('ksname')
-        success, _ = self.perform_simple_statement(SimpleStatement(parsed.extract_orig()))
-        if success:
+        result, future = self.perform_simple_statement(SimpleStatement(parsed.extract_orig()))
+        if result:
             if ksname[0] == '"' and ksname[-1] == '"':
                 self.current_keyspace = self.cql_unprotect_name(ksname)
             else:
@@ -1140,7 +1139,7 @@ class Shell(cmd.Cmd):
 
     def perform_statement(self, statement):
         stmt = SimpleStatement(statement, consistency_level=self.consistency_level, serial_consistency_level=self.serial_consistency_level, fetch_size=self.page_size if self.use_paging else None)
-        success, future = self.perform_simple_statement(stmt)
+        result, future = self.perform_simple_statement(stmt)
 
         if future:
             if future.warnings:
@@ -1148,17 +1147,19 @@ class Shell(cmd.Cmd):
 
             if self.tracing_enabled:
                 try:
-                    for trace in future.get_all_query_traces(self.max_trace_wait):
+                    trace = future.get_query_trace(self.max_trace_wait)
+                    if trace:
                         print_trace(self, trace)
-                except TraceUnavailable:
-                    msg = "Statement trace did not complete within %d seconds; trace data may be incomplete." % (self.session.max_trace_wait,)
-                    self.writeresult(msg, color=RED)
-                    for trace_id in future.get_query_trace_ids():
-                        self.show_session(trace_id)
+                    elif stmt.trace_id:
+                        self.writeresult("This statement trace may be incomplete", color=RED)
+                        self.show_session(stmt.trace_id)
+                    else:
+                        msg = "Statement trace did not complete within %d seconds" % (self.session.max_trace_wait)
+                        self.writeresult(msg, color=RED)
                 except Exception, err:
                     self.printerr("Unable to fetch query trace: %s" % (str(err),))
 
-        return success
+        return result
 
     def parse_for_table_meta(self, query_string):
         try:
@@ -1176,7 +1177,7 @@ class Shell(cmd.Cmd):
         while True:
             try:
                 future = self.session.execute_async(statement, trace=self.tracing_enabled)
-                result = future.result()
+                rows = future.result(self.session.default_timeout)
                 break
             except cassandra.OperationTimedOut, err:
                 self.refresh_schema_metadata_best_effort()
@@ -1191,36 +1192,42 @@ class Shell(cmd.Cmd):
                 return False, None
 
         if statement.query_string[:6].lower() == 'select':
-            self.print_result(result, self.parse_for_table_meta(statement.query_string))
+            self.print_result(rows, self.parse_for_table_meta(statement.query_string))
         elif statement.query_string.lower().startswith("list users") or statement.query_string.lower().startswith("list roles"):
-            self.print_result(result, self.get_table_meta('system_auth', 'roles'))
+            self.print_result(rows, self.get_table_meta('system_auth', 'roles'))
         elif statement.query_string.lower().startswith("list"):
-            self.print_result(result, self.get_table_meta('system_auth', 'role_permissions'))
-        elif result:
+            self.print_result(rows, self.get_table_meta('system_auth', 'role_permissions'))
+        elif rows:
             # CAS INSERT/UPDATE
             self.writeresult("")
-            self.print_static_result(list(result), self.parse_for_table_meta(statement.query_string))
+            self.print_static_result(rows, self.parse_for_table_meta(statement.query_string))
         self.flush_output()
         return True, future
 
-    def print_result(self, result, table_meta):
+    def print_result(self, rows, table_meta):
         self.decoding_errors = []
 
         self.writeresult("")
-        if result.has_more_pages and self.tty:
+        if isinstance(rows, PagedResult) and self.tty:
             num_rows = 0
             while True:
-                page = result.current_rows
-                if page:
-                    num_rows += len(page)
-                    self.print_static_result(page, table_meta)
-                if result.has_more_pages:
-                    raw_input("---MORE---")
-                    result.fetch_next_page()
-                else:
+                page = list(rows.current_response)
+                if not page:
                     break
+                num_rows += len(page)
+                self.print_static_result(page, table_meta)
+                if not rows.response_future.has_more_pages:
+                    break
+                raw_input("---MORE---")
+
+                rows.response_future.start_fetching_next_page()
+                result = rows.response_future.result()
+                if rows.response_future.has_more_pages:
+                    rows.current_response = result.current_response
+                else:
+                    rows.current_response = iter(result)
         else:
-            rows = list(result)
+            rows = list(rows or [])
             num_rows = len(rows)
             self.print_static_result(rows, table_meta)
         self.writeresult("(%d rows)" % num_rows)
@@ -2328,7 +2335,7 @@ class ImportProcess(multiprocessing.Process):
         cqltypes = [table_meta.columns[name].typestring for name in self.columns]
         pk_indexes = [self.columns.index(col.name) for col in table_meta.primary_key]
         query = 'INSERT INTO %s.%s (%s) VALUES (%%s)' % (
-            protect_name(table_meta.keyspace_name),
+            protect_name(table_meta.keyspace.name),
             protect_name(table_meta.name),
             ', '.join(protect_names(self.columns)))