[hotfix/sqllab] setting up the connection in the try: block

apache · Apr 18, 2017 · 76042be · 76042be
1 parent d3f55a0
commit 76042be
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/superset/sql_lab.py b/superset/sql_lab.py
@@ -120,11 +120,11 @@ def handle_error(msg):
     session.commit()
     logging.info("Set query to 'running'")
 
-    engine = database.get_sqla_engine(schema=query.schema)
-    conn = engine.raw_connection()
-    cursor = conn.cursor()
-    logging.info("Running query: \n{}".format(executed_sql))
     try:
+        engine = database.get_sqla_engine(schema=query.schema)
+        conn = engine.raw_connection()
+        cursor = conn.cursor()
+        logging.info("Running query: \n{}".format(executed_sql))
         logging.info(query.executed_sql)
         cursor.execute(
             query.executed_sql, **db_engine_spec.cursor_execute_kwargs)