Add completed imamges count

Author: dshabin

src/superannotate/lib/core/usecases/projects.py

@@ -125,15 +125,18 @@ def execute(self):
         )
         data["project"] = project
         if self._include_complete_image_count:
-            projects = self._projects.get_all(
-                condition=(
-                    Condition("completeImagesCount", "true", EQ)
-                    & Condition("name", self._project.name, EQ)
-                    & Condition("team_id", self._project.team_id, EQ)
-                )
+            completed_images_data = self._service.bulk_get_folders(
+                self._project.team_id, [project.uuid]
             )
-            if projects:
-                data["project"] = projects[0]
+            root_completed_count = 0
+            total_completed_count = 0
+            for i in completed_images_data['data']:
+                total_completed_count += i['completedCount']
+                if i['is_root']:
+                    root_completed_count = i['completedCount']
+
+            project.root_folder_completed_images_count = root_completed_count
+            project.completed_images_count = total_completed_count
 
         if self._include_annotation_classes:
             self.annotation_classes_use_case.execute()

tests/integration/test_folders.py

@@ -444,6 +444,19 @@ def test_folder_export(self):
             )
             self.assertEqual(len(list((temp_dir).glob("*.*"))), 4)
 
+    def test_project_completed_count(self):
+        sa.upload_images_from_folder_to_project(
+            self.PROJECT_NAME, self.folder_path, annotation_status="Completed"
+        )
+        sa.create_folder(self.PROJECT_NAME, self.TEST_FOLDER_NAME_1)
+        project = f"{self.PROJECT_NAME}/{self.TEST_FOLDER_NAME_1}"
+        sa.upload_images_from_folder_to_project(
+            project, self.folder_path, annotation_status="Completed"
+        )
+        project_metadata = sa.get_project_metadata(self.PROJECT_NAME, include_complete_image_count=True)
+        self.assertEqual(project_metadata['completed_images_count'], 8)
+        self.assertEqual(project_metadata['rootFolderCompletedImagesCount'], 4)
+
     def test_folder_image_annotation_status(self):
         sa.upload_images_from_folder_to_project(
             self.PROJECT_NAME, self.folder_path, annotation_status="InProgress"