Merge pull request #223 from superannotateai/sdk-193-2

Sdk 193 2

Author: VaghinakDev

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

@@ -1,11 +1,9 @@
-import concurrent.futures
 import io
 import json
 import logging
 import os
 import tempfile
 import time
-from collections import namedtuple
 from pathlib import Path
 from typing import Iterable
 from typing import List
@@ -1750,7 +1748,9 @@ def upload_videos_from_folder_to_project(
                     f"{len(duplicates)} already existing images found that won't be uploaded."
                 )
             if not images_to_upload:
-                logger.warning(f"{len(duplicates)} already existing images found that won't be uploaded.")
+                logger.warning(
+                    f"{len(duplicates)} already existing images found that won't be uploaded."
+                )
                 continue
             if use_case.is_valid():
                 with tqdm(
@@ -2009,14 +2009,16 @@ def move_image(
     :type copy_pin: bool
     """
     source_project_name, source_folder_name = extract_project_folder(source_project)
-    destination_project_name, destination_folder = extract_project_folder(destination_project)
+    destination_project_name, destination_folder = extract_project_folder(
+        destination_project
+    )
     response = controller.move_image(
         from_project_name=source_project_name,
         from_folder_name=source_folder_name,
         to_project_name=destination_project_name,
         to_folder_name=destination_folder,
         image_name=image_name,
-        copy_annotation_status=copy_annotation_status
+        copy_annotation_status=copy_annotation_status,
     )
     if response.errors:
         raise AppException(response.errors)
@@ -2269,11 +2271,13 @@ def attach_image_urls_to_project(
         annotation_status=annotation_status,
     )
     if use_case.is_valid():
-        with tqdm(total=use_case.attachments_count, desc="Attaching urls") as progress_bar:
+        with tqdm(
+            total=use_case.attachments_count, desc="Attaching urls"
+        ) as progress_bar:
             for _ in use_case.execute():
                 progress_bar.update(1)
         uploaded, duplications = use_case.data
-        uploaded = [i['name'] for i in uploaded]
+        uploaded = [i["name"] for i in uploaded]
         duplications.extend(duplicate_images)
         failed_images = [
             image["name"]
@@ -2320,11 +2324,13 @@ def attach_video_urls_to_project(
         annotation_status=annotation_status,
     )
     if use_case.is_valid():
-        with tqdm(total=use_case.attachments_count, desc="Attaching urls") as progress_bar:
+        with tqdm(
+            total=use_case.attachments_count, desc="Attaching urls"
+        ) as progress_bar:
             for _ in use_case.execute():
                 progress_bar.update(1)
         uploaded, duplications = use_case.data
-        uploaded = [i['name'] for i in uploaded]
+        uploaded = [i["name"] for i in uploaded]
         duplications.extend(duplicate_images)
         failed_images = [
             image["name"]
@@ -3318,7 +3324,7 @@ def upload_image_to_project(
         image=img,
         annotation_status=annotation_status,
         from_s3_bucket=from_s3_bucket,
-        image_quality_in_editor=image_quality_in_editor
+        image_quality_in_editor=image_quality_in_editor,
     )
     if response.errors:
         raise AppException(response.errors)
@@ -3529,11 +3535,13 @@ def attach_document_urls_to_project(
         annotation_status=annotation_status,
     )
     if use_case.is_valid():
-        with tqdm(total=use_case.attachments_count, desc="Attaching urls") as progress_bar:
+        with tqdm(
+            total=use_case.attachments_count, desc="Attaching urls"
+        ) as progress_bar:
             for _ in use_case.execute():
                 progress_bar.update(1)
         uploaded, duplications = use_case.data
-        uploaded = [i['name'] for i in uploaded]
+        uploaded = [i["name"] for i in uploaded]
         duplications.extend(duplicate_images)
         failed_images = [
             image["name"]

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

@@ -18,10 +18,10 @@
 
 class CreateFolderUseCase(BaseUseCase):
     def __init__(
-            self,
-            project: ProjectEntity,
-            folder: FolderEntity,
-            folders: BaseManageableRepository,
+        self,
+        project: ProjectEntity,
+        folder: FolderEntity,
+        folders: BaseManageableRepository,
     ):
         super().__init__()
         self._project = project
@@ -32,12 +32,12 @@ def validate_folder(self):
         if not self._folder.name:
             raise AppValidationException("Folder name cannot be empty.")
         if (
-                len(
-                    set(self._folder.name).intersection(
-                        constances.SPECIAL_CHARACTERS_IN_PROJECT_FOLDER_NAMES
-                    )
+            len(
+                set(self._folder.name).intersection(
+                    constances.SPECIAL_CHARACTERS_IN_PROJECT_FOLDER_NAMES
                 )
-                > 0
+            )
+            > 0
         ):
             self._folder.name = "".join(
                 "_"
@@ -56,14 +56,13 @@ def execute(self):
         return self._response
 
 
-
 class GetFolderUseCase(BaseUseCase):
     def __init__(
-            self,
-            project: ProjectEntity,
-            folders: BaseReadOnlyRepository,
-            folder_name: str,
-            team_id: int,
+        self,
+        project: ProjectEntity,
+        folders: BaseReadOnlyRepository,
+        folder_name: str,
+        team_id: int,
     ):
         super().__init__()
         self._project = project
@@ -73,9 +72,9 @@ def __init__(
 
     def execute(self):
         condition = (
-                Condition("name", self._folder_name, EQ)
-                & Condition("team_id", self._team_id, EQ)
-                & Condition("project_id", self._project.uuid, EQ)
+            Condition("name", self._folder_name, EQ)
+            & Condition("team_id", self._team_id, EQ)
+            & Condition("project_id", self._project.uuid, EQ)
         )
         try:
             self._response.data = self._folders.get_one(condition)
@@ -86,12 +85,12 @@ def execute(self):
 
 class SearchFoldersUseCase(BaseUseCase):
     def __init__(
-            self,
-            project: ProjectEntity,
-            folders: BaseReadOnlyRepository,
-            condition: Condition,
-            folder_name: str = None,
-            include_users=False,
+        self,
+        project: ProjectEntity,
+        folders: BaseReadOnlyRepository,
+        condition: Condition,
+        folder_name: str = None,
+        include_users=False,
     ):
         super().__init__()
         self._project = project
@@ -102,10 +101,10 @@ def __init__(
 
     def execute(self):
         condition = (
-                self._condition
-                & Condition("project_id", self._project.uuid, EQ)
-                & Condition("team_id", self._project.team_id, EQ)
-                & Condition("includeUsers", self._include_users, EQ)
+            self._condition
+            & Condition("project_id", self._project.uuid, EQ)
+            & Condition("team_id", self._project.team_id, EQ)
+            & Condition("includeUsers", self._include_users, EQ)
         )
         if self._folder_name:
             condition &= Condition("name", self._folder_name, EQ)
@@ -115,10 +114,10 @@ def execute(self):
 
 class DeleteFolderUseCase(BaseUseCase):
     def __init__(
-            self,
-            project: ProjectEntity,
-            folders: BaseManageableRepository,
-            folders_to_delete: List[FolderEntity],
+        self,
+        project: ProjectEntity,
+        folders: BaseManageableRepository,
+        folders_to_delete: List[FolderEntity],
     ):
         super().__init__()
         self._project = project
@@ -137,7 +136,7 @@ def execute(self):
 
 class UpdateFolderUseCase(BaseUseCase):
     def __init__(
-            self, folders: BaseManageableRepository, folder: FolderEntity,
+        self, folders: BaseManageableRepository, folder: FolderEntity,
     ):
         super().__init__()
         self._folders = folders
@@ -147,12 +146,12 @@ def validate_folder(self):
         if not self._folder.name:
             raise AppValidationException("Folder name cannot be empty.")
         if (
-                len(
-                    set(self._folder.name).intersection(
-                        constances.SPECIAL_CHARACTERS_IN_PROJECT_FOLDER_NAMES
-                    )
+            len(
+                set(self._folder.name).intersection(
+                    constances.SPECIAL_CHARACTERS_IN_PROJECT_FOLDER_NAMES
                 )
-                > 0
+            )
+            > 0
         ):
             self._folder.name = "".join(
                 "_"
@@ -175,11 +174,11 @@ def execute(self):
 
 class AssignFolderUseCase(BaseUseCase):
     def __init__(
-            self,
-            service: SuerannotateServiceProvider,
-            project_entity: ProjectEntity,
-            folder: FolderEntity,
-            users: List[str],
+        self,
+        service: SuerannotateServiceProvider,
+        project_entity: ProjectEntity,
+        folder: FolderEntity,
+        users: List[str],
     ):
         super().__init__()
         self._service = service