Merge pull request #599 from superannotateai/1900_auth

changed SDK auth flow

Author: nareksa

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

@@ -132,11 +132,11 @@ def get_mp_instance(self) -> Mixpanel:
         return Mixpanel(mp_token)
 
     @staticmethod
-    def get_default_payload(team_name, user_id):
+    def get_default_payload(team_name, user_email):
         return {
             "SDK": True,
             "Team": team_name,
-            "Team Owner": user_id,
+            "User Email": user_email,
             "Version": __version__,
             "Python version": platform.python_version(),
             "Python interpreter type": platform.python_implementation(),
@@ -207,13 +207,13 @@ def _track_method(self, args, kwargs, success: bool):
             function_name = self.function.__name__ if self.function else ""
             arguments = self.extract_arguments(self.function, *args, **kwargs)
             event_name, properties = self.default_parser(function_name, arguments)
-            user_id = client.controller.team_data.creator_id
+            user_email = client.controller.current_user.email
             team_name = client.controller.team_data.name
 
             properties["Success"] = success
-            default = self.get_default_payload(team_name=team_name, user_id=user_id)
+            default = self.get_default_payload(team_name=team_name, user_email=user_email)
             self._track(
-                user_id,
+                user_email,
                 event_name,
                 {**default, **properties, **CONFIG.get_current_session().data},
             )

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

@@ -1717,7 +1717,7 @@ def upload_annotations_from_folder_to_project(
         response = self.controller.annotations.upload_from_folder(
             project=project,
             folder=folder,
-            team=self.controller.team,
+            user=self.controller.current_user,
             annotation_paths=annotation_paths,  # noqa: E203
             client_s3_bucket=from_s3_bucket,
             folder_path=folder_path,
@@ -1794,7 +1794,7 @@ def upload_image_annotations(
             folder=folder,
             image=image,
             annotations=annotation_json,
-            team=self.controller.team,
+            user=self.controller.current_user,
             mask=mask,
             verbose=verbose,
             keep_status=keep_status,
@@ -2155,7 +2155,9 @@ def add_contributors_to_project(
             for email in emails
         ]
         response = self.controller.projects.add_contributors(
-            team=self.controller.get_team().data, project=project, contributors=contributors
+            team=self.controller.get_team().data,
+            project=project,
+            contributors=contributors,
         )
         if response.errors:
             raise AppException(response.errors)

src/superannotate/lib/core/entities/__init__.py

@@ -16,6 +16,7 @@
 from lib.core.entities.project import ProjectEntity
 from lib.core.entities.project import SettingEntity
 from lib.core.entities.project import TeamEntity
+from lib.core.entities.project import UserEntity
 from lib.core.entities.project import WorkflowEntity
 from lib.core.entities.project_entities import BaseEntity
 from lib.core.entities.project_entities import ImageInfoEntity
@@ -47,6 +48,7 @@
     "S3FileEntity",
     "AnnotationClassEntity",
     "TeamEntity",
+    "UserEntity",
     "MLModelEntity",
     "IntegrationEntity",
 ]

src/superannotate/lib/core/entities/project.py

@@ -169,9 +169,11 @@ class UserEntity(BaseModel):
     first_name: Optional[str]
     last_name: Optional[str]
     email: Optional[str]
-    picture: Optional[str]
     user_role: Optional[int]
 
+    class Config:
+        extra = Extra.ignore
+
 
 class TeamEntity(BaseModel):
     id: Optional[int]

src/superannotate/lib/core/repositories.py

@@ -1,5 +1,4 @@
 from abc import ABC
-from abc import abstractmethod
 from typing import Any
 from typing import List
 from typing import Optional
@@ -12,24 +11,20 @@
 
 
 class BaseReadOnlyRepository(ABC):
-
     def get_one(self, uuid: Union[Condition, int]) -> Optional[Union[BaseModel]]:
         raise NotImplementedError
 
-
     def get_all(self, condition: Optional[Condition] = None) -> List[Union[BaseModel]]:
         raise NotImplementedError
 
 
 class BaseManageableRepository(BaseReadOnlyRepository):
-
     def insert(self, entity: BaseEntity) -> BaseEntity:
         raise NotImplementedError
 
     def update(self, entity: BaseEntity) -> BaseEntity:
         raise NotImplementedError
 
-
     def delete(self, uuid: Any):
         raise NotImplementedError
 

src/superannotate/lib/core/service_types.py

@@ -150,6 +150,10 @@ class TeamResponse(ServiceResponse):
     data: entities.TeamEntity = None
 
 
+class UserResponse(ServiceResponse):
+    data: entities.UserEntity = None
+
+
 class ModelListResponse(ServiceResponse):
     data: List[entities.AnnotationClassEntity] = None
 

src/superannotate/lib/core/serviceproviders.py

@@ -26,6 +26,7 @@
 from lib.core.service_types import UploadAnnotationsResponse
 from lib.core.service_types import UploadCustomFieldValuesResponse
 from lib.core.service_types import UserLimitsResponse
+from lib.core.service_types import UserResponse
 from lib.core.types import Attachment
 from lib.core.types import AttachmentMeta
 
@@ -507,6 +508,10 @@ class BaseServiceProvider:
     def get_team(self, team_id: int) -> TeamResponse:
         raise NotImplementedError
 
+    @abstractmethod
+    def get_user(self, team_id: int) -> UserResponse:
+        raise NotImplementedError
+
     @abstractmethod
     def list_templates(self) -> ServiceResponse:
         raise NotImplementedError

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

@@ -37,7 +37,7 @@
 from lib.core.entities import FolderEntity
 from lib.core.entities import ImageEntity
 from lib.core.entities import ProjectEntity
-from lib.core.entities import TeamEntity
+from lib.core.entities import UserEntity
 from lib.core.exceptions import AppException
 from lib.core.reporter import Reporter
 from lib.core.response import Response
@@ -482,7 +482,7 @@ def __init__(
         reporter: Reporter,
         project: ProjectEntity,
         folder: FolderEntity,
-        team: TeamEntity,
+        user: UserEntity,
         annotation_paths: List[str],
         service_provider: BaseServiceProvider,
         pre_annotation: bool = False,
@@ -493,7 +493,7 @@ def __init__(
         super().__init__(reporter)
         self._project = project
         self._folder = folder
-        self._team = team
+        self._user = user
         self._service_provider = service_provider
         self._annotation_classes = service_provider.annotation_classes.list(
             Condition("project_id", project.id, EQ)
@@ -581,7 +581,7 @@ def prepare_annotation(self, annotation: dict, size) -> dict:
             raise AppException(errors)
 
         annotation = UploadAnnotationUseCase.set_defaults(
-            self._team.creator_id, annotation, self._project.type
+            self._user.email, annotation, self._project.type
         )
         return annotation
 
@@ -827,7 +827,7 @@ def __init__(
         project: ProjectEntity,
         folder: FolderEntity,
         image: ImageEntity,
-        team: TeamEntity,
+        user: UserEntity,
         service_provider: BaseServiceProvider,
         reporter: Reporter,
         annotation_upload_data: UploadAnnotationAuthData = None,
@@ -844,7 +844,7 @@ def __init__(
         self._project = project
         self._folder = folder
         self._image = image
-        self._team = team
+        self._user = user
         self._service_provider = service_provider
         self._annotation_classes = service_provider.annotation_classes.list(
             Condition("project_id", project.id, EQ)
@@ -973,7 +973,7 @@ def execute(self):
             annotation_json, mask = self._get_annotation_json()
             errors = self._validate_json(annotation_json)
             annotation_json = UploadAnnotationUseCase.set_defaults(
-                self._team.creator_id, annotation_json, self._project.type
+                self._user.email, annotation_json, self._project.type
             )
             if not errors:
                 annotation_file = io.StringIO()

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

@@ -507,7 +507,9 @@ def execute(self):
                 failed_images = []
                 while len(success_images) + len(failed_images) != len(image_ids):
                     images_metadata = self._service_provider.items.list_by_names(
-                        project=self._project, folder=self._folder, names=self._images_list
+                        project=self._project,
+                        folder=self._folder,
+                        names=self._images_list,
                     ).data
 
                     success_images = [

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

@@ -609,6 +609,23 @@ def execute(self):
         return self._response
 
 
+class GetCurrentUserUseCase(BaseUseCase):
+    def __init__(self, service_provider: BaseServiceProvider, team_id: int):
+        super().__init__()
+        self._service_provider = service_provider
+        self._team_id = team_id
+
+    def execute(self):
+        response = self._service_provider.get_user(self._team_id)
+        if not response.ok:
+            self._response.errors = AppException(
+                "Unable to retrieve user data. Please verify your credentials."
+            )
+        else:
+            self._response.data = response.data
+        return self._response
+
+
 class SearchContributorsUseCase(BaseUseCase):
     def __init__(
         self,