refactor: OPTIC-1621: Revert "Remove Stale Feature Flag - fflag_perf_…

…back_lsdv_4695_update_prediction_query_to_use_direct_project_relation" (#7012)

label_studio/core/feature_flags/stale_feature_flags.py

@@ -3,6 +3,7 @@
     'fflag_feat_front_optic_66_lazy_chart_evaluation_19092023_short': False,
     'fflag_fix_back_leap_24_tasks_api_optimization_05092023_short': False,
     'fflag_feat_optic_2_ensure_draft_saved_short': True,
+    'fflag_perf_back_lsdv_4695_update_prediction_query_to_use_direct_project_relation': True,
     'fflag_feat_front_lsdv_4620_outliner_optimization_310723_short': True,
     'fflag_fix_front_lsdv_4600_lead_time_27072023_short': False,
     'fflag_feat_front_lops_e_10_updated_ux_short': False,

label_studio/projects/functions/__init__.py

@@ -15,7 +15,13 @@ def annotate_finished_task_number(queryset):
 
 
 def annotate_total_predictions_number(queryset):
-    predictions = Prediction.objects.filter(project=OuterRef('id')).values('id')
+    if flag_set(
+        'fflag_perf_back_lsdv_4695_update_prediction_query_to_use_direct_project_relation',
+        user='auto',
+    ):
+        predictions = Prediction.objects.filter(project=OuterRef('id')).values('id')
+    else:
+        predictions = Prediction.objects.filter(task__project=OuterRef('id')).values('id')
     return queryset.annotate(total_predictions_number=SQCount(predictions))
 
 

label_studio/projects/models.py

@@ -5,6 +5,7 @@
 from typing import Any, Mapping, Optional
 
 from annoying.fields import AutoOneToOneField
+from core.feature_flags import flag_set
 from core.label_config import (
     check_control_in_config_by_regex,
     check_toname_in_config_by_regex,
@@ -326,7 +327,13 @@ def has_predictions(self):
 
     @property
     def has_any_predictions(self):
-        return Prediction.objects.filter(Q(project=self.id)).exists()
+        if flag_set(
+            'fflag_perf_back_lsdv_4695_update_prediction_query_to_use_direct_project_relation',
+            user='auto',
+        ):
+            return Prediction.objects.filter(Q(project=self.id)).exists()
+        else:
+            return Prediction.objects.filter(Q(task__project=self.id)).exists()
 
     @property
     def business(self):
@@ -926,7 +933,14 @@ def get_model_versions(self, with_counters=False, extended=False):
         :param extended: Boolean, if True, returns additional information. Default is False.
         :return: Dict or list containing model versions and their count predictions.
         """
-        predictions = Prediction.objects.filter(project=self)
+        if flag_set(
+            'fflag_perf_back_lsdv_4695_update_prediction_query_to_use_direct_project_relation',
+            user='auto',
+        ):
+            predictions = Prediction.objects.filter(project=self)
+        else:
+            predictions = Prediction.objects.filter(task__project=self)
+        # model_versions = set(predictions.values_list('model_version', flat=True).distinct())
 
         if extended:
             model_versions = list(

label_studio/tasks/api.py

@@ -778,7 +778,13 @@ class PredictionAPI(viewsets.ModelViewSet):
     filterset_fields = ['task', 'task__project', 'project']
 
     def get_queryset(self):
-        return Prediction.objects.filter(project__organization=self.request.user.active_organization)
+        if flag_set(
+            'fflag_perf_back_lsdv_4695_update_prediction_query_to_use_direct_project_relation',
+            user='auto',
+        ):
+            return Prediction.objects.filter(project__organization=self.request.user.active_organization)
+        else:
+            return Prediction.objects.filter(task__project__organization=self.request.user.active_organization)
 
 
 @method_decorator(name='get', decorator=swagger_auto_schema(auto_schema=None))

label_studio/tasks/models.py

@@ -933,8 +933,15 @@ def created_ago(self):
         return timesince(self.created_at)
 
     def has_permission(self, user):
-        user.project = self.project  # link for activity log
-        return self.project.has_permission(user)
+        if flag_set(
+            'fflag_perf_back_lsdv_4695_update_prediction_query_to_use_direct_project_relation',
+            user='auto',
+        ):
+            user.project = self.project  # link for activity log
+            return self.project.has_permission(user)
+        else:
+            user.project = self.task.project  # link for activity log
+            return self.task.project.has_permission(user)
 
     @classmethod
     def prepare_prediction_result(cls, result, project):
@@ -1002,8 +1009,13 @@ def save(self, *args, update_fields=None, **kwargs):
                 update_fields = {'project_id'}.union(update_fields)
 
         # "result" data can come in different forms - normalize them to JSON
-        self.result = self.prepare_prediction_result(self.result, self.project)
-
+        if flag_set(
+            'fflag_perf_back_lsdv_4695_update_prediction_query_to_use_direct_project_relation',
+            user='auto',
+        ):
+            self.result = self.prepare_prediction_result(self.result, self.project)
+        else:
+            self.result = self.prepare_prediction_result(self.result, self.task.project)
         if update_fields is not None:
             update_fields = {'result'}.union(update_fields)
         # set updated_at field of task to now()