django-orchestra/orchestra/contrib/resources/forms.py

37 lines
1.5 KiB
Python
Raw Normal View History

2014-07-08 15:19:15 +00:00
from django import forms
from django.utils.translation import ugettext_lazy as _
from orchestra.forms.widgets import ShowTextWidget, ReadOnlyWidget
class ResourceForm(forms.ModelForm):
2014-11-13 15:34:00 +00:00
verbose_name = forms.CharField(label=_("Name"), required=False,
2015-04-26 13:53:00 +00:00
widget=ShowTextWidget(tag='<b>'))
allocated = forms.DecimalField(label=_("Allocated"))
2014-07-16 15:20:16 +00:00
unit = forms.CharField(label=_("Unit"), widget=ShowTextWidget(), required=False)
2014-07-08 15:19:15 +00:00
class Meta:
2014-07-16 15:20:16 +00:00
fields = ('verbose_name', 'used', 'last_update', 'allocated', 'unit')
2014-07-08 15:19:15 +00:00
def __init__(self, *args, **kwargs):
self.resource = kwargs.pop('resource', None)
super(ResourceForm, self).__init__(*args, **kwargs)
if self.resource:
2014-11-10 15:40:51 +00:00
self.fields['verbose_name'].initial = self.resource.get_verbose_name()
2014-07-16 15:20:16 +00:00
self.fields['unit'].initial = self.resource.unit
2014-09-26 15:05:20 +00:00
if self.resource.on_demand:
2014-07-09 16:17:43 +00:00
self.fields['allocated'].required = False
self.fields['allocated'].widget = ReadOnlyWidget(None, '')
2014-07-08 15:19:15 +00:00
else:
2014-07-09 16:17:43 +00:00
self.fields['allocated'].required = True
self.fields['allocated'].initial = self.resource.default_allocation
2014-10-27 13:29:02 +00:00
# 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()
2014-07-08 15:19:15 +00:00
def save(self, *args, **kwargs):
self.instance.resource_id = self.resource.pk
return super(ResourceForm, self).save(*args, **kwargs)