From 075fb84bfe11c80e1f30d04734aee213d78561df Mon Sep 17 00:00:00 2001
From: sphenlee <54027992+sphenlee@users.noreply.github.com>
Date: Wed, 4 Sep 2019 15:10:56 +1000
Subject: [PATCH] hive_ds: show a user friendly error message when possible
 (#4121)

---
 redash/query_runner/hive_ds.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/redash/query_runner/hive_ds.py b/redash/query_runner/hive_ds.py
index 361bdf2105..6b10e23ebc 100644
--- a/redash/query_runner/hive_ds.py
+++ b/redash/query_runner/hive_ds.py
@@ -9,6 +9,7 @@
 
 try:
     from pyhive import hive
+    from pyhive.exc import DatabaseError
     from thrift.transport import THttpClient
     enabled = True
 except ImportError:
@@ -129,6 +130,12 @@ def run_query(self, query, user):
                 connection.cancel()
             error = "Query cancelled by user."
             json_data = None
+        except DatabaseError as e:
+            try:
+                error = e.args[0].status.errorMessage
+            except AttributeError:
+                error = str(e)
+            json_data = None
         finally:
             if connection:
                 connection.close()