From fee6b3fafaee3fca18454a70d9c3de2a8ea3a267 Mon Sep 17 00:00:00 2001
From: Niharika Ray <nikki.ray@airbnb.com>
Date: Fri, 25 Mar 2016 11:26:59 -0700
Subject: [PATCH] Adding a DRUID_IS_ACTIVE flag and changing nav bar

---
 dashed/config.py |  1 +
 dashed/views.py  | 39 ++++++++++++++++++++++-----------------
 2 files changed, 23 insertions(+), 17 deletions(-)

diff --git a/dashed/config.py b/dashed/config.py
index 915b9a9e5a..31a7990435 100644
--- a/dashed/config.py
+++ b/dashed/config.py
@@ -53,6 +53,7 @@ APP_NAME = "Dashed"
 # tz.tzutc() : Using utc timezone
 # tz.tzlocal() : Using local timezone
 # other tz can be overridden by providing a local_config
+DRUID_IS_ACTIVE = True
 DRUID_TZ = tz.tzutc()
 
 # ----------------------------------------------------
diff --git a/dashed/views.py b/dashed/views.py
index 42aff94078..b43b43243a 100644
--- a/dashed/views.py
+++ b/dashed/views.py
@@ -207,12 +207,14 @@ class DruidClusterModelView(DashedModelView, DeleteMixin):  # noqa
     edit_columns = add_columns
     list_columns = ['cluster_name', 'metadata_last_refreshed']
 
-appbuilder.add_view(
-    DruidClusterModelView,
-    "Druid Clusters",
-    icon="fa-cubes",
-    category="Sources",
-    category_icon='fa-database',)
+
+if config['DRUID_IS_ACTIVE']:
+    appbuilder.add_view(
+        DruidClusterModelView,
+        "Druid Clusters",
+        icon="fa-cubes",
+        category="Sources",
+        category_icon='fa-database',)
 
 
 class SliceModelView(DashedModelView, DeleteMixin):  # noqa
@@ -326,11 +328,12 @@ class DruidDatasourceModelView(DashedModelView, DeleteMixin):  # noqa
     def post_update(self, datasource):
         self.post_add(datasource)
 
-appbuilder.add_view(
-    DruidDatasourceModelView,
-    "Druid Datasources",
-    category="Sources",
-    icon="fa-cube")
+if config['DRUID_IS_ACTIVE']:
+    appbuilder.add_view(
+        DruidDatasourceModelView,
+        "Druid Datasources",
+        category="Sources",
+        icon="fa-cube")
 
 
 @app.route('/health')
@@ -778,12 +781,14 @@ class Dashed(BaseView):
             utils=utils)
 
 appbuilder.add_view_no_menu(Dashed)
-appbuilder.add_link(
-    "Refresh Druid Metadata",
-    href='/dashed/refresh_datasources/',
-    category='Sources',
-    category_icon='fa-database',
-    icon="fa-cog")
+
+if config['DRUID_IS_ACTIVE']:
+    appbuilder.add_link(
+        "Refresh Druid Metadata",
+        href='/dashed/refresh_datasources/',
+        category='Sources',
+        category_icon='fa-database',
+        icon="fa-cog")
 
 
 class CssTemplateModelView(DashedModelView, DeleteMixin):