tod

Author: VaghinakDev

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

@@ -2112,7 +2112,7 @@ def upload_image_annotations(
         mask=mask,
         verbose=verbose,
     )
-    if response.errors and not response.errors == constances.INVALID_JSON_MESSSAGE:
+    if response.errors and not response.errors == constances.INVALID_JSON_MESSAGE:
         raise AppException(response.errors)
 
 

src/superannotate/lib/core/__init__.py

@@ -123,7 +123,7 @@
     "which an annotation is invalid."
 )
 
-INVALID_JSON_MESSSAGE = "Invalid json"
+INVALID_JSON_MESSAGE = "Invalid json"
 
 __alL__ = (
     ProjectType,

src/superannotate/lib/core/reporter.py

@@ -24,6 +24,12 @@ def __init__(
         self.custom_messages = defaultdict(set)
         self.progress_bar = None
 
+    def disable_warnings(self):
+        self._log_warning = False
+
+    def enable_warnings(self):
+        self._log_warning = True
+
     def log_info(self, value: str):
         if self._log_info:
             self.logger.info(value)

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

@@ -168,6 +168,7 @@ def _upload_annotation(
         bucket,
     ):
         try:
+            self.reporter.disable_warnings()
             response = UploadAnnotationUseCase(
                 project=self._project,
                 folder=self._folder,
@@ -196,6 +197,8 @@ def _upload_annotation(
         except Exception as e:
             logger.debug(str(e), exc_info=True)
             return path, False
+        finally:
+            self.reporter.enable_warnings()
 
     def get_bucket_to_upload(self, ids: List[int]):
         upload_data = self.get_annotation_upload_data(ids)
@@ -485,13 +488,11 @@ def execute(self):
                 )
                 self._images.update(self._image)
                 if self._verbose:
-                    logger.info(
-                        "Uploading annotations for image %s in project %s.",
-                        str(self._image.name),
-                        self._project.name,
+                    self.reporter.log_info(
+                        f"Uploading annotations for image {str(self._image.name)} in project {self._project.name}."
                     )
             else:
-                self._response.errors = constances.INVALID_JSON_MESSSAGE
+                self._response.errors = constances.INVALID_JSON_MESSAGE
                 self.reporter.store_message("invalid_jsons", self._annotation_path)
                 self.reporter.log_warning(
                     f"Couldn't validate annotations. {constances.USE_VALIDATE_MESSAGE}"

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

@@ -40,7 +40,6 @@
 from lib.core.repositories import BaseManageableRepository
 from lib.core.repositories import BaseReadOnlyRepository
 from lib.core.response import Response
-from lib.core.reporter import Progress
 from lib.core.serviceproviders import SuerannotateServiceProvider
 from lib.core.usecases.base import BaseInteractiveUseCase
 from lib.core.usecases.base import BaseReportableUseCae

src/superannotate/lib/infrastructure/controller.py

@@ -216,9 +216,7 @@ def team_id(self) -> int:
 
     @property
     def default_reporter(self):
-        if not self._reporter:
-            self._reporter = Reporter()
-        return self._reporter
+        return Reporter()
 
     @timed_lru_cache(seconds=3600)
     def get_auth_data(self, project_id: int, team_id: int, folder_id: int):
@@ -1313,7 +1311,7 @@ def upload_image_annotations(
             backend_service_provider=self._backend_client,
             mask=mask,
             verbose=verbose,
-            reporter=Reporter(log_info=True, log_warning=True),
+            reporter=self.default_reporter,
             validators=self.annotation_validators,
         )
         return use_case.execute()