from django import forms from django.utils.translation import ugettext_lazy as _ from orchestra.forms.widgets import ShowTextWidget, ReadOnlyWidget class ResourceForm(forms.ModelForm): verbose_name = forms.CharField(label=_("Name"), required=False, widget=ShowTextWidget(bold=True)) allocated = forms.IntegerField(label=_("Allocated")) unit = forms.CharField(label=_("Unit"), widget=ShowTextWidget(), required=False) class Meta: fields = ('verbose_name', 'used', 'last_update', 'allocated', 'unit') def __init__(self, *args, **kwargs): self.resource = kwargs.pop('resource', None) super(ResourceForm, self).__init__(*args, **kwargs) if self.resource: self.fields['verbose_name'].initial = self.resource.get_verbose_name() self.fields['unit'].initial = self.resource.unit if self.resource.on_demand: self.fields['allocated'].required = False self.fields['allocated'].widget = ReadOnlyWidget(None, '') else: self.fields['allocated'].required = True self.fields['allocated'].initial = self.resource.default_allocation # def has_changed(self): # """ Make sure resourcedata objects are created for all resources """ # if not self.instance.pk: # return True # return super(ResourceForm, self).has_changed() def save(self, *args, **kwargs): self.instance.resource_id = self.resource.pk return super(ResourceForm, self).save(*args, **kwargs)