Move to db_engine_spec

superset/db_engine_specs.py

@@ -18,6 +18,8 @@
 
 from collections import namedtuple, defaultdict
 from flask_babel import lazy_gettext as _
+from superset import utils
+
 import inspect
 import textwrap
 import time
@@ -91,6 +93,11 @@ def handle_cursor(cls, cursor, query, session):
         query object"""
         pass
 
+    @classmethod
+    def extract_error_message(cls, e):
+        """Extract error message for queries"""
+        return utils.error_msg_from_exception(e)
+
     @classmethod
     def sql_preprocessor(cls, sql):
         """If the SQL needs to be altered prior to running it
@@ -312,6 +319,19 @@ def handle_cursor(cls, cursor, query, session):
             time.sleep(1)
             polled = cursor.poll()
 
+    @classmethod
+    def extract_error_message(cls, 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']
+            )
+        return utils.error_msg_from_exception(e)
+
 
 class MssqlEngineSpec(BaseEngineSpec):
     engine = 'mssql'

superset/sql_lab.py

@@ -108,16 +108,7 @@ 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))
+        handle_error(db_engine_spec.extract_error_message(e))
 
     cursor = result_proxy.cursor
     query.status = QueryStatus.RUNNING