Merge pull request #694 from superannotateai/FRIDAY-2919

Item interface

Author: VaghinakDev

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

@@ -4,6 +4,7 @@
 import logging
 import os
 import sys
+import uuid
 from pathlib import Path
 from typing import Callable
 from typing import Dict
@@ -879,12 +880,10 @@ def delete_items(self, project: str, items: Optional[List[str]] = None):
         :param items: to be deleted items' names. If None, all the items will be deleted
         :type items: list of str
         """
-        project, folder = self.controller.get_project_folder_by_path(project)
-        response = self.controller.items.delete(
-            project=project, folder=folder, item_names=items
-        )
-        if response.errors:
-            raise AppException(response.errors)
+        project_name, folder_name = extract_project_folder(project)
+        project = self.controller.get_project(project_name)
+        folder = project.get_folder(folder_name)
+        folder.delete_items(item_names=items)
 
     def assign_items(
         self, project: Union[NotEmptyStr, dict], items: List[str], user: str
@@ -904,13 +903,10 @@ def assign_items(
         :type user: str
         """
 
-        project, folder = self.controller.get_project_folder_by_path(project)
-        response = self.controller.projects.assign_items(
-            project, folder, item_names=items, user=user
-        )
-
-        if response.errors:
-            raise AppException(response.errors)
+        project_name, folder_name = extract_project_folder(project)
+        project = self.controller.get_project(project_name)
+        folder = project.get_folder(folder_name)
+        folder.assign_items(user=user, item_names=items)
 
     def unassign_items(
         self, project: Union[NotEmptyStr, dict], items: List[NotEmptyStr]
@@ -924,12 +920,10 @@ def unassign_items(
         :param items: list of items to unassign
         :type items: list of str
         """
-        project, folder = self.controller.get_project_folder_by_path(project)
-        response = self.controller.projects.un_assign_items(
-            project, folder, item_names=items
-        )
-        if response.errors:
-            raise AppException(response.errors)
+        project_name, folder_name = extract_project_folder(project)
+        project = self.controller.get_project(project_name)
+        folder = project.get_folder(folder_name)
+        folder.assign_items(item_names=items)
 
     def unassign_folder(self, project_name: NotEmptyStr, folder_name: NotEmptyStr):
         """Removes assignment of given folder for all assignees.
@@ -2676,8 +2670,6 @@ def attach_items(
         project = self.controller.get_project(project_name)
         folder = project.get_folder(folder_name)
         # todo validate Attachment
-        names = []
-        import uuid
 
         seen_names = set()
         unique_attachments = []

tests/integration/items/test_attach_items.py

@@ -1,6 +1,5 @@
 import os
 from pathlib import Path
-from unittest import TestCase
 
 import pytest
 from src.superannotate import AppException
@@ -88,10 +87,11 @@ def test_limitation(self):
         )
 
 
-class TestAttachItemsVectorArguments(TestCase):
+class TestAttachItemsVectorArguments(BaseTestCase):
     PROJECT_NAME = "TestAttachItemsVectorArguments"
 
     def test_attach_items_invalid_payload(self):
+        # todo update
         error_msg = [
             "attachments",
             "str type expected",