Better error handling for presto

apache · Feb 15, 2017 · b5e330d · b5e330d
1 parent acfe62e
commit b5e330d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/superset/sql_lab.py b/superset/sql_lab.py
@@ -108,6 +108,15 @@ def handle_error(msg):
         result_proxy = engine.execute(query.executed_sql, schema=query.schema)
     except Exception as e:
         logging.exception(e)
+        if hasattr(e, 'orig') \
+           and type(e.orig).__name__ == 'DatabaseError' \
+           and isinstance(e.orig[0], dict):
+            error_dict = e.orig[0]
+            e = '{} at {}: {}'.format(
+                error_dict['errorName'],
+                error_dict['errorLocation'],
+                error_dict['message']
+            )
         handle_error(utils.error_msg_from_exception(e))
 
     cursor = result_proxy.cursor