Merge pull request #229 from superannotateai/346

Fix

Author: VaghinakDev

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

@@ -2259,7 +2259,11 @@ def attach_image_urls_to_project(
         logger.warning(
             constances.ALREADY_EXISTING_FILES_WARNING.format(len(duplicate_images))
         )
-    logger.info(constances.ATTACHING_FILES_MESSAGE.format(len(images_to_upload),project_folder_name))
+    logger.info(
+        constances.ATTACHING_FILES_MESSAGE.format(
+            len(images_to_upload), project_folder_name
+        )
+    )
     if use_case.is_valid():
         with tqdm(
             total=use_case.attachments_count, desc="Attaching urls"
@@ -2312,7 +2316,11 @@ def attach_video_urls_to_project(
         logger.warning(
             constances.ALREADY_EXISTING_FILES_WARNING.format(len(duplicate_images))
         )
-    logger.info(constances.ATTACHING_FILES_MESSAGE.format(len(images_to_upload),project_folder_name))
+    logger.info(
+        constances.ATTACHING_FILES_MESSAGE.format(
+            len(images_to_upload), project_folder_name
+        )
+    )
     if use_case.is_valid():
         with tqdm(
             total=use_case.attachments_count, desc="Attaching urls"
@@ -2405,6 +2413,9 @@ def upload_annotations_from_folder_to_project(
                 progress_bar.update(1)
     else:
         raise AppException(use_case.response.errors)
+    if use_case.response.report:
+        for i in use_case.response.report_messages:
+            logger.info(i)
     return use_case.data
 
 
@@ -3528,7 +3539,11 @@ def attach_document_urls_to_project(
         logger.warning(
             constances.ALREADY_EXISTING_FILES_WARNING.format(len(duplicate_images))
         )
-    logger.info(constances.ATTACHING_FILES_MESSAGE.format(len(images_to_upload),project_folder_name))
+    logger.info(
+        constances.ATTACHING_FILES_MESSAGE.format(
+            len(images_to_upload), project_folder_name
+        )
+    )
     if use_case.is_valid():
         with tqdm(
             total=use_case.attachments_count, desc="Attaching urls"

src/superannotate/lib/core/response.py

@@ -5,6 +5,7 @@ class Response:
     def __init__(self, status: str = None, data: Union[dict, list] = None):
         self._status = status
         self._data = data
+        self._report = []
         self._errors = []
 
     @property
@@ -15,6 +16,18 @@ def data(self):
     def data(self, value):
         self._data = value
 
+    @property
+    def report(self):
+        return "\n".join(self._report)
+
+    @report.setter
+    def report(self, value: str):
+        self._report.append(value)
+
+    @property
+    def report_messages(self):
+        return self._report
+
     @property
     def status(self):
         return self.data

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

@@ -238,7 +238,6 @@ def _validate_limitations(self, to_upload_count):
         ):
             raise AppValidationException(constances.ATTACH_USER_LIMIT_ERROR_MESSAGE)
 
-
     @property
     def annotation_status_code(self):
         if self._annotation_status:
@@ -1466,7 +1465,8 @@ def execute(self):
                     folder=self._folder,
                     backend_service_provider=self._backend_client,
                     attachments=[
-                        image.entity for image in uploaded_images[i : i + 100]  # noqa: E203
+                        image.entity
+                        for image in uploaded_images[i : i + 100]  # noqa: E203
                     ],
                     annotation_status=self._annotation_status,
                     upload_state_code=constances.UploadState.BASIC.value,
@@ -1594,8 +1594,8 @@ def __init__(
     def auth_data(self):
         if not self._auth_data:
             self._auth_data = self._backend_service.get_s3_upload_auth_token(
-            self._project.team_id, self._folder.uuid, self._project.uuid
-        )
+                self._project.team_id, self._folder.uuid, self._project.uuid
+            )
         return self._auth_data
 
     @property
@@ -1926,7 +1926,9 @@ def execute(self):
                 response = AttachFileUrlsUseCase(
                     project=self._project,
                     folder=self._folder,
-                    attachments=self._attachments[i : i + self.CHUNK_SIZE],  # noqa: E203
+                    attachments=self._attachments[
+                        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,
@@ -2779,9 +2781,11 @@ def upload_to_s3(
                         file = io.BytesIO(mask_file.read())
                 bucket.put_object(Key=image_info["annotation_bluemap_path"], Body=file)
             return image_id_name_map[image_id], True
-        except Exception as _:
+        except Exception as e:
+            self._response.report = f"Couldn't upload annotation {image_id_name_map[image_id].name} - {str(e)}"
             return image_id_name_map[image_id], False
 
+
     def report_missing_data(self):
         if self.missing_classes:
             logger.warning(f"Couldn't find classes [{', '.join(self.missing_classes)}]")

src/superannotate/lib/infrastructure/controller.py

@@ -68,8 +68,10 @@ def init(self, config_path):
             self.configs.get_one("token"),
             self.configs.get_one("main_endpoint"),
         )
-        if token: token = token.value
-        if main_endpoint: main_endpoint = main_endpoint.value
+        if token:
+            token = token.value
+        if main_endpoint:
+            main_endpoint = main_endpoint.value
         if not main_endpoint:
             self.configs.insert(ConfigEntity("main_endpoint", constances.BACKEND_URL))
         if not token: