GITC-500: Adds ability to recalculate inactive rounds

app/grants/admin.py

@@ -489,9 +489,13 @@ def stats_link(self, instance):
     def response_change(self, request, obj):
         if "_recalculate_clr" in request.POST:
             from grants.tasks import recalc_clr
-            for grant in obj.grants:
-                recalc_clr.delay(grant.pk)
-            self.message_user(request, "submitted recaclulation to queue")
+            selected_clr = request.POST.get('_selected_clr', False)
+            if selected_clr:
+                recalc_clr.delay(False, int(selected_clr))
+                self.message_user(request, f"submitted recaclulation of GrantCLR:{ selected_clr } to queue")
+            else:
+                recalc_clr.delay(False)
+                self.message_user(request, "submitted recaclulation to queue")
 
         if "_set_current_grant_clr_calculations_to_false" in request.POST:
             latest_calculations = GrantCLRCalculation.objects.filter(grantclr=obj, latest=True)

app/grants/management/commands/estimate_clr_delay.py

@@ -46,8 +46,7 @@ def handle(self, *args, **options):
         if active_clr_rounds:
             for clr_round in active_clr_rounds:
                 print(f"CALCULATING CLR estimates for ROUND: {clr_round.round_num} {clr_round.sub_round_slug}")
-                for grant in clr_round.grants:
-                    recalc_clr.delay(grant.pk)
+                recalc_clr.delay(False, clr_round)
 
         else:
             print("No active CLRs found")

app/grants/models/grant.py

@@ -189,7 +189,7 @@ def record_clr_prediction_curve(self, grant, clr_prediction_curve):
             grantclr=self,
             grant=grant,
             clr_prediction_curve=clr_prediction_curve,
-            latest=True,
+            latest=True if self.is_active else False,
         )
 
 

app/grants/tasks.py

@@ -13,7 +13,7 @@
 from celery import app
 from celery.utils.log import get_task_logger
 from dashboard.models import Profile
-from grants.models import Grant, GrantCollection, Subscription
+from grants.models import Grant, GrantCLR, GrantCollection, Subscription
 from grants.utils import bsci_script, get_clr_rounds_metadata, save_grant_to_notion
 from marketing.mails import (
     new_contributions, new_grant, new_grant_admin, notion_failure_email, thank_you_for_supporting,
@@ -315,25 +315,38 @@ def recalc_clr_if_x_minutes_old(self, grant_id, minutes, retry: bool = True) ->
 
 
 @app.shared_task(bind=True, max_retries=1)
-def recalc_clr(self, grant_id, retry: bool = True) -> None:
+def recalc_clr(self, grant_id, grant_clr, retry: bool = True) -> None:
 
     if settings.FLUSH_QUEUE:
         return
 
-    obj = Grant.objects.get(pk=grant_id)
     from django.utils import timezone
-
     from grants.clr import predict_clr
-    for clr_round in obj.in_active_clrs.all():
+
+    if grant_clr:
+        clr_round = GrantCLR.objects.get(pk=grant_clr)
         network = 'mainnet'
         predict_clr(
             save_to_db=True,
             from_date=timezone.now(),
             clr_round=clr_round,
             network=network,
-            only_grant_pk=obj.pk,
+            only_grant_pk=grant_id,
             what='slim'
         )
+    elif grant_id:
+        obj = Grant.objects.get(pk=grant_id)
+
+        for clr_round in obj.in_active_clrs.all():
+            network = 'mainnet'
+            predict_clr(
+                save_to_db=True,
+                from_date=timezone.now(),
+                clr_round=clr_round,
+                network=network,
+                only_grant_pk=obj.pk,
+                what='slim'
+            )
 
 
 @app.shared_task(bind=True, max_retries=1)

app/grants/views.py

@@ -1617,7 +1617,7 @@ def grant_details_contributions(request, grant_id):
         all_contributions.append(contribution_json)
 
     response['contributions'] = json.loads(json.dumps(all_contributions, default=str))
-    response['next_page_number'] = page + 1 
+    response['next_page_number'] = page + 1
 
     return JsonResponse(response)
 

app/retail/templates/admin/change_form.html

@@ -70,6 +70,7 @@
     {% elif 'grantcollection/' in request.build_absolute_uri %}
       <input type="submit" value="Generate Cache" name="_generate_cache" value="1">
     {% elif 'grantclr/' in request.build_absolute_uri %}
+      <input type="hidden" value="{{ original.id }}" name="_selected_clr">
       <input type="submit" value="Recalculate CLR" name="_recalculate_clr" value="1">
       <input type="submit" value="Reset CLR calculations for this round" name="_set_current_grant_clr_calculations_to_false" value="1">
       <input type="submit" value="Reset all CLR calculations for all rounds" name="_set_all_grant_clr_calculations_to_false" value="1">