From bdc9cdab0d5e314b5d1a33a2d87fca22138d649d Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Mon, 4 May 2015 01:10:40 +0200 Subject: Imports: manage unmatched item links --- ishtar_common/forms_common.py | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) (limited to 'ishtar_common/forms_common.py') diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py index b6971c4f4..677656af6 100644 --- a/ishtar_common/forms_common.py +++ b/ishtar_common/forms_common.py @@ -98,6 +98,43 @@ class NewImportForm(forms.ModelForm): self.instance.user = user return super(NewImportForm, self).save(commit) +class TargetKeyForm(forms.ModelForm): + class Meta: + model = models.TargetKey + fields = ('target', 'value', 'key') + widgets = { + 'value': forms.TextInput(attrs={'readonly':'readonly'}), + 'key': forms.Select(), + } + + def __init__(self, *args, **kwargs): + super(TargetKeyForm, self).__init__(*args, **kwargs) + instance = getattr(self, 'instance', None) + if instance and instance.pk: + self.fields['target'].widget.attrs['readonly'] = True + self.fields['value'].widget.attrs['readonly'] = True + self.fields['key'].widget.choices = list(instance.target.get_choices()) + + def clean_target(self): + instance = getattr(self, 'instance', None) + if instance and instance.pk: + return instance.target + else: + return self.cleaned_data['target'] + + def clean_value(self): + instance = getattr(self, 'instance', None) + if instance and instance.pk: + return instance.value + else: + return self.cleaned_data['value'] + + def save(self, commit=True): + super(TargetKeyForm, self).save(commit) + if self.cleaned_data.get('value'): + self.instance.is_set = True + self.instance.save() + class OrganizationForm(NewItemForm): form_label = _(u"Organization") associated_models = {'organization_type':models.OrganizationType} -- cgit v1.2.3