Merge pull request #452 from superannotateai/1119_get_subsets

initial commit

Author: VaghinakDev

docs/source/superannotate.sdk.rst

@@ -73,11 +73,20 @@ ______
 .. automethod:: superannotate.SAClient.attach_items
 .. automethod:: superannotate.SAClient.copy_items
 .. automethod:: superannotate.SAClient.move_items
+.. automethod:: superannotate.SAClient.assign_items
+.. automethod:: superannotate.SAClient.unassign_items
 .. automethod:: superannotate.SAClient.get_item_metadata
 .. automethod:: superannotate.SAClient.set_annotation_statuses
 
 ----------
 
+Subsets
+______
+
+.. automethod:: superannotate.SAClient.get_subsets
+
+----------
+
 Images
 ______
 

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

@@ -2744,3 +2744,19 @@ def download_annotations(
         if response.errors:
             raise AppException(response.errors)
         return response.data
+
+    def get_subsets(self, project: Union[NotEmptyStr, dict]):
+        """Get Subsets
+
+        :param project: project name (e.g., “project1”)
+        :type project: str
+
+        :return: subsets’ metadata
+        :rtype: list of dicts
+        """
+        project_name, _ = extract_project_folder(project)
+
+        response = self.controller.list_subsets(project_name)
+        if response.errors:
+            raise AppException(response.errors)
+        return BaseSerializer.serialize_iterable(response.data)

src/superannotate/lib/core/entities/base.py

@@ -27,6 +27,13 @@ def validate(cls, v: datetime):
         return v.isoformat()
 
 
+class SubSetEntity(BaseModel):
+    name: str
+
+    class Config:
+        extra = Extra.ignore
+
+
 class TimedBaseModel(BaseModel):
     createdAt: StringDate = Field(None, alias="createdAt")
     updatedAt: StringDate = Field(None, alias="updatedAt")

src/superannotate/lib/core/service_types.py

@@ -6,6 +6,7 @@
 
 from pydantic import BaseModel
 from pydantic import Extra
+from pydantic import parse_obj_as
 
 
 class Limit(BaseModel):
@@ -81,7 +82,7 @@ def __init__(self, response, content_type=None):
         }
         try:
             if content_type and content_type is not self.__class__:
-                data["data"] = content_type(**response.json())
+                data["data"] = parse_obj_as(content_type, response.json())
             else:
                 data["data"] = response.json()
         except Exception as e:

src/superannotate/lib/core/serviceproviders.py

@@ -368,3 +368,6 @@ def saqul_query(
 
     def validate_saqul_query(self, team_id: int, project_id: int, query: str) -> dict:
         raise NotImplementedError
+
+    def list_sub_sets(self, team_id: int, project_id: int) -> ServiceResponse:
+        raise NotImplementedError

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

@@ -19,6 +19,7 @@
 from lib.core.reporter import Reporter
 from lib.core.repositories import BaseManageableRepository
 from lib.core.repositories import BaseReadOnlyRepository
+from lib.core.response import Response
 from lib.core.serviceproviders import SuperannotateServiceProvider
 from lib.core.usecases.base import BaseReportableUseCase
 from lib.core.usecases.base import BaseUseCase
@@ -1079,3 +1080,38 @@ def execute(self):
                     )
             self._response.data = invited, list(to_skip)
             return self._response
+
+
+class ListSubsetsUseCase(BaseReportableUseCase):
+    def __init__(
+        self,
+        reporter: Reporter,
+        project: ProjectEntity,
+        backend_client: SuperannotateServiceProvider,
+    ):
+        super().__init__(reporter)
+        self._project = project
+        self._backend_client = backend_client
+
+    def validate_project(self):
+        if self._project.sync_status != constances.ProjectState.SYNCED.value:
+            raise AppException("Data is not synced.")
+
+        response = self._backend_client.validate_saqul_query(
+            self._project.team_id, self._project.id, "_"
+        )
+        error = response.get("error")
+        if error:
+            raise AppException(response["error"])
+
+    def execute(self) -> Response:
+        if self.is_valid():
+            sub_sets_response = self._backend_client.list_sub_sets(
+                team_id=self._project.team_id, project_id=self._project.id
+            )
+            if sub_sets_response.ok:
+                self._response.data = []
+            else:
+                self._response.data = []
+
+        return self._response

src/superannotate/lib/infrastructure/controller.py

@@ -1689,3 +1689,12 @@ def download_annotations(
             callback=callback,
         )
         return use_case.execute()
+
+    def list_subsets(self, project_name: str):
+        project = self._get_project(project_name)
+        use_case = usecases.ListSubsetsUseCase(
+            reporter=self.get_default_reporter(),
+            project=project,
+            backend_client=self.backend_client,
+        )
+        return use_case.execute()

src/superannotate/lib/infrastructure/services.py

@@ -14,6 +14,7 @@
 
 import lib.core as constance
 import requests.packages.urllib3
+from lib.core import entities
 from lib.core.exceptions import AppException
 from lib.core.exceptions import BackendError
 from lib.core.reporter import Reporter
@@ -240,6 +241,7 @@ class SuperannotateBackendService(BaseBackendService):
     URL_ATTACH_INTEGRATIONS = "image/integration/create"
     URL_SAQUL_QUERY = "/images/search/advanced"
     URL_VALIDATE_SAQUL_QUERY = "/images/parse/query/advanced"
+    URL_LIST_SUBSETS = "/project/{project_id}/subset"
 
     def upload_priority_scores(
         self, team_id: int, project_id: int, folder_id: int, priorities: list
@@ -1190,3 +1192,11 @@ def validate_saqul_query(self, team_id: int, project_id: int, query: str) -> dic
         return self._request(
             validate_query_url, "post", params=params, data=data
         ).json()
+
+    def list_sub_sets(self, team_id: int, project_id: int) -> ServiceResponse:
+        return self._request(
+            urljoin(self.api_url, self.URL_LIST_SUBSETS.format(project_id=project_id)),
+            "get",
+            params=dict(team_id=team_id),
+            content_type=List[entities.SubSetEntity],
+        )

tests/integration/test_benchmark.py

@@ -3,7 +3,9 @@
 from os.path import dirname
 
 import pytest
+
 from src.superannotate import SAClient
+
 sa = SAClient()
 from tests.integration.base import BaseTestCase