Fixed streamed data chinks handeling

Author: VaghinakDev

src/superannotate/lib/app/interface/sdk_interface.py

@@ -2850,7 +2850,7 @@ def invite_contributors_to_team(
 
 @Trackable
 @validate_arguments
-def get_annotations(project: NotEmptyStr, items: Optional[List[NotEmptyStr]]):
+def get_annotations(project: NotEmptyStr, items: Optional[List[NotEmptyStr]] = None):
     """Returns annotations for the given list of items.
 
     :param project: project name

src/superannotate/lib/core/reporter.py

@@ -46,9 +46,9 @@ def log_debug(self, value: str):
         self.debug_messages.append(value)
 
     def start_progress(
-        self, iterations: Union[int, range], description: str = "Processing"
+        self, iterations: Union[int, range], description: str = "Processing", disable=False
     ):
-        self.progress_bar = self.get_progress_bar(iterations, description)
+        self.progress_bar = self.get_progress_bar(iterations, description, disable)
 
     @staticmethod
     def get_progress_bar(

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

@@ -9,6 +9,8 @@
 
 import boto3
 import lib.core as constances
+from lib.core.conditions import Condition
+from lib.core.conditions import CONDITION_EQ as EQ
 from lib.core.data_handlers import ChainedAnnotationHandlers
 from lib.core.data_handlers import DocumentTagHandler
 from lib.core.data_handlers import LastActionHandler
@@ -494,14 +496,18 @@ def __init__(
             reporter: Reporter,
             project: ProjectEntity,
             folder: FolderEntity,
+            images: BaseManageableRepository,
             item_names: Optional[List[str]],
-            backend_service_provider: SuerannotateServiceProvider
+            backend_service_provider: SuerannotateServiceProvider,
+            show_process: bool = True
     ):
         super().__init__(reporter)
         self._project = project
         self._folder = folder
+        self._images = images
         self._item_names = item_names
         self._client = backend_service_provider
+        self._show_process = show_process
 
     def validate_item_names(self):
         if self._item_names:
@@ -512,27 +518,36 @@ def validate_item_names(self):
                     f"Dropping duplicates. Found {len_unique_items}/{len_items} unique items."
                 )
                 self._item_names = item_names
+        else:
+            condition = (
+                    Condition("team_id", self._project.team_id, EQ)
+                    & Condition("project_id", self._project.uuid, EQ)
+                    & Condition("folder_id", self._folder.uuid, EQ)
+            )
+
+            self._item_names = [item.name for item in self._images.get_all(condition)]
 
     def execute(self):
-        items_count = len(self._item_names)
-        self.reporter.log_info(
-            f"Getting {items_count} annotations from "
-            f"{self._project.name}{f'/{self._folder.name}' if self._folder else ''}."
-        )
-        self.reporter.start_progress(items_count)
-        annotations = self._client.get_annotations(
-            team_id=self._project.team_id,
-            project_id=self._project.uuid,
-            folder_id=self._folder.uuid,
-            items=self._item_names,
-            reporter=self.reporter
-        )
-        received_items_count = len(annotations)
-        if items_count > received_items_count:
-            self.reporter.log_warning(
-                f"Could not find annotations for {items_count - received_items_count}/{items_count} items."
+        if self.is_valid():
+            items_count = len(self._item_names)
+            self.reporter.log_info(
+                f"Getting {items_count} annotations from "
+                f"{self._project.name}{f'/{self._folder.name}' if self._folder else ''}."
             )
-        self._response.data = annotations
+            self.reporter.start_progress(items_count, disable=not self._show_process)
+            annotations = self._client.get_annotations(
+                team_id=self._project.team_id,
+                project_id=self._project.uuid,
+                folder_id=self._folder.uuid,
+                items=self._item_names,
+                reporter=self.reporter
+            )
+            received_items_count = len(annotations)
+            if items_count > received_items_count:
+                self.reporter.log_warning(
+                    f"Could not find annotations for {items_count - received_items_count}/{items_count} items."
+                )
+            self._response.data = annotations
         return self._response
 
 
@@ -542,13 +557,15 @@ def __init__(
             reporter: Reporter,
             project: ProjectEntity,
             folder: FolderEntity,
+            images: BaseManageableRepository,
             video_name: str,
             fps: int,
             backend_service_provider: SuerannotateServiceProvider
     ):
         super().__init__(reporter)
         self._project = project
         self._folder = folder
+        self._images = images
         self._video_name = video_name
         self._fps = fps
         self._client = backend_service_provider
@@ -558,8 +575,10 @@ def execute(self):
             reporter=self.reporter,
             project=self._project,
             folder=self._folder,
+            images=self._images,
             item_names=[self._video_name],
-            backend_service_provider=self._client
+            backend_service_provider=self._client,
+            show_process=False
         ).execute()
         generator = VideoFrameGenerator(response.data[0], fps=self._fps)
         self.reporter.log_info(f"Getting annotations for {generator.frames_count} frames from {self._video_name}.")

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

@@ -1928,8 +1928,8 @@ def execute(self):
                     project=self._project,
                     folder=self._folder,
                     attachments=self._attachments[
-                        i : i + self.CHUNK_SIZE
-                    ],  # noqa: E203
+                        i : i + self.CHUNK_SIZE  # noqa: E203
+                    ],
                     backend_service_provider=self._backend_service,
                     annotation_status=self._annotation_status,
                     upload_state_code=self._upload_state_code,

src/superannotate/lib/infrastructure/controller.py

@@ -1626,6 +1626,7 @@ def get_annotations(self, project_name: str, folder_name: str, item_names: List[
             reporter=self.default_reporter,
             project=project,
             folder=folder,
+            images=self.images,
             item_names=item_names,
             backend_service_provider=self.backend_client
         )
@@ -1639,6 +1640,7 @@ def get_annotations_per_frame(self, project_name: str, folder_name: str, video_n
             reporter=self.default_reporter,
             project=project,
             folder=folder,
+            images=self.images,
             video_name=video_name,
             fps=fps,
             backend_service_provider=self.backend_client

src/superannotate/lib/infrastructure/services.py

@@ -176,6 +176,7 @@ class SuperannotateBackendService(BaseBackendService):
     DEFAULT_CHUNK_SIZE = 1000
 
     URL_USERS = "users"
+    URL_LIST_ALL_IMAGES = "/images/getImagesWithAnnotationPaths"
     URL_LIST_PROJECTS = "projects"
     URL_FOLDERS_IMAGES = "images-folders"
     URL_CREATE_PROJECT = "project"

src/superannotate/lib/infrastructure/stream_data_handler.py

@@ -25,18 +25,21 @@ async def fetch(self, method: str, session: aiohttp.ClientSession, url: str, dat
                     params: dict = None):
         response = await session._request(method, url, json=data, params=params)
         buffer = b""
-        async for line in response.content:
+        async for line in response.content.iter_any():
             slices = line.split(self.DELIMITER)
             if len(slices) == 1:
                 buffer += slices[0]
                 continue
             elif slices[0]:
                 self._annotations.append(json.loads(buffer + slices[0]))
+                self._reporter.update_progress()
             for data in slices[1:-1]:
                 self._annotations.append(json.loads(data))
+                self._reporter.update_progress()
             buffer = slices[-1]
         if buffer:
             self._annotations.append(json.loads(buffer))
+            self._reporter.update_progress()
         return self._annotations
 
     async def get_data(
@@ -54,9 +57,8 @@ async def get_data(
 
             if chunk_size:
                 for i in range(0, len(data), chunk_size):
-                    await self.fetch(method, session, url, map_function(data[i:i + chunk_size]), params=params)
-                    self._reporter.update_progress(chunk_size)
+                    data_to_process = data[i:i + chunk_size]
+                    await self.fetch(method, session, url, map_function(data_to_process), params=params)
             else:
                 await self.fetch(method, session, url, map_function(data), params=params)
-                self._reporter.update_progress(len(data))
         return self._annotations

tests/integration/annotations/test_get_annotations.py

@@ -1,12 +1,13 @@
-from pathlib import Path
+import json
 import os
+from pathlib import Path
 from typing import List
-import json
-import pytest
 
-import src.superannotate as sa
+import pytest
 from pydantic import parse_obj_as
 from superannotate_schemas.schemas.internal import VectorAnnotation
+
+import src.superannotate as sa
 from tests.integration.base import BaseTestCase
 
 
@@ -40,3 +41,18 @@ def test_get_annotations(self):
             annotation_data = json.load(annotation_file)
             self.assertEqual(len(annotation_data["instances"]), len(annotations[0]["instances"]))
         parse_obj_as(List[VectorAnnotation], annotations)
+
+    @pytest.mark.flaky(reruns=3)
+    def test_get_annotations_all(self):
+        sa.init()
+        sa.upload_images_from_folder_to_project(
+            self.PROJECT_NAME, self.folder_path, annotation_status="InProgress"
+        )
+        sa.create_annotation_classes_from_classes_json(
+            self.PROJECT_NAME, f"{self.folder_path}/classes/classes.json"
+        )
+        _, _, _ = sa.upload_annotations_from_folder_to_project(
+            self.PROJECT_NAME, self.folder_path
+        )
+        annotations = sa.get_annotations(f"{self.PROJECT_NAME}")
+        self.assertEqual(len(annotations), 4)