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):