Merge branch 'friday' of github.com:superannotateai/superannotate-python-sdk into friday

Author: VahagnTKA

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

@@ -795,13 +795,18 @@ def assign_folder(
         :type users: list of str
         """
 
-        contributors = (
-            self.controller.projects.get_by_name(
-                project_name=project_name, include_contributors=True
-            )
-            .data["project"]
-            .users
+        response = self.controller.projects.get_by_name(name=project_name)
+        if response.errors:
+            raise AppException(response.errors)
+        project = response.data
+        response = self.controller.projects.get_metadata(
+            project=project, include_contributors=True
         )
+
+        if response.errors:
+            raise AppException(response.errors)
+
+        contributors = response.data.users
         verified_users = [i["user_id"] for i in contributors]
         verified_users = set(users).intersection(set(verified_users))
         unverified_contributor = set(users) - verified_users

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

@@ -849,6 +849,7 @@ def execute(self) -> Response:
 
 class UploadImagesToProject(BaseInteractiveUseCase):
     MAX_WORKERS = 10
+    LIST_NAME_CHUNK_SIZE = 500
 
     def __init__(
         self,
@@ -1023,24 +1024,30 @@ def filter_paths(self, paths: List[str]):
         for path in paths:
             name_path_map[Path(path).name].append(path)
 
+        CHUNK_SIZE = UploadImagesToProject.LIST_NAME_CHUNK_SIZE
         filtered_paths = []
         duplicated_paths = []
         for file_name in name_path_map:
             if len(name_path_map[file_name]) > 1:
                 duplicated_paths.append(name_path_map[file_name][1:])
             filtered_paths.append(name_path_map[file_name][0])
 
-        response = self._service_provider.items.list_by_names(
-            project=self._project,
-            folder=self._folder,
-            names=[image.split("/")[-1] for image in filtered_paths],
-        )
-        if not response.ok:
-            self._response.errors = AppException(response.error)
-            return self._response
+        image_list = []
+        for i in range(0, len(filtered_paths), CHUNK_SIZE):
+            response = self._service_provider.items.list_by_names(
+                project=self._project,
+                folder=self._folder,
+                names=[
+                    image.split("/")[-1] for image in filtered_paths[i : i + CHUNK_SIZE]
+                ],
+            )
 
+            if not response.ok:
+                raise AppException(response.error)
+            image_list.extend([image.name for image in response.data])
+
+        image_list=set(image_list)
         images_to_upload = []
-        image_list = [image.name for image in response.data]
 
         for path in filtered_paths:
             if Path(path).name not in image_list:
@@ -1593,15 +1600,14 @@ def fill_classes_data(self, annotations: dict):
                 attribute["groupName"] = annotation_class["attribute_groups"][
                     attribute["groupId"]
                 ]["name"]
-                if attribute["id"] not in list(
+                if attribute.get("id") in list(
                     annotation_class["attribute_groups"][attribute["groupId"]][
                         "attributes"
                     ].keys()
                 ):
-                    continue
-                attribute["name"] = annotation_class["attribute_groups"][
-                    attribute["groupId"]
-                ]["attributes"][attribute["id"]]
+                    attribute["name"] = annotation_class["attribute_groups"][
+                        attribute["groupId"]
+                    ]["attributes"][attribute["id"]]
 
     def execute(self):
         if self.is_valid():