[sql lab] improve error messages

superset/data/__init__.py

@@ -950,10 +950,10 @@ def load_random_time_series_data():
 
 def load_country_map_data():
     """Loading data for map with country map"""
-    csvPath = os.path.join(DATA_FOLDER, 'birth_france_data_for_country_map.csv')
-    data = pd.read_csv(csvPath, encoding="utf-8")
+    csv_path = os.path.join(DATA_FOLDER, 'birth_france_data_for_country_map.csv')
+    data = pd.read_csv(csv_path, encoding="utf-8")
     data['date'] = datetime.datetime.now().date()
-    data.to_sql(
+    data.to_sql(  # pylint: disable=no-member
         'birth_france_by_region',
         db.engine,
         if_exists='replace',

superset/db_engine_specs.py

@@ -351,6 +351,17 @@ def adjust_database_uri(cls, uri, selected_schema=None):
     def epoch_to_dttm(cls):
         return "from_unixtime({col})"
 
+    @classmethod
+    def extract_error_message(cls, e):
+        """Extract error message for queries"""
+        message = str(e)
+        try:
+            if isinstance(e.args, tuple) and len(e.args) > 1:
+                message = e.args[1]
+        except:
+            pass
+        return message
+
 
 class PrestoEngineSpec(BaseEngineSpec):
     engine = 'presto'

superset/sql_lab.py

@@ -120,7 +120,7 @@ def handle_error(msg):
         session.commit()
         payload.update({
             'status': query.status,
-            'error_essage': msg,
+            'error': msg,
         })
         return payload
 

superset/views/base.py

@@ -29,13 +29,13 @@ def get_error_msg():
     return error_msg
 
 
-def json_error_response(msg, status=None, stacktrace=None):
-    data = {'error': str(msg)}
-    if stacktrace:
-        data['stacktrace'] = stacktrace
-    status = status if status else 500
+def json_error_response(msg=None, status=500, stacktrace=None, payload=None):
+    if not payload:
+        payload = {'error': str(msg)}
+        if stacktrace:
+            payload['stacktrace'] = stacktrace
     return Response(
-        json.dumps(data),
+        json.dumps(payload, default=utils.json_iso_dttm_ser),
         status=status, mimetype="application/json")
 
 

superset/views/core.py

@@ -2072,13 +2072,12 @@ def sql_json(self):
                 # pylint: disable=no-value-for-parameter
                 data = sql_lab.get_sql_results(
                     query_id=query_id, return_results=True)
-                payload = json.dumps(data, default=utils.json_iso_dttm_ser)
         except Exception as e:
             logging.exception(e)
             return json_error_response("{}".format(e))
         if data.get('status') == QueryStatus.FAILED:
-            return json_error_response(payload)
-        return json_success(payload)
+            return json_error_response(payload=data)
+        return json_success(json.dumps(data, default=utils.json_iso_dttm_ser))
 
     @has_access
     @expose("/csv/<client_id>")