Change share project

Author: dshabin

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

@@ -76,6 +76,7 @@ def __init__(
         folder_id: int = None,
         upload_state: int = None,
         users: Iterable = (),
+        unverified_users: Iterable = (),
         contributors: List = None,
         settings: List = None,
         annotation_classes: List = None,
@@ -97,6 +98,7 @@ def __init__(
         self.folder_id = folder_id
         self.upload_state = upload_state
         self.users = users
+        self.unverified_users = unverified_users
         self.contributors = contributors
         self.settings = settings
         self.annotation_classes = annotation_classes

src/superannotate/lib/core/serviceproviders.py

@@ -42,9 +42,7 @@ def get_annotation_classes(
         raise NotImplementedError
 
     @abstractmethod
-    def share_project(
-        self, project_id: int, team_id: int, user_id: str, user_role: int
-    ):
+    def share_project_bulk(self, project_id: int, team_id: int, users: list):
         raise NotImplementedError
 
     @abstractmethod

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

@@ -433,14 +433,20 @@ def _copy_include_contributors(self, to_project: ProjectEntity):
         from_project = self._projects.get_one(
             uuid=self._project.uuid, team_id=self._project.team_id
         )
+        users = []
         for user in from_project.users:
-            self._backend_service.share_project(
-                to_project.uuid,
-                to_project.team_id,
-                user.get("user_id"),
-                user.get("user_role"),
+            users.append(
+                {"user_id": user.get("user_id"), "user_role": user.get("user_role")}
             )
 
+        for user in from_project.unverified_users:
+            users.append(
+                {"user_id": user.get("email"), "user_role": user.get("user_role")}
+            )
+        self._backend_service.share_project_bulk(
+            to_project.uuid, to_project.team_id, users
+        )
+
     def _copy_settings(self, to_project: ProjectEntity):
         new_settings = self._settings_repo(self._backend_service, to_project)
         for setting in self.settings.get_all():
@@ -601,11 +607,10 @@ def user_role(self):
         return constances.UserRole.get_value(self._user_role)
 
     def execute(self):
-        self._response.data = self._service.share_project(
+        self._response.data = self._service.share_project_bulk(
             team_id=self._project_entity.team_id,
             project_id=self._project_entity.uuid,
-            user_id=self._user_id,
-            user_role=self.user_role,
+            users=[{"user_id": self._user_id, "user_role": self.user_role}],
         )
         if not self._response.errors:
             logger.info(

src/superannotate/lib/infrastructure/repositories.py

@@ -134,6 +134,7 @@ def dict2entity(data: dict):
                 description=data["description"],
                 folder_id=data.get("folder_id"),
                 users=data.get("users", ()),
+                unverified_users=data.get("unverified_users", ()),
                 completed_images_count=data.get("completedImagesCount"),
                 root_folder_completed_images_count=data.get(
                     "rootFolderCompletedImagesCount"

src/superannotate/lib/infrastructure/services.py

@@ -174,6 +174,7 @@ class SuperannotateBackendService(BaseBackendService):
     URL_PROJECT_SETTINGS = "project/{}/settings"
     URL_PROJECT_WORKFLOW = "project/{}/workflow"
     URL_SHARE_PROJECT = "project/{}/share"
+    URL_SHARE_PROJECT_BULK = "project/{}/share/bulk"
     URL_ANNOTATION_CLASSES = "classes"
     URL_TEAM = "team"
     URL_INVITE_CONTRIBUTOR = "team/{}/invite"
@@ -421,16 +422,14 @@ def set_project_workflow_bulk(self, project_id: int, team_id: int, steps: list):
         )
         return res.json()
 
-    def share_project(
-        self, project_id: int, team_id: int, user_id: int, user_role: int
-    ):
+    def share_project_bulk(self, project_id: int, team_id: int, users: list):
         share_project_url = urljoin(
-            self.api_url, self.URL_SHARE_PROJECT.format(project_id)
+            self.api_url, self.URL_SHARE_PROJECT_BULK.format(project_id)
         )
         res = self._request(
             share_project_url,
             "post",
-            data={"user_id": user_id, "user_role": user_role},
+            data={"users": users},
             params={"team_id": team_id},
         )
         return res.json()