Fixed wrong imports

Author: Vaghinak Basentsyan

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

@@ -3,7 +3,6 @@
 from typing import Union
 
 from lib.core.enums import AnnotationStatus
-from lib.core.exceptions import AppException
 from pydantic import constr
 from pydantic import StrictStr
 from pydantic import validate_arguments as pydantic_validate_arguments

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

@@ -1,9 +1,9 @@
 import functools
 import sys
 
+from lib.infrastructure.controller import Controller
 from mixpanel import Mixpanel
-from superannotate.lib.infrastructure.controller import Controller
-from superannotate.version import __version__
+from version import __version__
 
 from .config import TOKEN
 from .utils import parsers

src/superannotate/lib/app/mixp/utils/parsers.py

@@ -1,9 +1,9 @@
 import logging
 
-import superannotate.lib.core as constances
-from superannotate.lib.app.helpers import extract_project_folder
-from superannotate.lib.core.enums import ProjectType
-from superannotate.lib.infrastructure.controller import Controller
+import lib.core as constances
+from lib.app.helpers import extract_project_folder
+from lib.core.enums import ProjectType
+from lib.infrastructure.controller import Controller
 
 controller = Controller(logger=logging.getLogger())
 

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

@@ -833,7 +833,7 @@ def execute(self):
         try:
             self._response.data = self._teams.get_one(self._team_id)
         except Exception:
-            raise AppException("Can't get team data.")
+            raise AppException("Can't get team data.") from None
         return self._response
 
 

src/superannotate/lib/infrastructure/controller.py

@@ -68,12 +68,11 @@ def init(self, config_path):
             self.configs.get_one("token"),
             self.configs.get_one("main_endpoint"),
         )
-        if token:
-            token = token.value
-        if main_endpoint:
-            main_endpoint = main_endpoint.value
+        token = None if not token else token.value
+        main_endpoint = None if not main_endpoint else main_endpoint.value
         if not main_endpoint:
             self.configs.insert(ConfigEntity("main_endpoint", constances.BACKEND_URL))
+            main_endpoint = constances.BACKEND_URL
         if not token:
             self.configs.insert(ConfigEntity("token", ""))
             self._logger.warning("Fill config.json")
@@ -94,9 +93,18 @@ def init(self, config_path):
             self._backend_client.api_url = main_endpoint
             self._backend_client._auth_token = token
             self._backend_client.get_session.cache_clear()
-        self._team_id = int(self.configs.get_one("token").value.split("=")[-1])
+        token = self.configs.get_one("token").value
+        self.validate_token(token)
+        self._team_id = int(token.split("=")[-1])
         self._team = None
 
+    @staticmethod
+    def validate_token(token: str):
+        try:
+            return int(token.split("=")[-1])
+        except Exception:
+            raise AppException("Invalid token.")
+
     @property
     def config_path(self):
         return self._config_path

tests/unit/test_controller_init.py

@@ -3,6 +3,7 @@
 import json
 from contextlib import contextmanager
 import pkg_resources
+import tempfile
 from unittest import TestCase
 from unittest.mock import mock_open
 from unittest.mock import patch
@@ -65,3 +66,15 @@ def test_init_create(self, input_mock):
                     )
                 )
                 self.assertEqual(out.getvalue().strip(), "Configuration file successfully created.")
+
+
+class SKDInitTest(TestCase):
+
+    def test_init_flow(self):
+        with tempfile.TemporaryDirectory() as temp_dir:
+            token_path = f"{temp_dir}/config.json"
+            with open(token_path, "w") as temp_config:
+                json.dump({"token": "token=1234"}, temp_config)
+                temp_config.close()
+                import src.superannotate as sa
+                sa.init(token_path)