diff --git a/accounts_ext/forms.py b/accounts_ext/forms.py index eed5cf5..f928763 100644 --- a/accounts_ext/forms.py +++ b/accounts_ext/forms.py @@ -83,15 +83,13 @@ class RegistrationCompanyForm(forms.ModelForm): def save(self, user, commit=True): company = Company.objects.filter(user=user).first() - if not company: - company = self.instance - company.set_company_type(self.cleaned_data['is_individual']) - try: - company.save() - except Exception as e: - logger.error(e.__str__()) - raise e - return company + if company: + self.instance.id = company.id + self.instance.create_at = company.create_at + self.instance.updated_at = company.updated_at + self.instance.user = user + self.instance.set_company_type(self.cleaned_data['is_individual']) + return super().save(commit) class Meta: model = Company