Added approve/disapprove items

Author: VaghinakDev

docs/source/superannotate.sdk.rst

@@ -81,6 +81,8 @@ ______
 .. automethod:: superannotate.SAClient.unassign_items
 .. automethod:: superannotate.SAClient.get_item_metadata
 .. automethod:: superannotate.SAClient.set_annotation_statuses
+.. automethod:: superannotate.SAClient.set_approval_statuses
+.. automethod:: superannotate.SAClient.set_approval
 
 ----------
 

pytest.ini

@@ -2,5 +2,5 @@
 minversion = 3.7
 log_cli=true
 python_files = test_*.py
-;pytest_plugins = ['pytest_profiling']
+pytest_plugins = ['pytest_profiling']
 ;addopts = -n auto --dist=loadscope

requirements.txt

@@ -10,7 +10,7 @@ xmltodict==0.12.0
 opencv-python>=4.4.0.42
 wheel==0.35.1
 packaging>=20.4
-plotly==4.1.0
+plotly>=4.1.0
 ffmpeg-python>=0.2.0
 fire==0.4.0
 mixpanel==4.8.3

src/superannotate/__init__.py

@@ -2,7 +2,7 @@
 import sys
 import typing
 
-__version__ = "4.4.9dev2"
+__version__ = "4.4.9dev4"
 
 sys.path.append(os.path.split(os.path.realpath(__file__))[0])
 

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

@@ -23,6 +23,7 @@
 from lib.app.interface.types import AnnotationStatuses
 from lib.app.interface.types import AnnotationType
 from lib.app.interface.types import AnnotatorRole
+from lib.app.interface.types import ApprovalStatuses
 from lib.app.interface.types import AttachmentArg
 from lib.app.interface.types import AttachmentDict
 from lib.app.interface.types import ClassType
@@ -2332,7 +2333,7 @@ def search_items(
                                 ♦ “QualityCheck” \n
                                 ♦ “Returned” \n
                                 ♦ “Completed” \n
-                                ♦ “Skippe
+                                ♦ “Skip” \n
         :type annotation_status: str
 
         :param annotator_email: returns those items’ names that are assigned to the specified annotator.
@@ -2569,13 +2570,13 @@ def set_annotation_statuses(
         :param project: project name or folder path (e.g., “project1/folder1”).
         :type project: str
 
-        :param annotation_status: annotation status to set, should be one of.
-                                    “NotStarted”
-                                    “InProgress”
-                                    “QualityCheck”
-                                    “Returned”
-                                    “Completed”
-                                    “Skipped”
+        :param annotation_status: annotation status to set, should be one of. \n
+                                    ♦ “NotStarted” \n
+                                    ♦ “InProgress” \n
+                                    ♦ “QualityCheck” \n
+                                    ♦ “Returned” \n
+                                    ♦ “Completed” \n
+                                    ♦ “Skipped” \n
         :type annotation_status: str
 
         :param items:  item names to set the mentioned status for. If None, all the items in the project will be used.
@@ -3021,3 +3022,32 @@ def add_items_to_subset(
 
         return response.data
 
+    def set_approval_statuses(
+        self,
+        project: NotEmptyStr,
+        approval_status: Union[ApprovalStatuses, None],
+        items: Optional[List[NotEmptyStr]] = None,
+    ):
+        """Sets annotation statuses of items
+
+        :param project: project name or folder path (e.g., “project1/folder1”).
+        :type project: str
+
+        :param approval_status: approval status to set, should be one of. \n
+                                ♦ None \n
+                                ♦ “Approved” \n
+                                ♦ “Disapproved” \n
+        :type approval_status: str
+
+        :param items:  item names to set the mentioned status for. If None, all the items in the project will be used.
+        :type items: list of strs
+        """
+        project, folder = self.controller.get_project_folder_by_path(project)
+        response = self.controller.items.set_approval_statuses(
+            project=project,
+            folder=folder,
+            approval_status=approval_status,
+            item_names=items,
+        )
+        if response.errors:
+            raise AppException(response.errors)

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

@@ -5,6 +5,7 @@
 from typing import Union
 
 from lib.core.enums import AnnotationStatus
+from lib.core.enums import ApprovalStatus
 from lib.core.enums import BaseTitledEnum
 from lib.core.enums import ClassTypeEnum
 from lib.core.enums import FolderStatus
@@ -199,6 +200,22 @@ def validate(cls, value: Union[str]) -> Union[str]:
         return value
 
 
+class ApprovalStatuses(StrictStr):
+    @classmethod
+    def validate(cls, value: Union[str]) -> Union[str]:
+        if value is None:
+            return value
+        if value.lower() not in ApprovalStatus.values() or not isinstance(value, str):
+            raise TypeError(
+                f"Available approval_status options are {', '.join(map(str, ApprovalStatus.titles()))}."
+            )
+        return value
+
+    @classmethod
+    def __get_validators__(cls):
+        yield cls.validate
+
+
 def validate_arguments(func):
     @wraps(func)
     def wrapped(self, *args, **kwargs):

src/superannotate/lib/app/server/README.rst

@@ -10,4 +10,4 @@ Usage
 
 .. code-block:: bash
 
-        ./run.sh --token=<your team token>
+        ./run.sh --token=<your team token>

src/superannotate/lib/core/__init__.py

@@ -2,6 +2,7 @@
 
 from superannotate.lib.core.config import Config
 from superannotate.lib.core.enums import AnnotationStatus
+from superannotate.lib.core.enums import ApprovalStatus
 from superannotate.lib.core.enums import FolderStatus
 from superannotate.lib.core.enums import ImageQuality
 from superannotate.lib.core.enums import ProjectStatus
@@ -16,7 +17,7 @@
 CONFIG_PATH = "~/.superannotate/config.json"
 CONFIG_FILE_LOCATION = expanduser(CONFIG_PATH)
 LOG_FILE_LOCATION = expanduser("~/.superannotate")
-BACKEND_URL = "https://api.annotate.online"
+BACKEND_URL = "https://api.superannotate.com"
 
 DEFAULT_IMAGE_EXTENSIONS = ["jpg", "jpeg", "png", "tif", "tiff", "webp", "bmp"]
 DEFAULT_FILE_EXCLUDE_PATTERNS = ["___save.png", "___fuse.png"]
@@ -121,6 +122,7 @@
     SegmentationStatus,
     ImageQuality,
     AnnotationStatus,
+    ApprovalStatus,
     CONFIG_FILE_LOCATION,
     CONFIG_PATH,
     BACKEND_URL,

src/superannotate/lib/core/enums.py

@@ -12,7 +12,7 @@ def __get__(self, instance, owner):
 
 class BaseTitledEnum(int, Enum):
     def __new__(cls, title, value):
-        obj = int.__new__(cls, value)
+        obj = super().__new__(cls, value)
         obj._value_ = value
         obj.__doc__ = title
         obj._type = "titled_enum"
@@ -40,15 +40,15 @@ def get_name(cls, value):
     @classmethod
     def get_value(cls, name):
         for enum in list(cls):
-            if enum.__doc__.lower() == name.lower():
+            if enum.__doc__ and name and enum.__doc__.lower() == name.lower():
                 if isinstance(enum.value, int):
                     if enum.value < 0:
                         return ""
                 return enum.value
 
     @classmethod
     def values(cls):
-        return [enum.__doc__.lower() for enum in list(cls)]
+        return [enum.__doc__.lower() if enum else None for enum in list(cls)]
 
     @classmethod
     def titles(cls):
@@ -64,6 +64,12 @@ def __hash__(self):
         return hash(self.name)
 
 
+class ApprovalStatus(BaseTitledEnum):
+    NONE = None, 0
+    DISAPPROVED = "Disapproved", 1
+    APPROVED = "Approved", 2
+
+
 class AnnotationTypes(str, Enum):
     BBOX = "bbox"
     EVENT = "event"
@@ -170,9 +176,3 @@ class SegmentationStatus(BaseTitledEnum):
     IN_PROGRESS = "InProgress", 2
     COMPLETED = "Completed", 3
     FAILED = "Failed", 4
-
-
-class ApprovalStatus(BaseTitledEnum):
-    NONE = None, 0
-    DISAPPROVED = "disapproved", 1
-    APPROVED = "approved", 2

src/superannotate/lib/core/serviceproviders.py

@@ -293,6 +293,16 @@ def set_statuses(
     ) -> ServiceResponse:
         raise NotImplementedError
 
+    @abstractmethod
+    def set_approval_statuses(
+        self,
+        project: entities.ProjectEntity,
+        folder: entities.FolderEntity,
+        item_names: List[str],
+        approval_status: int,
+    ) -> ServiceResponse:
+        raise NotImplementedError
+
     @abstractmethod
     def delete_multiple(
         self, project: entities.ProjectEntity, item_ids: List[int]