Merge pull request #200 from superannotateai/sdk_limitations

Added limitation validation

Author: VaghinakDev

src/superannotate/lib/app/common.py

@@ -148,13 +148,6 @@ def save_web_format(output_dir, classes, files_dict):
         json.dump(classes, fw)
 
 
-def dump_output(output_dir, platform, classes, files_dict):
-    if platform == "Web":
-        save_web_format(output_dir, classes, files_dict)
-    else:
-        save_desktop_format(output_dir, classes, files_dict)
-
-
 def write_to_json(output_path, json_data):
     with open(output_path, "w") as fw:
         json.dump(json_data, fw, indent=2)

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

@@ -6,9 +6,6 @@
 import tempfile
 import time
 import uuid
-from collections import Counter
-from collections import namedtuple
-from io import BytesIO
 from pathlib import Path
 from typing import Iterable
 from typing import List

src/superannotate/lib/core/__init__.py

@@ -68,6 +68,18 @@
     ProjectType.DOCUMENT.value: DEPRECATED_DOCUMENT_PROJECTS_MESSAGE,
 }
 
+UPLOAD_FOLDER_LIMIT_ERROR_MESSAGE = "The number of items you want to upload exceeds the limit of 50 000 items per folder."
+UPLOAD_PROJECT_LIMIT_ERROR_MESSAGE = "The number of items you want to upload exceeds the limit of 500 000 items per project."
+UPLOAD_USER_LIMIT_ERROR_MESSAGE = "The number of items you want to upload  exceeds the limit of your subscription plan."
+
+ATTACH_FOLDER_LIMIT_ERROR_MESSAGE = "The number of items you want to attach exceeds the limit of 50 000 items per folder."
+ATTACH_PROJECT_LIMIT_ERROR_MESSAGE = "The number of items you want to attach exceeds the limit of 500 000 items per project."
+ATTACH_USER_LIMIT_ERROR_MESSAGE = "The number of items you want to attach  exceeds the limit of your subscription plan."
+
+COPY_FOLDER_LIMIT_ERROR_MESSAGE = (
+    "The number of items you want to copy exceeds the limit of 50 000 items per folder."
+)
+
 __version__ = "?"
 
 __alL__ = (

src/superannotate/lib/core/service_types.py

@@ -1,32 +1,26 @@
 from typing import Any
 from typing import Dict
-from typing import List
+from typing import Optional
 from typing import Union
+from typing import List
 
-from lib.core.exceptions import AppException
 from pydantic import BaseModel
 from pydantic import Extra
 
 
+class ErrorMessage(BaseModel):
+    error: str
+
+
+class Limit(BaseModel):
+    max_image_count: int
+    remaining_image_count: int
+
+
 class UserLimits(BaseModel):
-    super_user_limit: int
-    project_limit: int
-    folder_limit: int
-
-    def has_enough_slots(self, count: int):
-        if count > self.super_user_limit:
-            raise AppException(
-                "The number of items you want to upload exceeds the limit of your subscription plan."
-            )
-        if count > self.project_limit:
-            raise AppException(
-                "You have exceeded the limit of 500 000 items per project."
-            )
-        if count > self.folder_limit:
-            raise AppException(
-                "“You have exceeded the limit of 50 000 items per folder."
-            )
-        return True
+    super_user_limit: Optional[Limit]
+    project_limit: Limit
+    folder_limit: Limit
 
 
 class UploadAnnotationAuthData(BaseModel):

src/superannotate/lib/core/serviceproviders.py

@@ -315,7 +315,7 @@ def get_annotations_delete_progress(
     ):
         raise NotImplementedError
 
-    def get_limits(
+    def get_limitations(
         self, team_id: int, project_id: int, folder_id: int = None
     ) -> ServiceResponse:
         raise NotImplementedError