From 8bceda8134c6e53f62b5b30910ec0eeb02f051e7 Mon Sep 17 00:00:00 2001
From: Maxime Beauchemin <maximebeauchemin@gmail.com>
Date: Thu, 13 Apr 2017 18:37:27 +0000
Subject: [PATCH] [hotfix] fix iframe viz

---
 superset/db_engine_specs.py | 12 ++++++++++++
 superset/viz.py             |  3 +++
 2 files changed, 15 insertions(+)

diff --git a/superset/db_engine_specs.py b/superset/db_engine_specs.py
index aa042683f0..5fb3836ebe 100644
--- a/superset/db_engine_specs.py
+++ b/superset/db_engine_specs.py
@@ -401,6 +401,18 @@ class PrestoEngineSpec(BaseEngineSpec):
                 row['table_schema'], row['table_name']))
         return result_sets
 
+    @classmethod
+    def extract_error_message(cls, e):
+        """Extract error message for queries"""
+        logging.error(e)
+        print("-="*100)
+        print("-="*100)
+        print("-="*100)
+        print("-="*100)
+        print(type(e))
+        print(dir(e))
+        return error_msg_from_exception(e)
+
     @classmethod
     def extra_table_metadata(cls, database, table_name, schema_name):
         indexes = database.get_indexes(table_name, schema_name)
diff --git a/superset/viz.py b/superset/viz.py
index b4f95f2197..e7e0bd14da 100755
--- a/superset/viz.py
+++ b/superset/viz.py
@@ -1320,6 +1320,9 @@ class IFrameViz(BaseViz):
     credits = 'a <a href="https://github.com/airbnb/superset">Superset</a> original'
     is_timeseries = False
 
+    def get_df(self):
+       return None
+
 
 class ParallelCoordinatesViz(BaseViz):