superannotate 4.4.28__tar.gz → 4.4.29.dev1__tar.gz

{superannotate-4.4.28/src/superannotate.egg-info → superannotate-4.4.29.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: superannotate
-Version: 4.4.28
+Version: 4.4.29.dev1
 Summary: Python SDK to SuperAnnotate platform
 Home-page: https://github.com/superannotateai/superannotate-python-sdk
 Author: SuperAnnotate AI

{superannotate-4.4.28 → superannotate-4.4.29.dev1}/src/superannotate/__init__.py

@@ -3,7 +3,7 @@ import os
 import sys
 
 
-__version__ = "4.4.28"
+__version__ = "4.4.29dev1"
 
 os.environ.update({"sa_version": __version__})
 sys.path.append(os.path.split(os.path.realpath(__file__))[0])

{superannotate-4.4.28 → superannotate-4.4.29.dev1}/src/superannotate/lib/app/interface/sdk_interface.py

@@ -5,6 +5,7 @@ import json
 import logging
 import os
 import sys
+import typing
 import warnings
 from pathlib import Path
 from typing import Any
@@ -277,6 +278,55 @@ class SAClient(BaseInterfaceFacade, metaclass=TrackableMeta):
         response = self.controller.get_team()
         return TeamSerializer(response.data).serialize()
 
+    def get_component_config(self, project: Union[NotEmptyStr, int], component_id: str):
+        """
+        Retrieves the configuration for a given project and component ID.
+
+        :param project: The identifier of the project, which can be a string or an integer representing the project ID.
+        :type project: Union[str, int]
+
+        :param component_id: The ID of the component for which the context is to be retrieved.
+        :type component_id: str
+
+        :return: The context associated with the `webComponent`.
+        :rtype: Any
+
+        :raises AppException: If the project type is not `MULTIMODAL` or no `webComponent` context is found.
+        """
+
+        def retrieve_context(
+            component_data: List[dict], component_pk: str
+        ) -> Tuple[bool, typing.Any]:
+            for component in component_data:
+                if (
+                    component["type"] == "webComponent"
+                    and component["id"] == component_pk
+                ):
+                    return True, component.get("context")
+                if component["type"] == "group" and "children" in component:
+                    found, val = retrieve_context(component["children"], component_pk)
+                    if found:
+                        return found, val
+            return False, None
+
+        project = (
+            self.controller.get_project_by_id(project).data
+            if isinstance(project, int)
+            else self.controller.get_project(project)
+        )
+        if project.type != ProjectType.MULTIMODAL:
+            raise AppException(
+                "This function is only supported for Multimodal projects."
+            )
+
+        editor_template = self.controller.projects.get_editor_template(project)
+        components = editor_template.get("components", [])
+
+        _found, _context = retrieve_context(components, component_id)
+        if not _found:
+            raise AppException("No component context found for project.")
+        return _context
+
     def search_team_contributors(
         self,
         email: EmailStr = None,
@@ -2700,7 +2750,7 @@ class SAClient(BaseInterfaceFacade, metaclass=TrackableMeta):
             ]
         """
         project, folder = self.controller.get_project_folder_by_path(project)
-        query_kwargs = {}
+        query_kwargs = {"include": ["assignments"]}
         if name_contains:
             query_kwargs["name__contains"] = name_contains
         if annotation_status:
@@ -2718,7 +2768,9 @@ class SAClient(BaseInterfaceFacade, metaclass=TrackableMeta):
                     f"{project.name}{f'/{folder.name}' if not folder.is_root else ''}"
                 )
                 _items = self.controller.items.list_items(
-                    project, folder, **query_kwargs
+                    project,
+                    folder,
+                    **query_kwargs,
                 )
                 for i in _items:
                     i.path = path
@@ -2864,7 +2916,10 @@ class SAClient(BaseInterfaceFacade, metaclass=TrackableMeta):
                 ).data
             else:
                 folder = self.controller.get_folder(project, folder)
-        include = include or []
+        _include = {"assignments"}
+        if include:
+            _include.update(set(include))
+        include = list(_include)
         include_custom_metadata = "custom_metadata" in include
         if include_custom_metadata:
             include.remove("custom_metadata")
@@ -2876,7 +2931,7 @@ class SAClient(BaseInterfaceFacade, metaclass=TrackableMeta):
                 project=project, item_ids=[i.id for i in res]
             )
             for i in res:
-                i["custom_metadata"] = item_custom_fields[i.id]
+                i.custom_metadata = item_custom_fields[i.id]
         exclude = {"meta", "annotator_email", "qa_email"}
         if include:
             if "custom_metadata" not in include:

{superannotate-4.4.28 → superannotate-4.4.29.dev1}/src/superannotate/lib/infrastructure/controller.py

@@ -88,6 +88,10 @@ class BaseManager:
 
 
 class ProjectManager(BaseManager):
+    def __init__(self, service_provider: ServiceProvider, team: TeamEntity):
+        super().__init__(service_provider)
+        self._team = team
+
     def get_by_id(self, project_id):
         use_case = usecases.GetProjectByIDUseCase(
             project_id=project_id, service_provider=self.service_provider
@@ -239,6 +243,13 @@ class ProjectManager(BaseManager):
         )
         return use_case.execute()
 
+    def get_editor_template(self, project: ProjectEntity) -> dict:
+        response = self.service_provider.projects.get_editor_template(
+            team=self._team, project=project
+        )
+        response.raise_for_status()
+        return response.data
+
 
 class AnnotationClassManager(BaseManager):
     @timed_lru_cache(seconds=3600)
@@ -974,7 +985,7 @@ class BaseController(metaclass=ABCMeta):
         self._user = self.get_current_user()
         self._team = self.get_team().data
         self.annotation_classes = AnnotationClassManager(self.service_provider)
-        self.projects = ProjectManager(self.service_provider)
+        self.projects = ProjectManager(self.service_provider, team=self._team)
         self.folders = FolderManager(self.service_provider)
         self.items = ItemManager(self.service_provider)
         self.annotations = AnnotationManager(self.service_provider, config)

{superannotate-4.4.28 → superannotate-4.4.29.dev1}/src/superannotate/lib/infrastructure/services/item_service.py

@@ -14,7 +14,7 @@ from superannotate import AppException
 
 
 class ItemService(SuperannotateServiceProvider):
-    MAX_URI_LENGTH = 16_000
+    MAX_URI_LENGTH = 15_000
     URL_LIST = "items"
     URL_GET = "items/{item_id}"
 

{superannotate-4.4.28 → superannotate-4.4.29.dev1/src/superannotate.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: superannotate
-Version: 4.4.28
+Version: 4.4.29.dev1
 Summary: Python SDK to SuperAnnotate platform
 Home-page: https://github.com/superannotateai/superannotate-python-sdk
 Author: SuperAnnotate AI