From 54f2bb5a6e20761555b20b2bde4bf3e53a9ecd7d Mon Sep 17 00:00:00 2001 From: Max Yakovenko Date: Tue, 31 Jul 2018 11:24:57 +0300 Subject: [PATCH] reffactor logic on user signal create and activate --- accounts_ext/models.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/accounts_ext/models.py b/accounts_ext/models.py index 75a8d70..84c7289 100644 --- a/accounts_ext/models.py +++ b/accounts_ext/models.py @@ -53,10 +53,15 @@ class User(AbstractStatusModel, PermissionsMixin, AbstractBaseUser): }, ) + referral_code = models.ForeignKey('referral.Referral', + verbose_name=_('Реферальная ссылка'), + on_delete=models.SET_NULL, + blank=True, null=True + ) referral_user = models.ForeignKey('self', verbose_name=_('Реферальный пользователь'), on_delete=models.SET_NULL, - blank=True, null=True + blank=True,null=True ) confirmed_at = models.DateTimeField(_('подвтержден в'), blank=True, null=True) @@ -176,7 +181,6 @@ class UserPoints(models.Model): def create_user_profile(sender, instance, created, **kwargs): if created: Profile.objects.create(user=instance).save() - Company.objects.create(user=instance).save() @receiver(user_activated) def save_activation_date(sender, user, request, **kwargs):