From 28dbf07f017c888577f6761079430acf9514baae Mon Sep 17 00:00:00 2001 From: Beto Dealmeida Date: Fri, 18 Jan 2019 10:11:59 -0800 Subject: [PATCH] Allow empty results in Hive (from SET, eg) (#6695) * Allow empty results in Hive (from SET, eg) * Remove patch * Merge heads * Delete merge heads (cherry picked from commit 00388811b6e5ea56c8e3c92428d6cd805a98c8d1) (cherry picked from commit e5358175153060d190e50d027fd8c6f6d04667e1) --- superset/db_engine_specs.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/superset/db_engine_specs.py b/superset/db_engine_specs.py index 227e0b9b6b7d9..26ed45d1e80af 100644 --- a/superset/db_engine_specs.py +++ b/superset/db_engine_specs.py @@ -1004,11 +1004,15 @@ def fetch_result_sets(cls, db, datasource_type): @classmethod def fetch_data(cls, cursor, limit): + import pyhive from TCLIService import ttypes state = cursor.poll() if state.operationState == ttypes.TOperationState.ERROR_STATE: raise Exception('Query error', state.errorMessage) - return super(HiveEngineSpec, cls).fetch_data(cursor, limit) + try: + return super(HiveEngineSpec, cls).fetch_data(cursor, limit) + except pyhive.exc.ProgrammingError: + return [] @staticmethod def create_table_from_csv(form, table):