From 5ec32e84cc256dd0580c40c9912455a13170d418 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Fri, 17 Mar 2023 10:01:15 +0100 Subject: [PATCH] origin before save --- teleforma/forms.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/teleforma/forms.py b/teleforma/forms.py index 060340b0..7fc4486a 100644 --- a/teleforma/forms.py +++ b/teleforma/forms.py @@ -127,6 +127,8 @@ class UserForm(ModelForm): captcha = ReCaptchaField() accept = BooleanField() + origin = None + class Meta: model = User fields = ['first_name', 'last_name', 'email'] @@ -186,8 +188,12 @@ class UserForm(ModelForm): telephone=data['telephone'], birthday=data['birthday'] ) + if self.origin: + self.profile.origin = self.origin + if commit: self.profile.save() + platform_only = data.get('platform_only') == 'True' and True or False fascicule = data.get('fascicule') == 'True' and True or False training = data.get('training') @@ -222,9 +228,8 @@ class UserForm(ModelForm): class UserUseYourLawOriginForm(UserForm): def save(self, commit=True): - super(UserUseYourLawOriginForm, self).save(commit=commit) - origin, c = Origin.objects.get_or_create(name="UseYourLaw") - self.profile.origin = origin + self.origin, c = Origin.objects.get_or_create(name="UseYourLaw") + super(UserUseYourLawOriginForm, self).save(commit=True) class CorrectorForm(ModelForm): -- 2.39.5