Changed prepare_export

Author: VaghinakDev

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

@@ -1630,7 +1630,6 @@ def set_image_annotation_status(
     :rtype: dict
     """
     project_name, folder_name = extract_project_folder(project)
-    project_entity = Controller.get_default()._get_project(project_name)
     response = Controller.get_default().set_images_annotation_statuses(
         project_name, folder_name, [image_name], annotation_status
     )
@@ -1639,7 +1638,7 @@ def set_image_annotation_status(
     image = (
         Controller.get_default().get_item(project_name, folder_name, image_name).data
     )
-    return ImageSerializer(image).serialize_by_project(project=project_entity)
+    return BaseSerializer(image).serialize()
 
 
 @Trackable
@@ -2902,7 +2901,7 @@ def query(project: NotEmptyStr, query: Optional[NotEmptyStr]):
     :param project: project name or folder path (e.g., “project1/folder1”)
     :type project: str
 
-    :param query: SAQuL query string.
+    :param query: SAQuL(https://doc.superannotate.com/docs/query-search-1) query string.
     :type query: str
 
     :return: queried items’ metadata list

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

@@ -39,6 +39,7 @@ def __init__(
         project: ProjectEntity,
         folder_names: List[str],
         backend_service_provider: SuperannotateServiceProvider,
+        folders: BaseManageableRepository,
         include_fuse: bool,
         only_pinned: bool,
         annotation_statuses: List[str] = None,
@@ -50,6 +51,7 @@ def __init__(
         self._annotation_statuses = annotation_statuses
         self._include_fuse = include_fuse
         self._only_pinned = only_pinned
+        self._folders = folders
 
     def validate_only_pinned(self):
         if (
@@ -69,6 +71,19 @@ def validate_fuse(self):
                 f"Include fuse functionality is not supported for  projects containing {self._project.type} attached with URLs"
             )
 
+    def validate_folder_names(self):
+        if self._folder_names:
+            condition = (
+                    Condition("team_id", self._project.team_id, EQ) &
+                    Condition("project_id", self._project.id, EQ)
+            )
+            existing_folders = {folder.name for folder in self._folders.get_all(condition)}
+            folder_names_set = set(self._folder_names)
+            if not folder_names_set.issubset(existing_folders):
+                raise AppException(
+                    f"Folder(s) {', '.join(folder_names_set - existing_folders)} does not exist"
+                )
+
     def execute(self):
         if self.is_valid():
             if self._project.upload_state == constances.UploadState.EXTERNAL.value:

src/superannotate/lib/infrastructure/controller.py

@@ -578,6 +578,7 @@ def prepare_export(
         use_case = usecases.PrepareExportUseCase(
             project=project,
             folder_names=folder_names,
+            folders=self.folders,
             backend_service_provider=self._backend_client,
             include_fuse=include_fuse,
             only_pinned=only_pinned,

src/superannotate/version.py

@@ -1 +1 @@
-__version__ = "4.3.3dev9"
+__version__ = "4.3.3dev10"

tests/integration/items/test_set_annotation_statuses.py

@@ -70,3 +70,12 @@ def test_image_annotation_status_via_invalid_names(self):
             sa.set_annotation_statuses(
                 self.PROJECT_NAME, "QualityCheck", ["self.EXAMPLE_IMAGE_1", "self.EXAMPLE_IMAGE_2"]
             )
+
+    def test_set_image_annotation_status(self):
+        sa.attach_items(
+            self.PROJECT_NAME, [self.ATTACHMENT_LIST[0]]
+        )
+        data = sa.set_image_annotation_status(
+            self.PROJECT_NAME, self.ATTACHMENT_LIST[0]["name"], annotation_status="Completed"
+        )
+        assert data["annotation_status"] == "Completed"

tests/integration/test_get_exports.py

@@ -4,6 +4,7 @@
 from os.path import dirname
 
 import src.superannotate as sa
+from src.superannotate import AppException
 from tests.integration.base import BaseTestCase