Fix error on accounting when no target_type specified

This commit is contained in:
Antoine Bartuccio 2017-05-06 21:15:44 +02:00
parent 7cb9ea40ac
commit 2ef3c0260a

View File

@ -304,14 +304,15 @@ class OperationForm(forms.ModelForm):
def clean(self): def clean(self):
self.cleaned_data = super(OperationForm, self).clean() self.cleaned_data = super(OperationForm, self).clean()
if self.cleaned_data['target_type'] == "USER": if 'target_type' in self.cleaned_data.keys():
self.cleaned_data['target_id'] = self.cleaned_data['user'].id if self.cleaned_data['target_type'] == "USER":
elif self.cleaned_data['target_type'] == "ACCOUNT": self.cleaned_data['target_id'] = self.cleaned_data['user'].id
self.cleaned_data['target_id'] = self.cleaned_data['club_account'].id elif self.cleaned_data['target_type'] == "ACCOUNT":
elif self.cleaned_data['target_type'] == "CLUB": self.cleaned_data['target_id'] = self.cleaned_data['club_account'].id
self.cleaned_data['target_id'] = self.cleaned_data['club'].id elif self.cleaned_data['target_type'] == "CLUB":
elif self.cleaned_data['target_type'] == "COMPANY": self.cleaned_data['target_id'] = self.cleaned_data['club'].id
self.cleaned_data['target_id'] = self.cleaned_data['company'].id elif self.cleaned_data['target_type'] == "COMPANY":
self.cleaned_data['target_id'] = self.cleaned_data['company'].id
return self.cleaned_data return self.cleaned_data
def save(self): def save(self):