From dd0cd0261a520de7236e4f63526faa8ec5703f74 Mon Sep 17 00:00:00 2001 From: Max Yakovenko Date: Sun, 19 Aug 2018 18:13:52 +0300 Subject: [PATCH] update referral app --- referral/middleware.py | 2 -- referral/models.py | 5 +++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/referral/middleware.py b/referral/middleware.py index 5949c99..a13b63d 100644 --- a/referral/middleware.py +++ b/referral/middleware.py @@ -33,8 +33,6 @@ class ReferralMiddleware: if referral and referral.is_active: referral.referralstats.visits += 1 referral.referralstats.save() - referral.user.partnerstats.total_visits += 1 - referral.user.partnerstats.save() set_cookie(response, 'referral', referral_code) elif request.user.is_authenticated: pop_cookie(response, 'referral') diff --git a/referral/models.py b/referral/models.py index 180c57b..832b01c 100644 --- a/referral/models.py +++ b/referral/models.py @@ -77,6 +77,7 @@ class ReferralStats(StatsFormatterMixin, AbstractStatusModel): verbose_name = _('Реферальная статистика') verbose_name_plural = _('Реферальная статистика') + @receiver(post_save, sender=get_user_model()) def create_user_referral(sender, instance, created, **kwargs): if created and Referral.objects.filter(user=instance).first() is None: @@ -105,6 +106,6 @@ def update_ref_stats(sender, user, request, **kwargs): logger.warning("Missing referral code in database: " + referral_code) -@receiver(buying_bought,sender=Buying) -def charge_referral_points(sender,buying,*args,**kwargs): +@receiver(buying_bought, sender=Buying) +def charge_referral_points(sender, buying, *args, **kwargs): pass