tod

Author: Vaghinak Basentsyan

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

@@ -3517,7 +3517,10 @@ def upload_images_to_project(
         with tqdm(total=len(images_to_upload), desc="Uploading images") as progress_bar:
             for _ in use_case.execute():
                 progress_bar.update(1)
-        return use_case.data
+        uploaded, failed_images, duplications = use_case.data
+        if duplications:
+            logger.info(f"Duplicated images {', '.join(duplications)}")
+        return uploaded, failed_images, duplications
     raise AppException(use_case.response.errors)
 
 

src/superannotate/lib/core/types.py

@@ -17,6 +17,7 @@ class Attribute(BaseModel):
 
 class AttributeGroup(BaseModel):
     name: StrictStr
+    is_multiselect: Optional[int]
     attributes: List[Attribute]
 
 

src/superannotate/lib/core/usecases.py

@@ -2353,7 +2353,6 @@ def fill_classes_data(self, annotations: dict):
                         attribute["groupId"]
                     ]["attributes"]
                 ):
-                    del attribute["groupId"]
                     continue
                 attribute["name"] = annotation_classes[annotation_class_id][
                     "attribute_groups"
@@ -3268,9 +3267,6 @@ def execute(self):
                 annotations,
                 fuse_image,
             )
-            logger.info(
-                "Downloaded image %s to %s.", self._image.name, str(download_path)
-            )
 
         return self._response
 
@@ -4067,6 +4063,8 @@ def download_to_local_storage(self, destination: str):
                 project_id=self._project.uuid,
                 export_id=export["id"],
             )
+            if "error" in export:
+                raise AppException(export["error"])
             export_status = export["status"]
             if export_status in (ExportStatus.ERROR.value, ExportStatus.CANCELED.value):
                 raise AppException("Couldn't download export.")
@@ -4877,7 +4875,7 @@ def _upload_image(self, image_path: str):
         else:
             try:
                 image_bytes = io.BytesIO(open(image_path, "rb").read())
-            except FileNotFoundError:
+            except OSError:
                 return ProcessedImage(
                     uploaded=False,
                     path=image_path,
@@ -4927,10 +4925,6 @@ def images_to_upload(self):
                     [extension in path for extension in self.exclude_file_patterns]
                 )
             ]
-            excluded_paths = [path for path in paths if path not in filtered_paths]
-            if excluded_paths:
-                logger.info(f"Excluded paths {', '.join(excluded_paths)}")
-
             image_entities = (
                 GetBulkImages(
                     service=self._backend_client,
@@ -4994,8 +4988,6 @@ def execute(self):
                 duplications.extend(attach_duplications)
             uploaded = [image["name"] for image in uploaded]
             failed_images = [image.split("/")[-1] for image in failed_images]
-            if duplications:
-                logger.info(f"Duplicated images {', '.join(duplications)}")
             self._response.data = uploaded, failed_images, duplications
         return self._response
 

src/superannotate/lib/infrastructure/services.py

@@ -39,6 +39,7 @@ def __init__(
         self._paginate_by = paginate_by
         self._verify_ssl = verify_ssl
         self.team_id = auth_token.split("=")[-1]
+        self.get_session()
 
     @timed_lru_cache(seconds=360)
     def get_session(self):

tests/integration/test_interface.py

@@ -138,6 +138,18 @@ def test_overlay_fuse(self):
             )
             self.assertIsNotNone(paths)
 
+    def test_upload_images_to_project_returned_data(self):
+        upload, not_uploaded, duplicated = sa.upload_images_to_project(
+            self.PROJECT_NAME,
+            [f"{self.folder_path}/{self.EXAMPLE_IMAGE_1}", f"{self.folder_path}/{self.EXAMPLE_IMAGE_2}"]
+        )
+        self.assertEqual(2, len(upload))
+        upload, not_uploaded, duplicated = sa.upload_images_to_project(
+            self.PROJECT_NAME,
+            [f"{self.folder_path}/{self.EXAMPLE_IMAGE_1}", f"{self.folder_path}/{self.EXAMPLE_IMAGE_2}"]
+        )
+        self.assertEqual(2, len(duplicated))
+
     def test_upload_images_to_project_image_quality_in_editor(self):
         self.assertRaises(
             AppException,