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
 
 ----------
 

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: Optional[ApprovalStatuses],
+        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,16 @@ def validate(cls, value: Union[str]) -> Union[str]:
         return value
 
 
+class ApprovalStatuses(StrictStr):
+    @classmethod
+    def validate(cls, value: Union[str]) -> Union[str]:
+        if value.lower() not in ApprovalStatus.values():
+            raise TypeError(
+                f"Available an approval_statuses are {', '.join(ApprovalStatus.titles())}. "
+            )
+        return value
+
+
 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]

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

@@ -53,7 +53,6 @@
 BIG_FILE_THRESHOLD = 15 * 1024 * 1024
 ANNOTATION_CHUNK_SIZE_MB = 10 * 1024 * 1024
 URI_THRESHOLD = 4 * 1024 - 120
-nest_asyncio.apply()
 
 
 @dataclass
@@ -376,6 +375,7 @@ def execute(self):
                 len(items_to_upload), description="Uploading Annotations"
             )
             try:
+                nest_asyncio.apply()
                 asyncio.run(self.run_workers(items_to_upload))
             except Exception:
                 logger.debug(traceback.format_exc())
@@ -720,6 +720,7 @@ def execute(self):
             except KeyError:
                 missing_annotations.append(name)
         try:
+            nest_asyncio.apply()
             asyncio.run(self.run_workers(items_to_upload))
         except Exception:
             logger.debug(traceback.format_exc())
@@ -917,6 +918,7 @@ def execute(self):
                 json.dump(annotation_json, annotation_file)
                 size = annotation_file.tell()
                 annotation_file.seek(0)
+                nest_asyncio.apply()
                 if size > BIG_FILE_THRESHOLD:
                     uploaded = asyncio.run(
                         self._service_provider.annotations.upload_big_annotation(
@@ -1109,6 +1111,7 @@ def execute(self):
             )
             small_annotations = [x["name"] for x in items["small"]]
             try:
+                nest_asyncio.apply()
                 annotations = asyncio.run(
                     self.run_workers(items["large"], small_annotations)
                 )
@@ -1355,14 +1358,6 @@ def download_annotation_classes(self, path: str):
     def get_items_count(path: str):
         return sum([len(files) for r, d, files in os.walk(path)])
 
-    @staticmethod
-    def coroutine_wrapper(coroutine):
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        count = loop.run_until_complete(coroutine)
-        loop.close()
-        return count
-
     async def download_big_annotations(self, queue_idx, export_path):
         while True:
             cur_queue = self._big_file_queues[queue_idx]
@@ -1463,6 +1458,7 @@ def execute(self):
             if not folders:
                 folders.append(self._folder)
             with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:
+                nest_asyncio.apply()
                 futures = []
                 for folder in folders:
                     if not self._item_names: