Merge pull request #423 from superannotateai/friday_580

Added query function.

Author: VaghinakDev

docs/source/superannotate.sdk.rst

@@ -77,6 +77,7 @@ ______
 .. _ref_search_images:
 .. autofunction:: superannotate.search_images
 .. autofunction:: superannotate.search_images_all_folders
+.. autofunction:: superannotate.query
 .. autofunction:: superannotate.get_image_metadata
 .. autofunction:: superannotate.download_image
 .. autofunction:: superannotate.set_image_annotation_status

src/superannotate/__init__.py

@@ -71,6 +71,7 @@
 from superannotate.lib.app.interface.sdk_interface import move_images
 from superannotate.lib.app.interface.sdk_interface import pin_image
 from superannotate.lib.app.interface.sdk_interface import prepare_export
+from superannotate.lib.app.interface.sdk_interface import query
 from superannotate.lib.app.interface.sdk_interface import rename_project
 from superannotate.lib.app.interface.sdk_interface import run_prediction
 from superannotate.lib.app.interface.sdk_interface import search_annotation_classes
@@ -171,6 +172,8 @@
     "search_folders",
     "assign_folder",
     "unassign_folder",
+    # Entities Section
+    "query",
     # Image Section
     "copy_images",
     "move_images",

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

@@ -23,7 +23,6 @@
 from lib.infrastructure.repositories import ConfigRepository
 
 
-
 class CLIFacade(BaseInterfaceFacade):
     """
     With SuperAnnotate CLI, basic tasks can be accomplished using shell commands:

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

@@ -2955,9 +2955,6 @@ def attach_items_from_integrated_storage(
     :param project: project name or folder path where items should be attached (e.g., “project1/folder1”).
     :type project: str
 
-    :param project: project name or folder path where items should be attached (e.g., “project1/folder1”).
-    :type project: str
-
     :param integration:  existing integration name or metadata dict to pull items from.
      Mandatory keys in integration metadata’s dict is “name”.
     :type integration: str or dict
@@ -2972,3 +2969,24 @@ def attach_items_from_integrated_storage(
     response = Controller.get_default().attach_integrations(project_name, folder_name, integration, folder_path)
     if response.errors:
         raise AppException(response.errors)
+
+
+@Trackable
+@validate_arguments
+def query(project: NotEmptyStr, query: Optional[NotEmptyStr]):
+    """Return items
+
+    :param project: project name or folder path (e.g., “project1/folder1”)
+    :type project: str
+
+    :param query: SAQuL query string.
+    :type query: str
+
+    :return: queried items’ metadata list
+    :rtype: list of dicts
+    """
+    project_name, folder_name = extract_project_folder(project)
+    response = Controller.get_default().query_entities(project_name, folder_name, query)
+    if response.errors:
+        raise AppException(response.errors)
+    return BaseSerializers.serialize_iterable(response.data)

src/superannotate/lib/app/serializers.py

@@ -15,8 +15,16 @@ class BaseSerializers(ABC):
     def __init__(self, entity: BaseEntity):
         self._entity = entity
 
+    @staticmethod
+    def _fill_enum_values(data: dict):
+        if isinstance(data, dict):
+            for key, value in data.items():
+                if hasattr(value, "_type") and value._type == "titled_enum":
+                    data[key] = value.__doc__
+        return data
+
     def serialize(self, fields: List[str] = None, by_alias: bool = True, flat: bool = False):
-        return self._serialize(self._entity, fields, by_alias, flat)
+        return self._fill_enum_values(self._serialize(self._entity, fields, by_alias, flat))
 
     @staticmethod
     def _serialize(entity: Any, fields: List[str] = None, by_alias: bool = False, flat: bool = False):
@@ -43,7 +51,9 @@ def serialize_iterable(
     ) -> List[Any]:
         serialized_data = []
         for i in data:
-            serialized_data.append(cls._serialize(i, fields, by_alias, flat))
+            serialized_data.append(
+                cls._fill_enum_values(cls._serialize(i, fields, by_alias, flat))
+            )
         return serialized_data
 
 

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

@@ -3,10 +3,14 @@
 from typing import Any
 from typing import Iterable
 from typing import List
+from typing import Optional
 from typing import Union
 
+from lib.core.enums import AnnotationStatus
 from lib.core.enums import ClassTypeEnum
 from lib.core.enums import SegmentationStatus
+from pydantic import BaseModel
+from pydantic import Field
 from superannotate_schemas.schemas.classes import AnnotationClass
 
 
@@ -477,3 +481,33 @@ def to_dict(self):
             "is_global": self.is_global,
             **self.hyper_parameters,
         }
+
+
+class Entity(BaseModel):
+    id: int
+    name: str
+    path: Optional[str] = Field(None, description="Item’s path in SuperAnnotate project")
+    url: Optional[str] = Field(None, description="Publicly available HTTP address")
+    annotation_status: AnnotationStatus = Field(description="Item annotation status")
+    annotator_name: Optional[str] = Field(description="Annotator email")
+    qa_name: Optional[str] = Field(description="QA email")
+    entropy_value: Optional[str] = Field(description="Priority score of given item")
+    created_at: str = Field(alias="createdAt", description="Date of creation")
+    updated_at: str = Field(alias="updatedAt", description="Update date")
+
+    class Config:
+        ignore_extra = True
+
+
+class TmpImageEntity(Entity):
+    prediction_status: Optional[SegmentationStatus] = Field(SegmentationStatus.NOT_STARTED)
+    segmentation_status: Optional[SegmentationStatus] = Field(SegmentationStatus.NOT_STARTED)
+    approval_status: bool = None
+
+
+class VideoEntity(Entity):
+    pass
+
+
+class DocumentEntity(Entity):
+    pass

src/superannotate/lib/core/enums.py

@@ -2,37 +2,44 @@
 from types import DynamicClassAttribute
 
 
-class BaseTitledEnum(Enum):
+class BaseTitledEnum(int, Enum):
+    def __new__(cls, title, value):
+        obj = int.__new__(cls, value)
+        obj._value_ = value
+        obj.__doc__ = title
+        obj._type = "titled_enum"
+        return obj
+
     @DynamicClassAttribute
     def name(self) -> str:
-        return super().value[0]
+        return self.__doc__
 
     @DynamicClassAttribute
     def value(self):
-        return super().value[1]
+        return super().value
 
     @classmethod
     def get_name(cls, value):
         for enum in list(cls):
             if enum.value == value:
-                return enum.name
+                return enum.__doc__
 
     @classmethod
     def get_value(cls, name):
         for enum in list(cls):
-            if enum.name.lower() == name.lower():
+            if enum.__doc__.lower() == name.lower():
                 return enum.value
 
     @classmethod
     def values(cls):
-        return [enum.name.lower() for enum in list(cls)]
+        return [enum.__doc__.lower() for enum in list(cls)]
 
     @classmethod
     def titles(cls):
-        return [enum.name for enum in list(cls)]
+        return [enum.__doc__ for enum in list(cls)]
 
     def equals(self, other: Enum):
-        return self.name.lower() == other.lower()
+        return self.__doc__.lower() == other.__doc__.lower()
 
 
 class ProjectType(BaseTitledEnum):
@@ -92,9 +99,9 @@ class ClassTypeEnum(BaseTitledEnum):
     @classmethod
     def get_value(cls, name):
         for enum in list(cls):
-            if enum.name.lower() == name.lower():
+            if enum.__doc__.lower() == name.lower():
                 return enum.value
-        return "object"
+        return cls.OBJECT.value
 
 
 class TrainingStatus(BaseTitledEnum):
@@ -113,7 +120,7 @@ class SegmentationStatus(BaseTitledEnum):
     FAILED = "Failed", 4
 
 
-class TrainingTask(BaseTitledEnum):
+class TrainingTask:
     INSTANCE_SEGMENTATION_PIXEL = (
         "Instance Segmentation for Pixel Projects",
         "instance_segmentation_pixel",

src/superannotate/lib/core/service_types.py

@@ -77,14 +77,17 @@ class ServiceResponse(BaseModel):
     content: Union[bytes, str]
     data: Any
 
-    def __init__(self, response, content_type):
+    def __init__(self, response, content_type=None):
         data = {
             "status": response.status_code,
             "reason": response.reason,
             "content": response.content,
         }
         if response.ok:
-            data["data"] = content_type(**response.json())
+            if content_type:
+                data["data"] = content_type(**response.json())
+            else:
+                data["data"] = response.json()
         super().__init__(**data)
 
     @property

src/superannotate/lib/core/serviceproviders.py

@@ -326,3 +326,9 @@ def get_integrations(self, team_id: int) -> List[dict]:
     def attach_integrations(self, team_id: int, project_id: int, integration_id: int, folder_id: int,
                             folder_name: str) -> bool:
         raise NotImplementedError
+
+    def saqul_query(self, team_id: int, project_id: int, folder_id: int, query: str) -> ServiceResponse:
+        raise NotImplementedError
+
+    def validate_saqul_query(self, team_id: int, project_id: int, query: str) -> dict:
+        raise NotImplementedError

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

@@ -1,4 +1,5 @@
 from lib.core.usecases.annotations import *  # noqa: F403 F401
+from lib.core.usecases.entities import *  # noqa: F403 F401
 from lib.core.usecases.folders import *  # noqa: F403 F401
 from lib.core.usecases.images import *  # noqa: F403 F401
 from lib.core.usecases.integrations import *  # noqa: F403 F401