Implement give rewards scheduler

api/logics.py

@@ -384,8 +384,10 @@ class Logics:
 
         fee_sats = order.last_satoshis * fee_fraction
 
+        reward_tip = int(config('REWARD_TIP')) if user.profile.is_referred else 0
+
         if cls.is_buyer(order, user):
-            invoice_amount = round(order.last_satoshis - fee_sats)  # Trading fee to buyer is charged here.
+            invoice_amount = round(order.last_satoshis - fee_sats - reward_tip)  # Trading fee to buyer is charged here.
 
         return True, {"invoice_amount": invoice_amount}
 
@@ -401,8 +403,10 @@ class Logics:
 
         fee_sats = order.last_satoshis * fee_fraction 
 
+        reward_tip = int(config('REWARD_TIP')) if user.profile.is_referred else 0
+
         if cls.is_seller(order, user):
-            escrow_amount = round(order.last_satoshis + fee_sats)  # Trading fee to seller is charged here.
+            escrow_amount = round(order.last_satoshis + fee_sats + reward_tip)  # Trading fee to seller is charged here.
 
         return True, {"escrow_amount": escrow_amount}
 
@@ -1092,14 +1096,19 @@ class Logics:
     @classmethod
     def add_rewards(cls, order):
         '''
-        This order is called after a trade is finished. 
+        This function is called when a trade is finished. 
-        If order participants were referred, we add the reward to the referees.
+        If participants of the order were referred, the reward is given to the referees.
         '''
 
         if order.maker.profile.is_referred:
-            order.maker.profile.referred_by.pending_rewards += int(config('REWARD_TIP'))
+            profile = order.maker.profile.referred_by
+            profile.pending_rewards += int(config('REWARD_TIP'))
+            profile.save()
+            
         if order.taker.profile.is_referred:
-            order.taker.profile.referred_by.pending_rewards += int(config('REWARD_TIP'))
+            profile = order.taker.profile.referred_by
+            profile.pending_rewards += int(config('REWARD_TIP'))
+            profile.save()
 
         return
 

api/tasks.py

@@ -35,6 +35,28 @@ def users_cleansing():
     }
     return results
 
+@shared_task(name="give_rewards")
+def users_cleansing():
+    """
+    Referral rewards go from pending to earned.
+    Happens asynchronously so the referral program cannot be easily used to spy.
+    """
+    from api.models import Profile
+
+    # Users who's last login has not been in the last 6 hours
+    queryset = Profile.objects.filter(pending_rewards__gt=0)
+
+    # And do not have an active trade, any past contract or any reward.
+    results = {}
+    for profile in queryset:
+        given_reward = profile.pending_rewards
+        profile.earned_rewards += given_reward
+        profile.pending_rewards = 0
+        profile.save()
+
+        results[profile.user.username] = {'given_reward':given_reward,'earned_rewards':profile.earned_rewards}
+
+    return results
 
 @shared_task(name="follow_send_payment")
 def follow_send_payment(lnpayment):

robosats/celery/__init__.py

@@ -31,9 +31,13 @@ app.conf.beat_scheduler = "django_celery_beat.schedulers:DatabaseScheduler"
 
 # Configure the periodic tasks
 app.conf.beat_schedule = {
-    "users-cleansing": {  # Cleans abandoned users every 6 hours
+    "users-cleansing": {  # Cleans abandoned users at midnight
         "task": "users_cleansing",
-        "schedule": timedelta(hours=6),
+        "schedule": crontab(hour=0, minute=0),
+    },
+    "give-rewards": {  # Referral rewards go from 'pending' to 'earned' at midnight
+        "task": "give_rewards",
+        "schedule": crontab(hour=0, minute=0),
     },
     "cache-market-prices": {  # Cache market prices every minutes for now.
         "task": "cache_external_market_prices",