Add company creation logic for user creation signal

remotes/origin/HEAD
Max Yakovenko 8 years ago
parent e3701c494a
commit fa97894a3b
  1. 13
      accounts_ext/forms.py
  2. 2
      accounts_ext/models.py

@ -81,9 +81,16 @@ class RegistrationCompanyForm(forms.ModelForm):
is_individual = forms.BooleanField(label=_('Я физическое лицо'), required=False) is_individual = forms.BooleanField(label=_('Я физическое лицо'), required=False)
def save(self, user, commit=True): def save(self, user, commit=True):
self.instance.user = user company = Company.objects.filter(user=user).first()
self.instance.set_company_type(self.cleaned_data['is_individual']) if not company:
return super().save(commit) 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
class Meta: class Meta:
model = Company model = Company

@ -161,7 +161,7 @@ class Company(AbstractStatusModel):
def create_user_profile(sender, instance, created, **kwargs): def create_user_profile(sender, instance, created, **kwargs):
if created: if created:
Profile.objects.create(user=instance).save() Profile.objects.create(user=instance).save()
Company.objects.create(user=instance).save()
@receiver(user_activated) @receiver(user_activated)
def save_activation_date(sender, user, request, **kwargs): def save_activation_date(sender, user, request, **kwargs):

Loading…
Cancel
Save