From 9604e1dd1466521c0de4e81b53f2b1ae61b88942 Mon Sep 17 00:00:00 2001 From: John Bodley Date: Mon, 12 Mar 2018 12:54:32 -0700 Subject: [PATCH] [druid] Adding cluster filter for refresh --- superset/connectors/druid/models.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/superset/connectors/druid/models.py b/superset/connectors/druid/models.py index caef04b5d8..398f3c7e0c 100644 --- a/superset/connectors/druid/models.py +++ b/superset/connectors/druid/models.py @@ -157,16 +157,15 @@ class DruidCluster(Model, AuditMixinNullable, ImportMixin): def refresh(self, datasource_names, merge_flag, refreshAll): """ - Fetches metadata for the specified datasources andm + Fetches metadata for the specified datasources and merges to the Superset database """ session = db.session ds_list = ( session.query(DruidDatasource) - .filter(or_(DruidDatasource.datasource_name == name - for name in datasource_names)) + .filter(DruidDatasource.cluster_name == self.cluster_name) + .filter(DruidDatasource.datasource_name.in_(datasource_names)) ) - ds_map = {ds.name: ds for ds in ds_list} for ds_name in datasource_names: datasource = ds_map.get(ds_name, None)