Add support for hiding settings based on whether features are enabled in the license.

awx/conf/conf.py

@@ -78,6 +78,9 @@ register(
     # the other settings change, the cached value for this setting will be
     # cleared to require it to be recomputed.
     depends_on=['ANSIBLE_COW_SELECTION'],
+    # Optional; licensed feature required to be able to view or modify this
+    # setting.
+    feature_required='rebranding',
 )
 
 register(

awx/conf/license.py

@@ -14,7 +14,8 @@ from rest_framework.exceptions import APIException
 from awx.main.task_engine import TaskEnhancer
 from awx.main.utils import memoize
 
-__all__ = ['LicenseForbids', 'get_license', 'feature_enabled', 'feature_exists']
+__all__ = ['LicenseForbids', 'get_license', 'get_licensed_features',
+           'feature_enabled', 'feature_exists']
 
 
 class LicenseForbids(APIException):
@@ -42,6 +43,15 @@ def get_license(show_key=False):
     return license_data
 
 
+def get_licensed_features():
+    """Return a set of all features enabled by the active license."""
+    features = set()
+    for feature, enabled in _get_validated_license_data().get('features', {}).items():
+        if enabled:
+            features.add(feature)
+    return features
+
+
 def feature_enabled(name):
     """Return True if the requested feature is enabled, False otherwise."""
     return _get_validated_license_data().get('features', {}).get(name, False)

awx/conf/registry.py

@@ -50,7 +50,7 @@ class SettingsRegistry(object):
     def get_dependent_settings(self, setting):
         return self._dependent_settings.get(setting, set())
 
-    def get_registered_categories(self):
+    def get_registered_categories(self, features_enabled=None):
         categories = {
             'all': _('All'),
             'changed': _('Changed'),
@@ -59,6 +59,10 @@ class SettingsRegistry(object):
             category_slug = kwargs.get('category_slug', None)
             if category_slug is None or category_slug in categories:
                 continue
+            if features_enabled is not None:
+                feature_required = kwargs.get('feature_required', None)
+                if feature_required and feature_required not in features_enabled:
+                    continue
             if category_slug == 'user':
                 categories['user'] = _('User')
                 categories['user-defaults'] = _('User-Defaults')
@@ -66,7 +70,7 @@ class SettingsRegistry(object):
                 categories[category_slug] = kwargs.get('category', None) or category_slug
         return categories
 
-    def get_registered_settings(self, category_slug=None, read_only=None):
+    def get_registered_settings(self, category_slug=None, read_only=None, features_enabled=None):
         setting_names = []
         if category_slug == 'user-defaults':
             category_slug = 'user'
@@ -79,6 +83,10 @@ class SettingsRegistry(object):
                 # Note: Doesn't catch fields that set read_only via __init__;
                 # read-only field kwargs should always include read_only=True.
                 continue
+            if features_enabled is not None:
+                feature_required = kwargs.get('feature_required', None)
+                if feature_required and feature_required not in features_enabled:
+                    continue
             setting_names.append(setting)
         return setting_names
 
@@ -95,6 +103,7 @@ class SettingsRegistry(object):
         category = field_kwargs.pop('category', None)
         depends_on = frozenset(field_kwargs.pop('depends_on', None) or [])
         placeholder = field_kwargs.pop('placeholder', empty)
+        feature_required = field_kwargs.pop('feature_required', empty)
         if getattr(field_kwargs.get('child', None), 'source', None) is not None:
             field_kwargs['child'].source = None
         field_instance = field_class(**field_kwargs)
@@ -103,6 +112,8 @@ class SettingsRegistry(object):
         field_instance.depends_on = depends_on
         if placeholder is not empty:
             field_instance.placeholder = placeholder
+        if feature_required is not empty:
+            field_instance.feature_required = feature_required
         original_field_instance = field_instance
         if field_class != original_field_class:
             original_field_instance = original_field_class(**field_kwargs)

awx/conf/views.py

@@ -19,6 +19,7 @@ from rest_framework import status
 # Tower
 from awx.api.generics import *  # noqa
 from awx.main.utils import *  # noqa
+from awx.conf.license import get_licensed_features
 from awx.conf.models import Setting
 from awx.conf.serializers import SettingCategorySerializer, SettingSingletonSerializer
 from awx.conf import settings_registry
@@ -37,7 +38,7 @@ class SettingCategoryList(ListAPIView):
 
     def get_queryset(self):
         setting_categories = []
-        categories = settings_registry.get_registered_categories()
+        categories = settings_registry.get_registered_categories(features_enabled=get_licensed_features())
         if self.request.user.is_superuser or self.request.user.is_system_auditor:
             pass  # categories = categories
         elif 'user' in categories:
@@ -60,7 +61,7 @@ class SettingSingletonDetail(RetrieveUpdateDestroyAPIView):
 
     def get_queryset(self):
         self.category_slug = self.kwargs.get('category_slug', 'all')
-        all_category_slugs = settings_registry.get_registered_categories().keys()
+        all_category_slugs = settings_registry.get_registered_categories(features_enabled=get_licensed_features()).keys()
         if self.request.user.is_superuser or getattr(self.request.user, 'is_system_auditor', False):
             category_slugs = all_category_slugs
         else:
@@ -70,7 +71,7 @@ class SettingSingletonDetail(RetrieveUpdateDestroyAPIView):
         if self.category_slug not in category_slugs:
             raise PermissionDenied()
 
-        registered_settings = settings_registry.get_registered_settings(category_slug=self.category_slug, read_only=False)
+        registered_settings = settings_registry.get_registered_settings(category_slug=self.category_slug, read_only=False, features_enabled=get_licensed_features())
         if self.category_slug == 'user':
             return Setting.objects.filter(key__in=registered_settings, user=self.request.user)
         else:
@@ -78,7 +79,7 @@ class SettingSingletonDetail(RetrieveUpdateDestroyAPIView):
 
     def get_object(self):
         settings_qs = self.get_queryset()
-        registered_settings = settings_registry.get_registered_settings(category_slug=self.category_slug)
+        registered_settings = settings_registry.get_registered_settings(category_slug=self.category_slug, features_enabled=get_licensed_features())
         all_settings = {}
         for setting in settings_qs:
             all_settings[setting.key] = setting.value