awx/lib/main/forms.py

import json
from django import forms
from django.utils.translation import ugettext_lazy as _
from jsonfield.fields import JSONFormField
from lib.main.models import *


EMPTY_CHOICE = ('', '---------')

class PlaybookOption(object):

    def __init__(self, project, playbook):
        self.project, self.playbook = project, playbook

    def __unicode__(self):
        return self.playbook

class PlaybookSelect(forms.Select):
    '''Custom select widget for playbooks related to a project.'''

    def render_option(self, selected_choices, option_value, obj):
        opt = super(PlaybookSelect, self).render_option(selected_choices,
                                                        option_value,
                                                        unicode(obj))
        # Add a class with the project ID so JS can filter the options.
        if hasattr(obj, 'project'):
            opt = opt.replace('">', '" class="project-%s">' % obj.project.pk)
        return opt

class HostAdminForm(forms.ModelForm):

    class Meta:
        model = Host
    
    vdata = JSONFormField(label=_('Variable data'), required=False, widget=forms.Textarea(attrs={'class': 'vLargeTextField'}))

    def __init__(self, *args, **kwargs):
        super(HostAdminForm, self).__init__(*args, **kwargs)
        if self.instance.variable_data:
            print repr(self.instance.variable_data.data)
            self.initial['vdata'] = self.instance.variable_data.data

    def save(self, commit=True):
        instance = super(HostAdminForm, self).save(commit=commit)
        save_m2m = getattr(self, 'save_m2m', lambda: None)
        vdata = self.cleaned_data.get('vdata', '')
        print 'vdata', repr(vdata)
        def new_save_m2m():
            save_m2m()
            if not instance.variable_data:
                instance.variable_data = VariableData.objects.create(data=vdata)
                instance.save()
            else:
                variable_data = instance.variable_data
                # FIXME!!!
                #variable_data.data = vdata
                #variable_data.save()
        if commit:
            new_save_m2m()
        else:
            self.save_m2m = new_save_m2m
        return instance

class GroupForm(forms.ModelForm):

    class Meta:
        model = Host

    variable_data = JSONFormField(required=False, widget=forms.Textarea(attrs={'class': 'vLargeTextField'}))

class JobTemplateAdminForm(forms.ModelForm):
    '''Custom admin form for creating/editing JobTemplates.'''

    playbook = forms.ChoiceField(choices=[EMPTY_CHOICE], widget=PlaybookSelect)
    
    class Meta:
        model = JobTemplate

    def __init__(self, *args, **kwargs):
        super(JobTemplateAdminForm, self).__init__(*args, **kwargs)
        playbook_choices = []
        for project in Project.objects.all():
            for playbook in project.available_playbooks:
                playbook_choices.append((playbook,
                                         PlaybookOption(project, playbook)))
        self.fields['playbook'].choices = [EMPTY_CHOICE] + playbook_choices

class JobAdminForm(JobTemplateAdminForm):
    '''Custom admin form for creating Jobs.'''

    start_job = forms.BooleanField(initial=False, required=False)

    class Meta:
        model = Job

    def __init__(self, *args, **kwargs):
        super(JobAdminForm, self).__init__(*args, **kwargs)
        if self.instance.pk and self.instance.status != 'new':
            self.fields.pop('playbook', None)

    def clean_start_job(self):
        return self.cleaned_data.get('start_job', False)

    def save(self, commit=True):
        instance = super(JobAdminForm, self).save(commit)
        save_m2m = getattr(self, 'save_m2m', lambda: None)
        should_start = bool(self.cleaned_data.get('start_job', '') and
                            instance.status == 'new')
        def new_save_m2m():
            save_m2m()
            if should_start:
                instance.start()
        if commit:
            new_save_m2m()
        else:
            self.save_m2m = new_save_m2m
        return instance