commit e9d4749f4470aee294f1351baa9422173a2a62c8
author Maxime Beauchemin <maximebeauchemin@gmail.com> Fri Oct 21 13:35:21 2016 -0700
committer Maxime Beauchemin <maximebeauchemin@gmail.com> Fri Oct 21 13:35:21 2016 -0700
tree f671b062562e7b5421e2a460d4db2b22cd535450
parent 93f8e7d8e9b25a1aaf07e2b0c3772c327e523f0e

[hotfix] sqllab presto

caravel/db_engine_specs.py

diff --git a/caravel/db_engine_specs.py b/caravel/db_engine_specs.py
index dfb8690..319700e 100644
--- a/caravel/db_engine_specs.py
+++ b/caravel/db_engine_specs.py
@@ -55,7 +55,7 @@
         return "'{}'".format(dttm.strftime('%Y-%m-%d %H:%M:%S'))
 
     @classmethod
-    def handle_cursor(cls, cursor, query):
+    def handle_cursor(cls, cursor, query, session):
         """Handle a live cursor between the execute and fetchall calls
 
         The flow works without this method doing anything, but it allows

caravel/sql_lab.py

diff --git a/caravel/sql_lab.py b/caravel/sql_lab.py
index 23f3eb0..914f0c2 100644
--- a/caravel/sql_lab.py
+++ b/caravel/sql_lab.py
@@ -98,7 +98,7 @@
     cursor = result_proxy.cursor
     query.status = QueryStatus.RUNNING
     session.flush()
-    db_engine_spec.handle_cursor(cursor, query)
+    db_engine_spec.handle_cursor(cursor, query, session)
 
     cdf = None
     if result_proxy.cursor: