Merge pull request #6029 from AlanCoding/group_v1

Group serializer special v1 functionality

awx/main/access.py

@@ -346,6 +346,17 @@ class BaseAccess(object):
             elif display_method == 'copy' and isinstance(obj, WorkflowJobTemplate) and obj.organization_id is None:
                 user_capabilities[display_method] = self.user.is_superuser
                 continue
+            elif display_method in ['start', 'schedule'] and isinstance(obj, Group):  # TODO: remove in 3.3
+                try:
+                    if obj.deprecated_inventory_source and not obj.deprecated_inventory_source._can_update():
+                        user_capabilities[display_method] = False
+                        continue
+                except Group.deprecated_inventory_source.RelatedObjectDoesNotExist:
+                    user_capabilities[display_method] = False
+                    continue
+                if obj.inventory_source and not obj.inventory_source._can_update():
+                    user_capabilities[display_method] = False
+                    continue
             elif display_method in ['start', 'schedule'] and isinstance(obj, (Project)):
                 if obj.scm_type == '':
                     user_capabilities[display_method] = False
@@ -720,6 +731,19 @@ class GroupAccess(BaseAccess):
                                  "active_jobs": active_jobs})
         return True
 
+    def can_start(self, obj, validate_license=True):
+        # TODO: Delete for 3.3, only used by v1 serializer
+        # Used as another alias to inventory_source start access for user_capabilities
+        if obj:
+            try:
+                return self.user.can_access(
+                    InventorySource, 'start', obj.deprecated_inventory_source,
+                    validate_license=validate_license)
+                obj.deprecated_inventory_source
+            except Group.deprecated_inventory_source.RelatedObjectDoesNotExist:
+                return False
+        return False
+
 
 class InventorySourceAccess(BaseAccess):
     '''

awx/main/migrations/0037_v320_release.py

@@ -28,7 +28,7 @@ class Migration(migrations.Migration):
         migrations.AlterField(
             model_name='inventorysource',
             name='deprecated_group',
-            field=models.ForeignKey(related_name='deprecated_inventory_source', default=None, null=True, to='main.Group'),
+            field=models.OneToOneField(related_name='deprecated_inventory_source', null=True, default=None, to='main.Group'),
         ),
         migrations.AlterField(
             model_name='inventorysource',

awx/main/migrations/0038_v320_data_migrations.py

@@ -19,7 +19,6 @@ class Migration(migrations.Migration):
     operations = [
         # Inventory Refresh
         migrations.RunPython(migration_utils.set_current_apps_for_migrations),
-        migrations.RunPython(invsrc.remove_manual_inventory_sources),
         migrations.RunPython(invsrc.remove_inventory_source_with_no_inventory_link),
         migrations.RunPython(invsrc.rename_inventory_sources),
     ]

awx/main/migrations/_inventory_source.py

@@ -10,6 +10,7 @@ def remove_manual_inventory_sources(apps, schema_editor):
     Group creation and we would use the parent Group as our interface for the user.
     During that process we would create InventorySource that had a source of "manual".
     '''
+    # TODO: use this in the 3.3 data migrations
     InventorySource = apps.get_model('main', 'InventorySource')
     # see models/inventory.py SOURCE_CHOICES - ('', _('Manual'))
     logger.debug("Removing all Manual InventorySource from database.")

awx/main/models/inventory.py

@@ -1063,7 +1063,7 @@ class InventorySource(UnifiedJobTemplate, InventorySourceOptions):
         on_delete=models.CASCADE,
     )
 
-    deprecated_group = models.ForeignKey(
+    deprecated_group = models.OneToOneField(
         'Group',
         related_name='deprecated_inventory_source',
         null=True,
@@ -1178,6 +1178,16 @@ class InventorySource(UnifiedJobTemplate, InventorySourceOptions):
                     success=list(success_notification_templates),
                     any=list(any_notification_templates))
 
+    def clean_source(self):  # TODO: remove in 3.3
+        source = self.source
+        if source and self.deprecated_group:
+            qs = self.deprecated_group.inventory_sources.filter(source__in=CLOUD_INVENTORY_SOURCES)
+            existing_sources = qs.exclude(pk=self.pk)
+            if existing_sources.count():
+                s = u', '.join([x.deprecated_group.name for x in existing_sources])
+                raise ValidationError(_('Unable to configure this item for cloud sync. It is already managed by %s.') % s)
+        return source
+
 
 class InventoryUpdate(UnifiedJob, InventorySourceOptions, JobNotificationMixin):
     '''
@@ -1212,6 +1222,8 @@ class InventoryUpdate(UnifiedJob, InventorySourceOptions, JobNotificationMixin):
 
     def websocket_emit_data(self):
         websocket_data = super(InventoryUpdate, self).websocket_emit_data()
+        if self.inventory_source.deprecated_group is not None:  # TODO: remove in 3.3
+            websocket_data.update(dict(group_id=self.inventory_source.deprecated_group.id))
         return websocket_data
 
     def save(self, *args, **kwargs):

awx/main/signals.py

@@ -372,6 +372,10 @@ model_serializer_mapping = {
 
 def activity_stream_create(sender, instance, created, **kwargs):
     if created and activity_stream_enabled:
+        # TODO: remove deprecated_group conditional in 3.3
+        # Skip recording any inventory source directly associated with a group.
+        if isinstance(instance, InventorySource) and instance.deprecated_group:
+            return
         object1 = camelcase_to_underscore(instance.__class__.__name__)
         changes = model_to_dict(instance, model_serializer_mapping)
         # Special case where Job survey password variables need to be hidden
@@ -417,6 +421,10 @@ def activity_stream_update(sender, instance, **kwargs):
 def activity_stream_delete(sender, instance, **kwargs):
     if not activity_stream_enabled:
         return
+    # TODO: remove deprecated_group conditional in 3.3
+    # Skip recording any inventory source directly associated with a group.
+    if isinstance(instance, InventorySource) and instance.deprecated_group:
+        return
     changes = model_to_dict(instance)
     object1 = camelcase_to_underscore(instance.__class__.__name__)
     activity_entry = ActivityStream(