fix: OPTIC-79: Add queueTotal to lsfProperties from project (#4742)

* [submodules] Build static HumanSignal/dm2

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6099175716

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6099186034

* [submodules] Build static HumanSignal/dm2

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6126697671

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6191135530

* [submodules] Build static HumanSignal/dm2

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6191149554

* [submodules] Build static HumanSignal/dm2

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6199486597

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6201782822

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6215277169

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6237615551

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6250094239

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6250115389

* Add queue_total and queue_done fields to project serializer

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6403602996

* docs: LSF Update

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6403602996

* ci: Build frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6403661920

* [submodules] Build static HumanSignal/dm2

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6410235609

* [submodules] Build static HumanSignal/label-studio-frontend

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6410289300

* [submodules] Build static HumanSignal/dm2

Workflow run: https://github.com/HumanSignal/label-studio/actions/runs/6421795914

* merge develop

---------

Co-authored-by: Travis1282 <travisjosephclark@gmail.com>
Co-authored-by: Andreas Divaris <dredivaris@gmail.com>
Co-authored-by: AndrejOros <AndrejOros@users.noreply.github.com>
Co-authored-by: robot-ci-heartex <robot-ci-heartex@users.noreply.github.com>
Co-authored-by: Brandon Martel <brandonmartel@gmail.com>

Author: 5 people

label_studio/projects/serializers.py

@@ -2,10 +2,12 @@
 """
 import bleach
 from constants import SAFE_HTML_ATTRIBUTES, SAFE_HTML_TAGS
+from django.db.models import Q
 from projects.models import Project, ProjectImport, ProjectOnboarding, ProjectReimport, ProjectSummary
 from rest_flex_fields import FlexFieldsModelSerializer
 from rest_framework import serializers
 from rest_framework.serializers import SerializerMethodField
+from tasks.models import Task
 from users.serializers import UserSimpleSerializer
 
 
@@ -66,6 +68,16 @@ class ProjectSerializer(FlexFieldsModelSerializer):
     )
     finished_task_number = serializers.IntegerField(default=None, read_only=True, help_text='Finished tasks')
 
+    queue_total = serializers.SerializerMethodField()
+    queue_done = serializers.SerializerMethodField()
+
+    @property
+    def user_id(self):
+        try:
+            return self.context['request'].user.id
+        except KeyError:
+            return next(iter(self.context['user_cache']))
+
     @staticmethod
     def get_config_has_control_tags(project):
         return len(project.get_parsed_config()) > 0
@@ -141,6 +153,8 @@ class Meta:
             'reveal_preannotations_interactively',
             'pinned_at',
             'finished_task_number',
+            'queue_total',
+            'queue_done',
         ]
 
     def validate_label_config(self, value):
@@ -152,6 +166,27 @@ def validate_label_config(self, value):
             self.instance.validate_config(value)
         return value
 
+    def get_queue_total(self, project):
+        remain = project.tasks.filter(
+            Q(is_labeled=False) & ~Q(annotations__completed_by_id=self.user_id)
+            | Q(annotations__completed_by_id=self.user_id)
+        ).distinct()
+        return remain.count()
+
+    def get_queue_done(self, project):
+        tasks_filter = {
+            'project': project,
+            'annotations__completed_by_id': self.user_id,
+        }
+
+        if project.skip_queue == project.SkipQueue.REQUEUE_FOR_ME:
+            tasks_filter['annotations__was_cancelled'] = False
+
+        already_done_tasks = Task.objects.filter(**tasks_filter)
+        result = already_done_tasks.distinct().count()
+
+        return result
+
 
 class ProjectOnboardingSerializer(serializers.ModelSerializer):
     class Meta: