Setting adjust_database_uri for HiveEngineSpec

apache · Apr 18, 2017 · 5f5d576 · 5f5d576
1 parent d3f55a0
commit 5f5d576
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/superset/db_engine_specs.py b/superset/db_engine_specs.py
@@ -650,6 +650,12 @@ def fetch_result_sets(cls, db, datasource_type, force=False):
         return BaseEngineSpec.fetch_result_sets(
             db, datasource_type, force=force)
 
+    @classmethod
+    def adjust_database_uri(cls, uri, selected_schema=None):
+        if selected_schema:
+            uri.database = selected_schema
+        return uri
+
     @classmethod
     def progress(cls, logs):
         # 17/02/07 19:36:38 INFO ql.Driver: Total jobs = 5