Merge pull request #203 from superannotateai/friday_140

Fixed SDK | Error when importing superannotate

Author: VaghinakDev

src/superannotate/lib/app/mixp/decorators.py

@@ -26,6 +26,8 @@ def get_default(team_name, user_id, project_name=None):
 
 
 class Trackable:
+    TEAM_DATA = None
+
     def __init__(self, function):
         self.function = function
         self._success = False
@@ -38,8 +40,8 @@ def track(self, *args, **kwargs):
             properties = data["properties"]
             properties["Success"] = self._success
             default = get_default(
-                team_name=controller.team_name,
-                user_id=controller.user_id,
+                team_name=self.__class__.TEAM_DATA[1],
+                user_id=self.__class__.TEAM_DATA[0],
                 project_name=properties.get("project_name", None),
             )
             properties.pop("project_name", None)
@@ -52,11 +54,17 @@ def track(self, *args, **kwargs):
 
     def __call__(self, *args, **kwargs):
         try:
+
+            self.__class__.TEAM_DATA = controller.get_team()
             ret = self.function(*args, **kwargs)
             self._success = True
-            self.track(*args, **kwargs)
         except Exception as e:
             self._success = False
-            self.track(*args, **kwargs)
             raise e
-        return ret
+        else:
+            return ret
+        finally:
+            try:
+                self.track(*args, **kwargs)
+            except Exception:
+                pass

src/superannotate/lib/core/__init__.py

@@ -59,6 +59,15 @@
     "The function does not support projects containing videos attached with URLs"
 )
 
+DEPRECATED_DOCUMENT_PROJECTS_MESSAGE = (
+    "The function does not support projects containing documents attached with URLs"
+)
+
+LIMITED_FUNCTIONS = {
+    ProjectType.VIDEO.value: DEPRECATED_VIDEO_PROJECTS_MESSAGE,
+    ProjectType.DOCUMENT.value: DEPRECATED_DOCUMENT_PROJECTS_MESSAGE,
+}
+
 __version__ = "?"
 
 __alL__ = (

src/superannotate/lib/core/base_usecases.py

@@ -0,0 +1,41 @@
+from abc import ABC
+from abc import abstractmethod
+
+from lib.core.exceptions import AppValidationException
+from lib.core.response import Response
+
+
+class BaseUseCase(ABC):
+    def __init__(self):
+        self._response = Response()
+
+    @abstractmethod
+    def execute(self) -> Response:
+        raise NotImplementedError
+
+    def _validate(self):
+        for name in dir(self):
+            try:
+                if name.startswith("validate_"):
+                    method = getattr(self, name)
+                    method()
+            except AppValidationException as e:
+                self._response.errors = e
+
+    def is_valid(self):
+        self._validate()
+        return not self._response.errors
+
+
+class BaseInteractiveUseCase(BaseUseCase):
+    @property
+    def response(self):
+        return self._response
+
+    @property
+    def data(self):
+        return self.response.data
+
+    @abstractmethod
+    def execute(self):
+        raise NotImplementedError