validate

Author: dshabin

src/superannotate/lib/core/__init__.py

@@ -83,12 +83,18 @@
 ATTACH_USER_LIMIT_ERROR_MESSAGE = "The number of items you want to attach  exceeds the limit of your subscription plan."
 
 
-COPY_FOLDER_LIMIT_ERROR_MESSAGE = "The number of items you want to copy exceeds the limit of 50 000 items per folder."
+COPY_FOLDER_LIMIT_ERROR_MESSAGE = (
+    "The number of items you want to copy exceeds the limit of 50 000 items per folder."
+)
 COPY_PROJECT_LIMIT_ERROR_MESSAGE = "The number of items you want to copy exceeds the limit of 500 000 items per project."
-COPY_SUPER_LIMIT_ERROR_MESSAGE = "The number of items you want to copy exceeds the limit of your subscription plan."
+COPY_SUPER_LIMIT_ERROR_MESSAGE = (
+    "The number of items you want to copy exceeds the limit of your subscription plan."
+)
 
 
-MOVE_FOLDER_LIMIT_ERROR_MESSAGE = "The number of items you want to move exceeds the limit of 50 000 items per folder."
+MOVE_FOLDER_LIMIT_ERROR_MESSAGE = (
+    "The number of items you want to move exceeds the limit of 50 000 items per folder."
+)
 MOVE_PROJECT_LIMIT_ERROR_MESSAGE = "The number of items you want to move exceeds the limit of 500 000 items per project."
 
 __alL__ = (

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

@@ -833,14 +833,17 @@ def execute(self):
 
                 outline_color = 4 * (255,)
                 for instance in self.annotations["instances"]:
-                    if (not instance.get("className")) or (not class_color_map.get(instance["className"])):
+                    if (not instance.get("className")) or (
+                        not class_color_map.get(instance["className"])
+                    ):
                         continue
                     color = class_color_map.get(instance["className"])
                     if not color:
                         class_color_map[instance["className"]] = self.generate_color()
                     for image in images:
                         fill_color = (
-                            *class_color_map[instance["className"]], 255 if image.type == "fuse" else self.TRANSPARENCY
+                            *class_color_map[instance["className"]],
+                            255 if image.type == "fuse" else self.TRANSPARENCY,
                         )
                         if instance["type"] == "bbox":
                             image.content.draw_bbox(
@@ -909,7 +912,9 @@ def execute(self):
                 weight, height = image.get_size()
                 empty_image_arr = np.full((height, weight, 4), [0, 0, 0, 255], np.uint8)
                 for annotation in self.annotations["instances"]:
-                    if (not annotation.get("className")) or (not class_color_map.get(annotation["className"])):
+                    if (not annotation.get("className")) or (
+                        not class_color_map.get(annotation["className"])
+                    ):
                         continue
                     fill_color = *class_color_map[annotation["className"]], 255
                     for part in annotation["parts"]:
@@ -1989,17 +1994,28 @@ def validate_limitations(self):
         if self._move:
             if self._from_project.uuid == self._to_project.uuid:
                 if self._from_folder.uuid == self._to_folder.uuid:
-                    raise AppValidationException("Cannot move image if source_project == destination_project.")
+                    raise AppValidationException(
+                        "Cannot move image if source_project == destination_project."
+                    )
                 elif response.data.folder_limit.remaining_image_count < 1:
-                    raise AppValidationException(constances.MOVE_FOLDER_LIMIT_ERROR_MESSAGE)
+                    raise AppValidationException(
+                        constances.MOVE_FOLDER_LIMIT_ERROR_MESSAGE
+                    )
             elif response.data.project_limit.remaining_image_count < 1:
-                raise AppValidationException(constances.MOVE_PROJECT_LIMIT_ERROR_MESSAGE)
+                raise AppValidationException(
+                    constances.MOVE_PROJECT_LIMIT_ERROR_MESSAGE
+                )
         else:
             if response.data.folder_limit.remaining_image_count < 1:
                 raise AppValidationException(constances.COPY_FOLDER_LIMIT_ERROR_MESSAGE)
             if response.data.project_limit.remaining_image_count < 1:
-                raise AppValidationException(constances.COPY_PROJECT_LIMIT_ERROR_MESSAGE)
-            if response.data.super_user_limit and response.data.super_user_limit.remaining_image_count < 1:
+                raise AppValidationException(
+                    constances.COPY_PROJECT_LIMIT_ERROR_MESSAGE
+                )
+            if (
+                response.data.super_user_limit
+                and response.data.super_user_limit.remaining_image_count < 1
+            ):
                 raise AppValidationException(constances.COPY_SUPER_LIMIT_ERROR_MESSAGE)
 
     def execute(self) -> Response:
@@ -2722,16 +2738,16 @@ def upload_to_s3(
                 logger.warning(f"Invalid json {image_id_name_map[image_id].path}")
                 return image_id_name_map[image_id], False
             bucket.put_object(
-                Key=image_info["annotation_json_path"], Body=json.dumps(annotation_json),
+                Key=image_info["annotation_json_path"],
+                Body=json.dumps(annotation_json),
             )
             if self._project.project_type == constances.ProjectType.PIXEL.value:
-                mask_path = image_id_name_map[image_id].path.replace("___pixel.json", constances.ANNOTATION_MASK_POSTFIX)
+                mask_path = image_id_name_map[image_id].path.replace(
+                    "___pixel.json", constances.ANNOTATION_MASK_POSTFIX
+                )
                 if from_s3:
                     file = io.BytesIO()
-                    s3_object = from_s3.Object(
-                        self._client_s3_bucket,
-                        mask_path
-                    )
+                    s3_object = from_s3.Object(self._client_s3_bucket, mask_path)
                     s3_object.download_fileobj(file)
                     file.seek(0)
                 else: