Add logger to response

Author: dshabin

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

@@ -2409,6 +2409,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
 
 

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

@@ -2502,7 +2502,6 @@ def __init__(
         self.missing_attribute_groups = set()
         self.missing_classes = set()
         self.missing_attributes = set()
-        self._failed_annotation_reports = []
 
     @property
     def s3_client(self):
@@ -2744,7 +2743,6 @@ def execute(self):
                 missing_annotations,
             )
         self.report_missing_data()
-        self.report_failed_annotations()
         return self._response
 
     def upload_to_s3(
@@ -2784,16 +2782,9 @@ def upload_to_s3(
                 bucket.put_object(Key=image_info["annotation_bluemap_path"], Body=file)
             return image_id_name_map[image_id], True
         except Exception as e:
-            self._failed_annotation_reports.append(
-                {"image": image_id_name_map[image_id].name, "message": str(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_failed_annotations(self):
-        for info in self._failed_annotation_reports:
-            logger.warning(
-                f"Couldn't upload annotation {info['image']} - {info['message']}"
-            )
 
     def report_missing_data(self):
         if self.missing_classes: