Merge pull request #1305 from chrismeyersfsu/feature-labels

job template labels init implementation

awx/main/access.py

@@ -1509,7 +1509,21 @@ class NotificationAccess(BaseAccess):
         if self.user.is_superuser:
             return qs
         return qs
-    
+
+class LabelAccess(BaseAccess):
+    '''
+    I can see/use a Label if I have permission to
+    '''
+    model = Label
+
+    def get_queryset(self):
+        qs = self.model.objects.filter(active=True).distinct()
+        if self.user.is_superuser:
+            return qs
+        return qs
+
+    def can_delete(self, obj):
+        return False
 
 class ActivityStreamAccess(BaseAccess):
     '''
@@ -1712,3 +1726,4 @@ register_access(CustomInventoryScript, CustomInventoryScriptAccess)
 register_access(TowerSettings, TowerSettingsAccess)
 register_access(Notifier, NotifierAccess)
 register_access(Notification, NotificationAccess)
+register_access(Label, LabelAccess)

awx/main/migrations/0008_v300_create_labels.py

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+from django.conf import settings
+import taggit.managers
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('taggit', '0002_auto_20150616_2121'),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('main', '0007_v300_credential_domain_field'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Label',
+            fields=[
+                ('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
+                ('created', models.DateTimeField(default=None, editable=False)),
+                ('modified', models.DateTimeField(default=None, editable=False)),
+                ('description', models.TextField(default=b'', blank=True)),
+                ('active', models.BooleanField(default=True, editable=False)),
+                ('name', models.CharField(max_length=512)),
+                ('created_by', models.ForeignKey(related_name="{u'class': 'label', u'app_label': 'main'}(class)s_created+", on_delete=django.db.models.deletion.SET_NULL, default=None, editable=False, to=settings.AUTH_USER_MODEL, null=True)),
+                ('modified_by', models.ForeignKey(related_name="{u'class': 'label', u'app_label': 'main'}(class)s_modified+", on_delete=django.db.models.deletion.SET_NULL, default=None, editable=False, to=settings.AUTH_USER_MODEL, null=True)),
+                ('organization', models.ForeignKey(related_name='labels', to='main.Organization', help_text='Organization this label belongs to.')),
+                ('tags', taggit.managers.TaggableManager(to='taggit.Tag', through='taggit.TaggedItem', blank=True, help_text='A comma-separated list of tags.', verbose_name='Tags')),
+            ],
+            options={
+                'ordering': ('organization', 'name'),
+            },
+        ),
+        migrations.AddField(
+            model_name='activitystream',
+            name='label',
+            field=models.ManyToManyField(to='main.Label', blank=True),
+        ),
+        migrations.AddField(
+            model_name='job',
+            name='labels',
+            field=models.ManyToManyField(related_name='job_labels', to='main.Label', blank=True),
+        ),
+        migrations.AddField(
+            model_name='jobtemplate',
+            name='labels',
+            field=models.ManyToManyField(related_name='jobtemplate_labels', to='main.Label', blank=True),
+        ),
+        migrations.AlterUniqueTogether(
+            name='label',
+            unique_together=set([('name', 'organization')]),
+        ),
+    ]

awx/main/models/__init__.py

@@ -19,6 +19,7 @@ from awx.main.models.ha import * # noqa
 from awx.main.models.configuration import * # noqa
 from awx.main.models.notifications import * # noqa
 from awx.main.models.fact import * # noqa
+from awx.main.models.label import * # noqa
 
 # Monkeypatch Django serializer to ignore django-taggit fields (which break
 # the dumpdata command; see https://github.com/alex/django-taggit/issues/155).
@@ -64,3 +65,4 @@ activity_stream_registrar.connect(CustomInventoryScript)
 activity_stream_registrar.connect(TowerSettings)
 activity_stream_registrar.connect(Notifier)
 activity_stream_registrar.connect(Notification)
+activity_stream_registrar.connect(Label)

awx/main/models/activity_stream.py

@@ -55,6 +55,7 @@ class ActivityStream(models.Model):
     custom_inventory_script = models.ManyToManyField("CustomInventoryScript", blank=True)
     notifier = models.ManyToManyField("Notifier", blank=True)
     notification = models.ManyToManyField("Notification", blank=True)
+    label = models.ManyToManyField("Label", blank=True)
 
     def get_absolute_url(self):
         return reverse('api:activity_stream_detail', args=(self.pk,))

awx/main/models/jobs.py

@@ -122,7 +122,11 @@ class JobOptions(BaseModel):
     become_enabled = models.BooleanField(
         default=False,
     )
-
+    labels = models.ManyToManyField(
+        "Label",
+        blank=True,
+        related_name='%(class)s_labels'
+    )
 
     extra_vars_dict = VarsDictProperty('extra_vars', True)
 
@@ -190,7 +194,8 @@ class JobTemplate(UnifiedJobTemplate, JobOptions):
         return ['name', 'description', 'job_type', 'inventory', 'project',
                 'playbook', 'credential', 'cloud_credential', 'forks', 'schedule',
                 'limit', 'verbosity', 'job_tags', 'extra_vars', 'launch_type',
-                'force_handlers', 'skip_tags', 'start_at_task', 'become_enabled']
+                'force_handlers', 'skip_tags', 'start_at_task', 'become_enabled',
+                'labels',]
 
     def create_job(self, **kwargs):
         '''

awx/main/models/label.py

@@ -0,0 +1,33 @@
+# Copyright (c) 2016 Ansible, Inc.
+# All Rights Reserved.
+
+# Django
+from django.db import models
+from django.core.urlresolvers import reverse
+from django.utils.translation import ugettext_lazy as _
+
+# AWX
+from awx.main.models.base import CommonModelNameNotUnique
+
+__all__ = ('Label', )
+
+class Label(CommonModelNameNotUnique):
+    '''
+    Generic Tag. Designed for tagging Job Templates, but expandable to other models.
+    '''
+
+    class Meta:
+        app_label = 'main'
+        unique_together = (("name", "organization"),)
+        ordering = ('organization', 'name')
+
+    organization = models.ForeignKey(
+        'Organization',
+        related_name='labels',
+        help_text=_('Organization this label belongs to.'),
+        on_delete=models.CASCADE,
+    )
+
+    def get_absolute_url(self):
+        return reverse('api:label_detail', args=(self.pk,))
+

awx/main/models/unified_jobs.py

@@ -310,11 +310,11 @@ class UnifiedJobTemplate(PolymorphicModel, CommonModelNameNotUnique, Notificatio
         '''
         Create a new unified job based on this unified job template.
         '''
-        save_unified_job = kwargs.pop('save', True)
         unified_job_class = self._get_unified_job_class()
         parent_field_name = unified_job_class._get_parent_field_name()
         kwargs.pop('%s_id' % parent_field_name, None)
         create_kwargs = {}
+        m2m_fields = {}
         create_kwargs[parent_field_name] = self
         for field_name in self._get_unified_job_field_names():
             # Foreign keys can be specified as field_name or field_name_id.
@@ -332,14 +332,25 @@ class UnifiedJobTemplate(PolymorphicModel, CommonModelNameNotUnique, Notificatio
             elif field_name in kwargs:
                 if field_name == 'extra_vars' and isinstance(kwargs[field_name], dict):
                     create_kwargs[field_name] = json.dumps(kwargs['extra_vars'])
+                # We can't get a hold of django.db.models.fields.related.ManyRelatedManager to compare
+                # so this is the next best thing.
+                elif kwargs[field_name].__class__.__name__ is 'ManyRelatedManager':
+                    m2m_fields[field_name] = kwargs[field_name]
                 else:
                     create_kwargs[field_name] = kwargs[field_name]
             elif hasattr(self, field_name):
-                create_kwargs[field_name] = getattr(self, field_name)
+                field_obj = self._meta.get_field_by_name(field_name)[0]
+                # Many to Many can be specified as field_name
+                if isinstance(field_obj, models.ManyToManyField):
+                    m2m_fields[field_name] = getattr(self, field_name)
+                else:
+                    create_kwargs[field_name] = getattr(self, field_name)
         new_kwargs = self._update_unified_job_kwargs(**create_kwargs)
         unified_job = unified_job_class(**new_kwargs)
-        if save_unified_job:
-            unified_job.save()
+        unified_job.save()
+        for field_name, src_field_value in m2m_fields.iteritems():
+            dest_field = getattr(unified_job, field_name)
+            dest_field.add(*list(src_field_value.all().values_list('id', flat=True)))
         return unified_job
 
 

awx/main/tests/functional/conftest.py

@@ -16,6 +16,7 @@ from django.conf import settings
 # AWX
 from awx.main.models.projects import Project
 from awx.main.models.organization import Organization, Permission
+from awx.main.models.jobs import JobTemplate
 from awx.main.models.base import PERM_INVENTORY_READ
 from awx.main.models.ha import Instance
 from awx.main.models.fact import Fact
@@ -264,3 +265,14 @@ def team(organization):
 def permission_inv_read(organization, inventory, team):
     return Permission.objects.create(inventory=inventory, team=team, permission_type=PERM_INVENTORY_READ)
 
+
+@pytest.fixture
+def job_template_labels(organization):
+    jt = JobTemplate(name='test-job_template')
+    jt.save()
+
+    jt.labels.create(name="label-1", organization=organization)
+    jt.labels.create(name="label-2", organization=organization)
+
+    return jt
+

awx/main/tests/functional/models/test_unified_job.py

@@ -0,0 +1,34 @@
+import pytest
+
+
+class TestCreateUnifiedJob:
+    '''
+    Ensure that copying a job template to a job handles many to many field copy
+    '''
+    @pytest.mark.django_db
+    def test_many_to_many(self, mocker, job_template_labels):
+        jt = job_template_labels
+        _get_unified_job_field_names = mocker.patch('awx.main.models.jobs.JobTemplate._get_unified_job_field_names', return_value=['labels'])
+        j = jt.create_unified_job()
+
+        _get_unified_job_field_names.assert_called_with()
+        assert j.labels.all().count() == 2
+        assert j.labels.all()[0] == jt.labels.all()[0]
+        assert j.labels.all()[1] == jt.labels.all()[1]
+
+    '''
+    Ensure that data is looked for in parameter list before looking at the object
+    '''
+    @pytest.mark.django_db
+    def test_many_to_many_kwargs(self, mocker, job_template_labels):
+        jt = job_template_labels
+        mocked = mocker.MagicMock()
+        mocked.__class__.__name__ = 'ManyRelatedManager'
+        kwargs = {
+            'labels': mocked
+        }
+        _get_unified_job_field_names = mocker.patch('awx.main.models.jobs.JobTemplate._get_unified_job_field_names', return_value=['labels'])
+        jt.create_unified_job(**kwargs)
+
+        _get_unified_job_field_names.assert_called_with()
+        mocked.all.assert_called_with()